【51Nod 1103】N的倍数

Description

一个长度为N的数组A，从A中选出若干个数，使得这些数的和是N的倍数。
例如：N = 8，数组A包括：2 5 6 3 18 7 11 19，可以选2 6，因为2 + 6 = 8，是8的倍数。

Solution

好机智的一道题目。想了好久才想出来。
首先想到的是每个数都要去mod N。然后这些数只会出现在[0,N-1]之间。
之后就不知道要怎么做了。
然后观察了一下，那么数的范围，如果是0的话，那么直接就找到答案了。
那么还剩下N-1个数值。
噢，发现了有N个数值，那么根据抽屉原理，一定会有两个数值是相同的。
但是这两个数又不能互相相减，那么怎么办的？
我们现在需要一个可以互相相减的又能表示一些数的和的方法——前缀和。
不这个n个前缀和mod n一下也符合抽屉原理。
那么就很简单了。

Code

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#define fo(i,a,b) for(i=a;i<=b;i++)
using namespace std;
int i,j,k,l,t,n,m,ans;
int a[50007],sum[50007],c[50007];
int main(){
    scanf("%d",&n);
    fo(i,1,n){
        scanf("%d",&a[i]);
        sum[i]=(sum[i-1]+a[i])%n;
        if(!sum[i]){
            printf("%d\n",i);
            fo(j,1,i){
                printf("%d\n",a[j]);
            }
            return 0;
        }
        if(c[sum[i]]){
            printf("%d\n",i-c[sum[i]]);
            fo(j,c[sum[i]]+1,i){
                printf("%d\n",a[j]);
            }    
            return 0;
        }
        c[sum[i]]=i;
    }
}