当前位置: 代码迷 >> 综合 >> POJ 3370 Halloween treats(抽屉原理,前缀和)
  详细解决方案

POJ 3370 Halloween treats(抽屉原理,前缀和)

热度:77   发布时间:2023-12-13 19:21:37.0

抽屉原理
题目意思:
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 */