给定一个序列,你会被问到是否存在一个连续的子序列,其总和可以被m整除。输出YES,否则输出NO。
Input
输入包含多组测试用例,输入的第一行是单个整数T(0<T≤10),它是测试用例的数量。T的测试用例如下:你将在每个测试用例里输入许多案例,每个案例第一行包括两个数n,m(1≤n≤100000, 1≤m≤5000);第二行是按顺序输入n个正整数x(1≤x≤100)。
Output
输出T行,每行打印YES或NO。
Sample Input
2 3 3 1 2 3 5 7 6 6 6 6 6
Sample Output
YES NO
这题我们可以用到前缀和,[(d+m)%k=d]==[d%k=d].
#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
int t=0,n=0,m=0,a[100005],sum[100005];
bool vis[5005];
int main(){
cin>>t;
while(t--){
memset(vis,0,sizeof(vis));
memset(a,0,sizeof(a));
memset(sum,0,sizeof(sum));
cin>>n>>m;
bool f=0;
for(int i=1;i<=n;i++)scanf("%d",&a[i]);
for(int i=1;i<=n;i++){
sum[i]=a[i]+sum[i-1];
sum[i]%=m;
if(sum[i]==0){
f=1;
break;
}
}
if(f!=0){
printf("YES\n");
continue;
}
for(int i=1;i<=n;i++){
if(!vis[sum[i]])vis[sum[i]]=1;
else{
f=1;
break;
}
}
if(f!=0)cout<<"YES"<<endl;
else cout<<"NO"<<endl;
}
return 0;
}