HUST team contest #2 C Divisible Subsequences ||poj 3844 (剩余类)
算是签到帖,竟然卡住了。
我数学还是太差了。。
然后去找题解。。竟然看不懂!
我数学真的有这么差嘛。。。
然后多亏了队友 @zcy 妹子的讲解。。
其实很好理解啊。。。
不过数到底还是数学太差了==
今天七夕,废话有点多==
这道题的题意是,找序列中连续的一段,使得和 可以整除d
我们观察到数列中的数的范围是1..1 000 000 000 的
而d只有1 000 000
我们考虑余数相同,读入的时候就可以直接先 % 掉 d
因为只会和余数有关。
我们可以先处理出来一个前缀和数组sum[i],表示前i个元素的和。
如果有一段序列从a[i] 到 a[j] 满足题意
根据题意那么这段序列的和一定%d=0
a[i] 到 a[j]的和为 sum[j]-sum[i-1]
也就是(sum[j]-sum[i-1] )%d==0
也就是sum[j] 和 sum[i-1] 关于 d 同余
如果我们在处理得到前缀和的时候就%dshuxue
那么就是sum[j]==sum[i-1]
所以我只要对于d的每一个剩余类有多少个统计出来。
然后对于每个剩余类,只需要任意取出两个,就是一种答案。
需要注意的是如果只有一个0,也是一种答案。
我们可以定义sum[i] = 0
还有一个需要注意的是要开long long
/* ***********************************************
Author :111qqz
Created Time :2016年02月29日 星期一 21时20分01秒
File Name :code/poj/3844.cpp
************************************************ */
#include<iostream>
#include<iomanip>
#include<cstdio>
#include<algorithm>
#include<cmath>
#include<cstring>
#include<string>
#include<map>
#include<set>
#include<queue>
#include<vector>
#include<stack>
#define y0 abc111qqz
#define y1 hust111qqz
#define yn hez111qqz
#define j1 cute111qqz
#define tm crazy111qqz
#define lr dying111qqz
using namespace std;
#define REP(i, n) for (int i=0;i<int(n);++i)
typedef long long LL;
typedef unsigned long long ULL;
const int inf = 0x7fffffff;
const int N=1E6+7;
int sum[50005];
LL p[N];
void solve ()
{
int d,n;
int x;
memset(p,0,sizeof(p));
sum[0] = 0;
p [0] = 1;
for ( int i = 1 ; i <= n ; i++){
scanf("%d",&x);
sum[i] = (sum[i-1] + x) % d;
p[sum[i]]++;
}
LL ans = 0 ;
for ( int i = 0 ; i < d ; i++){
if (p[i]>0){
ans = ans + p[i]*(p[i]-1)/2;
}
}
cout<<ans<<endl;
}
int main()
{
int T;
cin>>T;
while (T--){
solve();
}
return 0;
}