YBTOJ E. 1.合法序列

思路：

找到两个当两个前缀和mod k值相等时 就证明这里有一个子序列（好像完事了

所以计算所有子前缀和值mod k的值的个数，再用杨辉三角算组合数即可

Code

#include<bits/stdc++.h>
#define re register 

#define inl inline

#define ll long long

using namespace std;
const int N=5e4+10;
int read(){
	int sum=0,f=1;char c=getchar();
	while(!isdigit(c)){if(c=='-') f=-1;c=getchar();}
	while(isdigit(c)){sum=(sum<<3)+(sum<<1)+(c^48);c=getchar();}
	return sum*f;
}
int T,n,k;
int s[N];
ll c[N][3];
int cnt[1000010];
void init(){
	for(re int i=0;i<=N;i++) c[i][0]=1;
	for(int i=1;i<=N;i++){
		for(int j=1;j<=2;j++){
			c[i][j]=c[i-1][j]+c[i-1][j-1];
		}
	}
}
int ans=0;
int main(){
	T=read();
	init();
	while(T--){
		ans=0;
		memset(cnt,0,sizeof(cnt));
		cnt[0]=1;
		memset(s,0,sizeof(s));
		k=read(),n=read();
		for(re int i=1;i<=n;i++){
			int x=read();
			s[i]=(s[i-1]+x)%k;
			cnt[s[i]]++;
		}
		for(re int i=0;i<k;i++){
			ans+=c[cnt[i]][2];
			//cout<<cnt[i]<<endl;
		}
		printf("%lld\n",ans);
	}
}