poj 2356 Find a multiple【抽屉原理】

题目链接：http://poj.org/problem?id=2356

题目大意：题目大意就是先给出一个数N，接着再给出N个数，要你从这N个数中任意选择1个或多个数，使得其和是N的倍数如果找不到这样的答案 则输出0，答案可能有多个，但只用任意输出一个解就行。

思路：可以先计算出S1,S2,S3……Sn（表示从第一个到第n个数字的和）；如果其中Si%N==0的，直接输出；

抽屉原理：把多于n个的物体放到n个抽屉里，则至少有一个抽屉里有2个或2个以上的物体。

           否则，就让每个S1~n对n取模，如果最终答案不是0，那么肯定有Si%N==Sj%N,那么假设j>n,则(Sj-Si)%N==0,所以输出从i+1到 j 的数就OK了；

#include<stdio.h>
#include<string.h>
#include<string>
#include<queue>
#include<stack>
#include<map>
#include<vector>
#include<set>
#include<algorithm>
#define inf 0x3f3f3f3f
using namespace std;
typedef long long ll;
const int N=1e6+10;

int sum[N];
int num[N];
int vis[N];

int main()
{
    int n;
    while(~scanf("%d",&n))
    {
        int flag=0;
        memset(vis,0,sizeof(vis));
        memset(sum,0,sizeof(sum));
        for(int i=1;i<=n;i++)
            scanf("%d",&num[i]);
        for(int i=1;i<=n;i++)
        {
            sum[i]=sum[i-1]+num[i];
            if(sum[i]%n==0)//如果当前sum[i]%n==0,直接输出
            {
                flag=1;
                printf("%d\n",i);
                for(int j=1;j<=i;j++)
                    printf("%d\n",num[j]);
                break;//记得加上break，第一次就是这里没加WA了
            }
        }
        if(!flag)
        {
            int t;
            for(int i=1;i<=n;i++)
            {
                t=sum[i]%n;
                if(vis[t]!=0)
                {
                    printf("%d\n",i-vis[t]);
                    for(int j=vis[t]+1;j<=i;j++)
                        printf("%d\n",num[j]);
                    break;
                }
                vis[t]=i;
            }
        }
    }
    return 0;
}