POJ 3844 Divisible Subsequences G++

#include <iostream>
#include <cstring>
#include <cstdio>
using namespace std;
//巧妙 抄博友程序 
long long mp[1000002];
int main()
{
	int T;
	//cin>>T;
	scanf("%d",&T);
	for(int o=0;o<T;o++)
	{
		int mod,n;
		scanf("%d%d",&mod,&n);
		//cin>>mod>>n;
		memset(mp,0,sizeof(mp));
		int b=0;
		for(int i=1;i<=n;i++)
		{
			int a;
			//cin>>a;
			scanf("%d",&a);
			b=(b+a)%mod;
			mp[b]++;
		}
		long long jg=0;		
		for(int i=0;i<mod;i++)//抄博友程序
		{
			if(i==0)
			{
				jg=jg+mp[0];
			}
			jg=jg+(mp[i]*(mp[i]-1))/2;
		}
		printf("%lld\n",jg);		
		//cout<<jg<<endl;
	}
	return 0;
}