POJ 2356 Find a multiple（鸽巢定理）

题目链接

知道了用这种方法做，想了会，没整出来，翻翻书，看到一个类似的。组合数学17页有个类似的例题，证明一定有解。

大体过程，用sum[i]，表示前i个的和对n取余，共有n个元素，这n个数里，如果存在0，很明显这是答案，如果不存在，则至少有两个数相同，sum[j]-sum[k] = 0，则这一段的和就是所求解的答案。难得1Y啊。。

#include <cstdio>
#include <cstring>
#include <cmath>
#include <map>
using namespace std;
int sum[10001],p[10001],o[10001];
int main()
{
    int i,n,j;
    scanf("%d",&n);
    for(i = 1;i <= n;i ++)
    {
        scanf("%d",&p[i]);
        sum[i] = (sum[i-1]+p[i])%n;
    }
    for(i = 1;i <= n;i ++)
    {
        if(sum[i] == 0)
        {
            printf("%d\n",i);
            for(j = 1;j <= i;j ++)
            printf("%d\n",p[j]);
            break;
        }
        if(!o[sum[i]])
        o[sum[i]] = i;
        else
        {
            printf("%d\n",i-o[sum[i]]);
            for(j = o[sum[i]]+1;j <= i;j ++)
            {
                printf("%d\n",p[j]);
            }
            break;
        }
    }
    return 0;
}