Find a multiple POJ - 2356

题面

题意

给出n个数,是否可以选出一些数使它们的和为n的倍数,能则输出方案,反之输出0.

方法

这题一开始只想到n^2的搜索,TLE了.
看了题解之后发现可以用抽屉原理证明一定存在一个解是一些连续的数.
因为一共n个前缀和,它们对n取模一定存在两个相同或是为0.
如果为0,输出这个前缀.
如果有两个相同,则输出它们中间的这段数.

代码

#include<iostream>
#include<cstdio>
#define N 10010
using namespace std;

int n,qz[N],jl[N],num[N];

int main()
{
    int i,j;
    cin>>n;
    for(i=1;i<=n;i++)
    {
        scanf("%d",&num[i]);
        qz[i]=(qz[i-1]+num[i])%n;
        if(jl[qz[i]]||!qz[i])
        {
            printf("%d\n",i-jl[qz[i]]);
            for(j=jl[qz[i]]+1;j<=i;j++)
            {
                printf("%d\n",num[j]);
            }
            return 0;
        }
        jl[qz[i]]=i;
    }
}