原题链接:选数 - 题目 - Daimayuan Online Judge
解题思路:首先先开一个数组记录原数据的前缀和,并且余n。如果余后有0(后n项范围内),那么说明这个前缀和本身就可以被整除,符合题目要求,直接输出即可。
假如没有0,也必然有解。因为任何一个数余n都只可能产生0到n-1这n种情况。而没有0,说明只产生了n-1种情况。但是题目中有n个数字,说明必然有余后结果相同的值。总所周知,a%n=(a+k*n)%n,k为任意整数。
假设第i位的前缀和与第j位的前缀和同余n后相等,那么说明i+1位累加到j这一段必然余n为0,那么这一段就是答案。
AC代码:
#include<bits/stdc++.h>
using namespace std;
long long n,a[100010],b[100010],c[100010];//n与a数组为题目本意,b数组记录a数组前缀和余n后的值,c数组记录b数组中每个值第一次出现的位置
int main(){
cin>>n;
for(int i=1;i<=n;i++){
scanf("%lld",&a[i]);
b[i]=(b[i-1]+a[i])%n;//前缀和的同时余n
}
for(int i=1;i<=n;i++){
if(b[i]==0){//如果有余后为0,说明从第1位累加到这一位的前缀和本身就是答案
cout<<i<<endl;
for(int j=1;j<=i;j++) cout<<j<<' ';
return 0;
}
if(c[b[i]]){//如果b[i]这个值第二次出现,说明第一次出现的位置后一位,累加到第二次出现的位置这一段和为答案
cout<<i-c[b[i]]<<endl;
for(int j=c[b[i]]+1;j<=i;j++) cout<<j<<' ';
return 0;
}
c[b[i]]=i;//如果第一次出现则不会触发上面的if,记录第一次出现的位置
}
return 0;
}