poj 3370 Halloween treats

题目链接：http://poj.org/problem?id=3370

题意：每组给你一个数c和一个数m，输入m个数，让你输出若干个数的下标，这些数加起来可以整除c（c小于等于n），鸽巢原理同样适用，

需要注意的是，sum可能会溢出int, 并且用scanf读入防止超时。

代码：

#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <ctype.h>
#include <math.h>
#include <iostream>
#include <string>
#include <queue>
#include <stack>
#include <vector>
#include <algorithm>
#define N 100000+100

using namespace std;

int a[N];
long long int sum[N];
int yu[N];

struct node
{
    bool k;
    int pos;

}q[N];


int main()
{
    int n, c;
    int i, j;
    int left, right;

    while(~scanf("%d %d", &c, &n))
    {
        if(c==0 && n==0) break;
        bool flag1=false;
        bool flag2=false;

        for(i=0; i<n; i++){
            scanf("%d", &a[i] );
            if(i==0) sum[i]=a[i];
            else sum[i]=sum[i-1]+a[i];
        }
        left=0;
        memset(q, 0, sizeof(q));

        for(i=0; i<n; i++){
            yu[i]=sum[i]%c;
            if(yu[i]==0){
                flag1=true; right=i; break;
            }
            else{
                if(q[yu[i]].k ){
                    flag2=true;
                    left=q[yu[i]].pos; right=i; break;
                }else{
                    q[yu[i]].k=true; q[yu[i]].pos=i;
                }
            }
        }
        if(flag1){
            //printf("%d
", right+1 );
            for(i=0; i<=right; i++){
                if(i==0) printf("%d", i+1 );
                else printf(" %d", i+1);
            }
            printf("
");
        }
        else if(flag2){
            //printf("%d
", right-left );
            for(i=left+1; i<=right; i++){
                if(i==left+1) printf("%d", i+1);
                else printf(" %d", i+1 );
            }
            printf("
");
        }
    }
    return 0;
}

 这是网上一个人写的代码，比较简短：http://www.cnblogs.com/ACShiryu/archive/2011/08/09/poj3370.html  可供参考学习。