POJ 2356 鸽巢原理

来源：互联网发布：数据库round down函数编辑：程序博客网时间：2024/05/16 07:30

题意：给定n个数，从中选出连续的若干个，使得和为n的倍数。多解时输出任意解。

这题运用鸽巢原理，首先用s[1],s[2],s[3]...s[n]表示第一个数到第n个数的和也就是s[n]=a[1]+...+a[n]，然后将s[i]分别对n取余，显然，如果s[i]=0那么此时1-i个数即为n的倍数也就是答案，如果s[i]均不为0，那么通过鸽巢原理可知s[i]个数为n，1<=s[i]<=n-1，那么至少有两个s[i] 即s[i]=s[j] (i!=j) 所以此时i+1至j 即为答案

#include <iostream>#include<cstdio>#include<cstring>using namespace std;int s[10005],a[10005];int main(){    int n,f,ans1,ans2;    while(~scanf("%d",&n))    {        s[0]=f=0;        for(int i=1; i<=n; i++)        {            scanf("%d",&a[i]);            if(f)                continue;            s[i]=(s[i-1]+a[i])%n;            if(s[i]==0)                f=1,ans1=i;            for(int j=1; j<i; j++)                if(s[j]==s[i])                {                    f=2,ans1=j,ans2=i;                    break;                }        }        if(f==1)        {            printf("%d\n",ans1);            for(int i=1; i<=ans1; i++)                printf("%d\n",a[i]);        }        if(f==2)        {            printf("%d\n",ans2-ans1);            for(int i=ans1+1; i<=ans2; i++)                printf("%d\n",a[i]);        }    }    return 0;}