POJ 3370 Halloween treats（抽屉原理，前缀和）

抽屉原理
题目意思：
n个数任取若干个， 这些数的和是c的倍数， 输出其中的任意一种取法，输出元素的下标即可

本题要点：
1、抽屉原理
有n+1件或n+1件以上的物品要放到n个抽屉中，那么至少有一个抽屉里有两个或两个以上物品。
2、一个结论
a1,a2,a3…am是正整数序列，至少存在整数k和r,1<=k<r<=m,使得ak+a(k+1)+…+a®是m的倍数。
3、具体步骤
用sum来记录数组a的前缀和，
a） 如果某个 sum[i] % c == 0, 说明 sum[i] = a[1] + a[2] + … + a[i]
是 c的倍数，输出下标 1~i 即可。
c） 如果 sum[i] 和 sum[j] （假设 i < j）模 c 同余，说明 sum[j] - sum[i] 是c 的倍数，输出下标
i + 1 ~ j 即可。

#include <cstdio>
#include <cstring>
#include <iostream>
using namespace std;
const int MaxN = 100010;
int a[MaxN], sum[MaxN];
int r[MaxN];	// r[i] 表示余数是i， 对应的前缀和的下标
int n, c;

int main()
{
	while(scanf("%d%d", &c, &n) != EOF)
	{
		if(0 == c && 0 == n)	break;
		sum[0] = 0;
		for(int i = 0; i <= c; ++i)
			r[i] = 0;
		for(int i = 1; i <= n; ++i)
		{
			scanf("%d", &a[i]);
		}
		for(int i = 1; i <= n; ++i)
		{
			sum[i] = (sum[i - 1] + a[i]) % c;
			if(sum[i] == 0)
			{
				for(int j = 1; j < i; j++)
					printf("%d ", j);
				printf("%d\n", i);
				break;
			}
			if(r[sum[i]])
			{
				for(int j = r[sum[i]] + 1; j < i; j++)
					printf("%d ", j);
				printf("%d\n", i);
				break;
			}
			r[sum[i]] = i;
		}
	}
	return 0;
}

/*
4 5
1 2 3 7 5
3 6
7 11 2 5 13 17
0 0
*/

/*
3 5
2 3 4
*/