知识点:前缀和,抽屉原理
题目大意:
给定n个数,要求选出一个不为空的子序列,令子序列之和取模于n为0,输出你选定子序列的大小和序列中每个数在原数组中的下标
思路:
1.给定n个数,那么做前缀和的话,前缀和共有n+1个数(包括sum[0]=0)
2.对于所有的前缀和都取模于n,最多能得到n个不同的数。
接下来证明答案必然存在
1.特判,我们取模过后的前缀和中有等于0的情况,那么,必然是这一段区间就是答案。
2.原本有n+1个数的前缀和经过变换,变为n个数,相当于用n个抽屉装n+1个物品,必然有至少两个物品在同一抽屉内,在这道题上就是必然有两个取模于n后的前缀和是同一个数。
我们不妨想想2所说的是什么意思
假设sum[3]=sum[2]
(a[3]+a[2]+a[1])%m=(a[1]+a[2])%m
根据取模运算的性质可知,a[3]%m必然等于0,则a[3]就是我们要找的这一段区间。
#include <bits/stdc++.h>
#define endl '\n'
using namespace std;
long long sum[10000000+10];
int main()
{
int n;
cin>>n;
for(register int i=1,temp;i<=n;i++)
{
cin>>temp;
sum[i]=(sum[i-1]+temp)%n;
}
map<int ,int>mp;
mp[0]=0;
for(register int i=1;i<=n;i++)
{
if(mp.count(sum[i]))
{
cout<<i-mp[sum[i]]<<endl;
for(int j=mp[sum[i]]+1;j<=i;j++)
cout<<j<<" ";
cout<<endl;
return 0;
}
mp[sum[i]]=i;
}
return 0;
}