一个长度为N的数组A,从A中选出若干个数,使得这些数的和是N的倍数。
例如:N = 8,数组A包括:2 5 6 3 18 7 11 19,可以选2 6,因为2 + 6 = 8,是8的倍数。
Input
第1行:1个数N,N为数组的长度,同时也是要求的倍数。(2 <= N <= 50000) 第2 - N + 1行:数组A的元素。(0 < A[i] <= 10^9)
Output
如果没有符合条件的组合,输出No Solution。 第1行:1个数S表示你所选择的数的数量。 第2 - S + 1行:每行1个数,对应你所选择的数。
Input示例
8 2 5 6 3 18 7 11 19
Output示例
2 2 6
把前缀和(mod n)求出来。
1.如果这些和中有一个0,那么我们便得到所求。
2.否则,这些和中必有两个是相等的(抽屉原理),他们相减为0,这便又找到我们想要的。
利用抽屉原理,因为N的倍数,A[i]中的任意一个数modN 都是在[1, N-1] 然后把N个数扔到这些个N-1个格子里,必定会有两个一样,这两个做减法就是一样的了。
#include<bits/stdc++.h>
using namespace std;
const int maxn=50000+100;
struct node
{
int num;
int index;
} pre[maxn];
int a[maxn];
bool cmp(node x,node y)
{
if(x.num<y.num) return true;
else if(x.num==y.num && x.index<y.index) return true;
return false;
}
int main()
{
int i,j,n,ok,mod;
cin>>n;mod=n;
pre[0].num=0;
for(i=1;i<=n;i++) {
cin>>a[i];
pre[i].num=(pre[i-1].num+a[i])%mod;
pre[i].index=i;
}
sort(pre+1,pre+1+n,cmp);
for(i=1;i<=n;i++) {
if(pre[i].num==0) {
cout<<pre[i].index<<endl;
for(j=1;j<=pre[i].index;j++) cout<<a[j]<<endl;
return 0;
}
}
for(i=2;i<=n;i++) {
if(pre[i].num==pre[i-1].num) {
cout<<pre[i].index-pre[i-1].index<<endl;
for(j=pre[i-1].index+1;j<=pre[i].index;j++) cout<<a[j]<<endl;
return 0;
}
}
return 0;
}