HDU 5776 sum(前缀和+取模)

题目链接：HDU 5776

题意：给定一个数列，求是否存在连续子列和为m的倍数，存在输出YES，否则输出NO。

题解：

预处理前缀和，一旦有两个数模m的值相同，说明中间一部分连续子列可以组成m的倍数。为什么？假设sum[1,i]%m=k，sum[1,j]%m=k,则sum[i+1,j]%m=0。

由鸽巢原理，可以知道一旦n>m,则必定会有某两个前缀和对m取模相等。

复杂度O(n).

代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int MAX=100000+10;
int a[MAX],sum[MAX];
int vis[MAX];
int main()
{
    //freopen("in.txt","r",stdin);
    //freopen("out.txt","w",stdout);
    int T;
    scanf("%d",&T);
    while(T--)
    {
        int n,m;
        scanf("%d%d",&n,&m);
        sum[0]=0;
        memset(vis,0,sizeof(vis));
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&a[i]);
            sum[i]=sum[i-1]+a[i];
            vis[sum[i]%m]++;
        }
        int flag=0;
        if(vis[0]) flag=1;
        else 
        {
            for(int i=1;i<m;i++)
            {
                if(vis[i]>=2)
                {
                    flag=1;
                    break;
                }
            }
        }
        printf("%s\n",flag?"YES":"NO");
    }
    return 0;
}