51nod 1103 抽屉原理

51nod 1103

#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
const int maxn=5e4+5;
int main()
{
    int a[maxn],n;
    scanf("%d",&n);
    bool flag=false;
    int solve;
    for(int i=0;i<n;i++)
    {
        scanf("%d",&a[i]);
        if(a[i]%n==0) {
            flag=true;
            solve=a[i];
        }
    }
    if(flag) {
        printf("1\n%d\n",solve);
        return 0;
    }
    int l,r;
    LL sum[maxn];
    sum[0]=a[0]%n;
    for(int i=1;i<n;i++)
    {
        sum[i]=(sum[i-1]+a[i]%n)%n;
        //cout<<sum[i]<<endl;
    }
    int done[maxn];
    memset(done,-1,sizeof(done));
    for(int i=0;i<n;i++)
    {
        if(sum[i]==0) {
            flag=true;
            l=0,r=i;
            break;
        }
        else {
            if(done[ sum[ i ] ]==-1) {
                done[ sum[i] ]=i;
            }
            else {
                flag=true;
                l=done[sum[i]]+1,r=i;
                break;
            }
        }
        if(flag) break;
    }
    printf("%d\n",r-l+1);
    for(int i=l;i<=r;i++)
        printf("%d\n",a[i]);
    return 0;
}