Codeoforces 577B Modulo Sum (鸽巢定理+dp)

题目链接: http://codeforces.com/problemset/problem/577/B

题意：给出n个数，冲中挑选若干个数加和，如果可以整除m则输出yes

思路：首先有一个数学定理:  设s1=a1+a2+a3+……+al，s2=a1+……+al+al+1+al+2+……ar，

           若s2%m = s1%m 

               ->s2%m - s1%m 

               -> (s2 - s1)%m

               -> (al+1 + …… + ar)%m = 0

当n>m 时  必定有2个序列和%m结果相同(鸽巢定理)

当n<= 时  用类似01背包(我觉得有点像数位dp……)的方式将所有的mod处理出来，如果出现同样的mod则成立，

dp[i][j] i表示第i个数，j表示mod的结果，当之前出现过 ((j-s[i])+m)%m的mod时加上s[j]%mod可以得出j(说的有些混乱具体看代码)

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#define maxn 1000030
using namespace std;

int s[maxn],dp[1030][1030];

int main()
{
    int n,m;
    while (scanf("%d%d",&n,&m)!=EOF)
    {
        memset(dp,0,sizeof(dp));
        for (int i=1;i<=n;i++)
        {
            scanf("%d",&s[i]);
        }

        if (n>m)
        {
            printf("YES\n");
        }
        else
        {
            int flag=0;
            dp[0][0]=1;
            for (int i=1;i<=n;i++)
            {
                for (int j=0;j<=m-1;j++)
                {
                    int tmp=((j-s[i])%m+m)%m;

                    if (dp[i-1][tmp])
                    {
                        dp[i][j]=1;

                        if (j==0) flag=1;
                    }
                    if (dp[i-1][j]) dp[i][j]=1;
                }
            }
            if (flag) printf("YES\n");
            else printf("NO\n");
        }
    }
}