HDU 5776 SUM （鸽巢原理 / 抽屉原理）

原题网址：

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 <iostream>
#include <string.h>
#include <cstdio>

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\n");
        else{
            if(flag == 1)printf("YES\n");
            else{
                printf("NO\n");
            }
        }
        sum = 0;
        flag = 0;
        memset(mm,0,sizeof(mm));
    }
    return 0;
}