HDU 5776 SUM (鸽巢原理 / 抽屉原理)

2019-04-14 17:37发布

原题网址:

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; }