[杂题] Codeforces #577B. Modulo Sum

来源:互联网 发布:java 效率最高的排序 编辑:程序博客网 时间:2024/06/07 01:06

简单题。
a1,a1+a2,a1+a2+a3,a1+a2+a3+a4...
抽屉原理,这些数模m一定会有相等的或为 0 的。容易想到,当 nm 时是一定有解的。
所以直接 O(m2) 搞搞就好了。

#include<cstdio>#include<cstring> #include<algorithm>using namespace std;const int maxn=5005;int n,m,a[maxn];int f[2][maxn];int main(){    freopen("cf577B.in","r",stdin);    freopen("cf577B.out","w",stdout);    scanf("%d%d",&n,&m);    if(n>=m) return puts("YES\n"),0;    for(int i=1;i<=n;i++) scanf("%d",&a[i]), a[i]%=m;    f[1][a[1]]=1;    for(int i=1;i<=n-1;i++){        memset(f[(i&1)^1],0,sizeof(f[(i&1)^1]));        f[(i&1)^1][a[i+1]]=1;        for(int j=0;j<=m-1;j++) if(f[(i&1)][j]) f[(i&1)^1][j]|=1, f[(i&1)^1][(j+a[i+1])%m]|=1;     }    if(f[n&1][0]) printf("YES\n"); else printf("NO\n");    return 0; }