【组合数学】 Find a multiple&&N的倍数 详解（鸽巢原理）

两道几乎一模一样的题

题目链接：N的倍数  Find a multiple

题目大意：

给定N和N个数，让你找出任意一个组合，使这个组合的数之和为N的倍数。

鸽巢原理：

又称抽屉原理：

桌上有十个苹果，要把这十个苹果放到九个抽屉里，无论怎样放，我们会发现至少会有一个抽屉里面放不少于两个苹果。这一现象就是我们所说的“抽屉原理”。 

应用： 

对于给定的N个数，将其存放于A数组中，并求其前缀和，存放于sum数组中。也就是说sum[i]中存放的是A[1]~A[i]之和。

然后让sum数组的每一个元素对N取模。

则sum数组每一个元素的取值都为0~N-1。

①如果存在sum[i]==0，那么也就是说取模前sum[i]%N==0，即前i个数之和为N的倍数。（数据保证给出的N个数不包含0）

②如果sum数组中不包含0，那么因为sum数组中的元素取值为0~N-1，现在保证不含0，则sum数组中的元素取值为1~N-1，而sum数组共有N个元素，则sum数组中必然存在两个元素值相等。相当于N-1个抽屉要放N个苹果，必然有一个抽屉放了两个苹果，这就是鸽巢原理在这题的应用了。

因为如果sum[j]-sum[i]==0，则代表取模前(sum[j]-sum[i])%N==0，即从i+1到j这段连续和为N的倍数。

综上所述，答案必然存在，且至少有一种方案是给出的N个数的一段连续和

 

代码：

两题的代码一模一样。。。

#include<iostream>
#include<cstdio>
#include<string.h>
using namespace std;

int A[50005];
int sum[50005];
int vis[50005];

int main()
{
    int N;
    memset(sum,0,sizeof(sum));
    memset(vis,0,sizeof(vis));

    scanf("%d",&N);
    for(int i=1;i<=N;i++){
        scanf("%d",&A[i]);
        sum[i]=(sum[i-1]%N)+(A[i]%N);
        sum[i]%=N;
    }

    for(int i=1;i<=N;i++)
        if(sum[i]==0){
           printf("%d\n",i);
           for(int j=1;j<=i;j++)
             printf("%d\n",A[j]);
           return 0;
        }

    for(int i=1;i<=N;i++){
        if(vis[sum[i]]!=0){ //vis[sum[i]]表示上一个和为sum[i]的元素的下标
            printf("%d\n",i-vis[sum[i]]);
            for(int j=vis[sum[i]]+1;j<=i;j++)
                printf("%d\n",A[j]);
            break;
        }
        vis[sum[i]]=i;   //记录i
    }
    return 0;

}