Find a multiple POJ

Problem Description
The input contains N natural (i.e. positive integer) numbers ( N <= 10000 ). Each of that numbers is not greater than 15000. This numbers are not necessarily different (so it may happen that two or more of them will be equal). Your task is to choose a few of given numbers ( 1 <= few <= N ) so that the sum of chosen numbers is multiple for N (i.e. N * k = (sum of chosen numbers) for some natural number k).

Input 
The first line of the input contains the single number N. Each of next N lines contains one number from the given set.

Output
In case your program decides that the target set of numbers can not be found it should print to the output the single number 0. Otherwise it should print the number of the chosen numbers in the first line followed by the chosen numbers themselves (on a separate line each) in arbitrary order. 

If there are more than one set of numbers with required properties you should print to the output only one (preferably your favorite) of them.

Sample Input
5
1
2
3
4
1

Sample Output
2
2
3

题意：给出一个含有 n 个数字的序列，要找一个连续的子序列，使他们的和一定是 n 的倍数

思路：抽屉原理经典应用

AC代码：

#include <stdio.h>
#include <iostream>
#include <string.h>
#include <algorithm>

using namespace std;
#define N 100502
int arr[N];
int vis[N];
int sum[N];
int main(){
    int n;
    while(~scanf("%d",&n)){
        int flag=0;
        sum[0]=0;
        for(int i=1;i<=n;i++){
            scanf("%d",&arr[i]);
            sum[i]=sum[i-1]+arr[i];
            if(sum[i]%n==0){
                flag=i;
            }
        } 
        if(flag){
            printf("%d
",flag);
            for(int i=1;i<=flag;i++){
                printf("%d
",arr[i]);
            }
            continue;
        }
        for(int i=1;i<=n;i++){
            if(vis[sum[i]%n]){
                int ans=i-(vis[sum[i]%n]);
                printf("%d
",ans);
                for(int j=vis[sum[i]%n]+1;j<=i;j++){
                    printf("%d
",arr[j]);
                }    
                break;
            } 
            vis[sum[i]%n]=i;
        }    
    }

    return 0;
}

/*
5
4 3 4  3
4 7 11 14

*/