Find a multiple-鸽巢原理的妙用

题意：
给你一个数n，和n个数，要你从这n个数中选取几个数，使得这些数的和是n的倍数。
思路:
这种题目属于那种有很多可行解的，要你随便输出一个，所以里面很可能有某种规律。不要想复杂了，从其中间隔的挑选，这就变成了多阶段鞠策问题了。
想到两个前缀和%n相等的区间之差就是答案，你就会豁然开朗。
鸽巢原理的妙用:
求前缀和，pre%n有n中可能，如果mod == 0，那么1->i就是我们要的结果。
否则%n有余数，1->n-1（排除余数为0的情况），中共n-1中状态，而有n个余数，所以必然有两个余数相等，那么余数相等的
两个区间差就是答案。

AC code:
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<vector>
#include<set>
#include<stack>
#include<queue>
#include<map>
#include<cstring>
#include<string>
#include<cmath>

using namespace std;

typedef long long LL;

#define INF 0x3f3f3f3f
#define PI acos(-1.0)
#define pii pair<int,int>
#define all(x) x.begin(),x.end()
#define mem(a,b) memset(a,b,sizeof(a))
#define per(i,a,b) for(int i = a;i <= b;++i)
#define rep(i,a,b) for(int i = a;i >= b;--i)
const int maxn = 1e4;
int n = 0;
int a[maxn+10],pre[maxn+10];
int mod[maxn+10];
/*
鸽巢原理的妙用: 
求前缀和，pre%n有n中可能，如果mod == 0，那么1->i就是我们要的结果。
否则%n有余数，1->n-1（排除余数为0的情况），中共n-1中状态，而有n个余数，所以必然有两个余数相等，那么余数相等的
两个区间差就是答案。 
*/
int main(){
	while(~scanf("%d",&n)){
		memset(pre,0,sizeof(pre));	memset(mod,0,sizeof(mod));
		bool flag = false;
		per(i,1,n){
			scanf("%d",&a[i]);
			if(flag == true){//已经输出了答案，只要读取数据就可以了 
				continue;
			}
			pre[i] = pre[i-1] + a[i];
			if(pre[i] % n == 0){
				flag = true;
				printf("%d\n",i);
				per(j,1,i){
					printf("%d\n",a[j]);
				}
			}else{
				if(mod[pre[i] % n] != 0){
					flag = true;
					printf("%d\n",i - mod[pre[i] % n]);
					per(j,mod[pre[i]%n]+1,i){
						printf("%d\n",a[j]);
					}
				}else{
					mod[pre[i] % n] = i;
				}
			}
		}
	}
	
	return 0;
}