POJ 3844Divisible Subsequences

[Description]
给定一个由正整数构成的序列，并给出一个数 d，求序列中存在多少个连续子序列，使得
子序列之和能被 d 整除。
[Input]
第一行一个数 T，表示有 T 组测试数据。
每组测试数据第一行两个数，d，n 表示除数和序列的长度
接下来一行 n 个不大于 10^9 的数，表示这个序列。
[Output]
对于每组测试数据输出一个数，表示找到符合条件的子串的数目。
[Sample Input]
2
7 3
1 2 3
4 8
2 1 2 1 1 2 1 2
[Sample Output]
0
6
[Hint]

1<=d<= 1,000,000  1<=n<=50,000

思路：

首先看数据范围，1<=n<=50000，用n^2的算法肯定要超时。所以应该用O(n)的算法。
可以发现对于一个数n，n满足n%d=x，要使n能整除d，只需用n减去一
个模上d也余x的数，所得到的数就一定能整除d。


我们可以现将1到每个数的和sum%d算出来，对于当前的数n，只需找在n之前的数模上d的值也为x的个数找到，将个数求和就是答案。
需要特判，如果n%d=0，ans=ans+1；

代码：

#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std;

int d,n;
long long ans;
int b[1000000+10];
int a[50000+10];


int main()
{
int t;
scanf("%d",&t);
for(int u=1;u<=t;u++)
{
memset(b,0,sizeof(b));
ans=0;
scanf("%d%d",&d,&n);
for(int i=1;i<=n;i++)
{
int x;
scanf("%d",&x);
a[i]=(a[i-1]+x%d)%d;
if(a[i]==0)
{
ans++;
}
ans=ans+b[a[i]];
b[a[i]]++;
}
printf("%I64d\n",ans);
}
return 0;
}