原题网址:
http://acm.hdu.edu.cn/showproblem.php?pid=5776
题意:
t组测试样例,每次给你n,m和n个数字,问你这个序列里存不存在某个子串(连续)的和模m的值等于0
思路:
预处理前缀和sum,因为问的是模m,所以最多有m种情况:0,1,2.。。。。m-1。如果我们找到两个取模后相等的前缀和sum,那么那么这两个序列相差的那部分一定模m等于0。所以当n >= m时,前缀和的情况有m种,这大于对m取模的结果数,就像m-1个抽屉,你非要放m个物品,那么必有一个抽屉会有大于等于两件物品。
特殊判断一种情况:当n < m 时,每个能出现的前缀和模m的值都只出现一次 ,但是有某个前缀和模m等于0,那么也输出YES。
#include
#include
#include
using namespace std;
int mm[5020];
int main()
{
int T;
cin>>T;
int n,m;
int sum = 0;
int flag = 0;
while(T--){
scanf("%d%d",&n,&m);
int temp;
for(int i = 0;i < n;i++){
scanf("%d",&temp);
sum += temp;
mm[sum%m]++;
if(mm[sum%m] == 2)flag = 1;
}
if(mm[0] != 0)flag = 1;
if(n >= m)printf("YES
");
else{
if(flag == 1)printf("YES
");
else{
printf("NO
");
}
}
sum = 0;
flag = 0;
memset(mm,0,sizeof(mm));
}
return 0;
}