poj 2356 and 3370 鸽巢原理

题意 ： 给你n个整数，取其中的任意个整数满足相加的和是n的倍数


处理出前k个数的sum[k] (1 <= k <= n) 并且对n取余 ， 如果有一个sum[k]等于0，自然这个sum就是n的倍数。

根据鸽巢原理，有n个余数 ， 余数0 ~ n ，如果没有余数0，至少有两个余数相同，那么这两个sum相减得到的就是n的倍数

POJ2365 code

#include <stdio.h>
int a[11111] , sum[11111], vis[11111];
int main()
{
	int n,i,j;
	while(scanf("%d", &n)!=-1)
	{
		for(i = 1;i <= n; i++)
		{
			scanf("%d", &a[i]);
			sum[i] = sum[i-1]+a[i];
			sum[i] %= n;
			vis[i] = 0;
		}
		for(i = 1;i <= n; i++)
		{
			if(sum[i]==0)
			{
				printf("%d\n", i);
				for(j = 1;j <= i; j++)
					printf("%d\n", a[j]);
				break;
			}
			else if(vis[sum[i]])
			{
				printf("%d\n", i-vis[sum[i]]);
				for(j = vis[sum[i]]+1;j <= i ;j++)
					printf("%d\n", a[j]);
				break;
			}
			vis[sum[i]] = i;
		}
	}
	return 0;
}

POJ 3370 code

#include <stdio.h>
int a[111111] , sum[111111], vis[111111];
int main()
{
	int n,i,j,c;
	while(scanf("%d%d", &c , &n)!=-1 && c)
	{
		for(i = 1;i <= n; i++)
		{
			scanf("%d", &a[i]);
			sum[i] = sum[i-1]+a[i];
			sum[i] %= c;
			vis[i] = 0;
		}
		for(i = 1;i <= n; i++)
		{
			if(sum[i]==0)
			{
				for(j = 1;j < i; j++)
					printf("%d ", j);
				printf("%d\n", j);
				break;
			}
			else if(vis[sum[i]])
			{
				for(j = vis[sum[i]]+1;j < i ;j++)
					printf("%d ", j);
				printf("%d\n", j);
				break;
			}
			vis[sum[i]] = i;
		}
	}
	return 0;
}