题目传送门
题意:
给你N个数,要你从这N个数中选出几个数,他们的和恰好是N的整数倍。输出所选数的个数,和数。
思路:
这道题要用到容斥原理。首先,N个数有N个前缀和,他们%N的余数,要么全部不相同,即0->N-1,这种情况,余数为0的前缀和就是所求的区间。
另外一种情况就是,至少有2个前缀和%N的余数是相同的,那么他们之间的差值就是N的倍数。他们之间的区间就是所求区间。
教训:
思考问题时,不能被样例的答案所局限,这是设题者的坑。还有不要总是用模拟的思路去思考问题,要将问题转化与化归,从变中寻找不变。
AC code:
#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
map<int,int> mp;
const int maxn = 5e4 + 10;
int sum[maxn];
int a[maxn];
int n = 0;
int main(){
while(~scanf("%d",&n)){
fill(sum,sum+maxn,0);
mp[0] = 0;
for(int i = 1;i <= n;++i){
scanf("%d",&a[i]);
sum[i] = (sum[i-1] + a[i]) % n;
if(mp.count(sum[i]) == 1){
printf("%d\n",i - mp[sum[i]]);
for(int j = mp[sum[i]] + 1;j <= i;++j){
printf("%d\n",a[j]);
}
return 0;
}
mp[sum[i]] = i;
}
}
return 0;
}