POJ 3370 && HDU 1808 Halloween treats（鸽巢原理）

Description
万圣节到了，有m个小孩要去n个地方要糖吃，现在给出n个地方的糖的数量，问是否可以从中选出某些地方，使得糖的数量可以整除m，如果可以则输出其中一组解，不可以的话输出-1
Input
多组输入，每组用例第一行为两个整数m和n(1 <= m <= n <= 10^5)，第二行为n个数分别表示这n个地方的糖数，以0 0结束输入
Output
对于每组用例，如果存在可行方案则输出其中一组，否则输出-1
Sample Input
4 5
1 2 3 7 5
3 6
7 11 2 5 13 17
0 0
Sample Output
3 5
2 3 4
Solution
首先得到这n个数的前缀和sum，其中sum[0]=0，这样我们就得到了n+1个数，将其模m后，因为m <= n，那么由鸽巢原理得这n+1个数中最少存在两个数相同，令其为sum[i],sum[j]，那么sum[j]-sum[i]必为m的倍数，即从原数列第i+1项到第j项之间的j-i个数的和必然为m的倍数
Code

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
#define maxn 111111
int sum[maxn];
int a[maxn];
int mark[maxn];
int main()
{
    int m,n;
    while(scanf("%d%d",&m,&n),m||n)
    {
        //初始化 
        memset(sum,0,sizeof(sum));
        memset(mark,-1,sizeof(mark));
        mark[0]=0;
        for(int i=1;i<=n;i++)//求前缀和 
        {
            scanf("%d",&a[i]);
            sum[i]=(sum[i-1]+a[i])%m;
        }
        for(int i=1;i<=n;i++)
        {
            if(mark[sum[i]]!=-1)//该前缀和以及出现过 
            {
                for(int j=mark[sum[i]]+1;j<=i;j++)
                    printf("%d%c",j,j==i?'\n':' ');
                break;
            }
            else//没有出现过则记录其出现位置 
                mark[sum[i]]=i;
        }
    }
    return 0;
}