poj 2356 Find a multiple (剩余类,抽屉原理)

http://poj.org/problem?id=2356

题意:有n个数,从中选取若干个(1..n),和能被n整除。问是否有解,无解输出0,有解的话,输出个数以及选择的ai

由抽屉原理可知一定有解: 做一个带模的前缀和 sum[i]=(sum[i-1]+a[i])%n n个数,sum[i]最多有n种。 如果某个sum[i]为0,那么表示从1到i的和能被n整除。 如果所有的sum[i]不为0,那么一共有n个sum[i],n-1个值(1..n-1),一定有sum[i]==sumj 那么a[i]到a[j]的和一定能被n整除。

 1/*************************************************************************
 2	> File Name: code/poj/2356.cpp
 3	> Author: 111qqz
 4	> Email: rkz2013@126.com 
 5	> Created Time: 2015年08月21日 星期五 13时43分41秒
 6 ************************************************************************/
 7
 8#include<iostream>
 9#include<iomanip>
10#include<cstdio>
11#include<algorithm>
12#include<cmath>
13#include<cstring>
14#include<string>
15#include<map>
16#include<set>
17#include<queue>
18#include<vector>
19#include<stack>
20#define y0 abc111qqz
21#define y1 hust111qqz
22#define yn hez111qqz
23#define j1 cute111qqz
24#define tm crazy111qqz
25#define lr dying111qqz
26using namespace std;
27#define REP(i, n) for (int i=0;i<int(n);++i)  
28typedef long long LL;
29typedef unsigned long long ULL;
30const int inf = 0x3f3f3f3f;
31const int N=2E4+7;
32int a[N];
33int sum[N];
34int n;
35int p[N];
36int main()
37{
38    scanf("%d",&n);
39    sum[0]= 0;
40    for ( int i = 1  ; i <= n ; i++){
41	scanf("%d",&a[i]);
42	sum[i] = (sum[i-1] + a[i])%n;
43    }
44    memset(p,0,sizeof(p));
45    for ( int i = 1 ; i <= n ; i++){
46	if (sum[i]==0){
47	    printf("%d\n",i);
48	    for ( int j = 1 ; j <= i ; j++){
49		printf("%d\n",j);
50	    }
51	    break;
52	}
53	if (p[sum[i]]){
54	   // cout<<"111qqz"<<endl;
55	    printf("%d\n",i-p[sum[i]]);
56	    for ( int j = p[sum[i]]+1 ; j <= i ; j++){
57		printf("%d\n",j);
58	    }
59	    break;
60	}
61	    p[sum[i]] =  i;
62
63    }
64
65	return 0;
66}