[数据结构与算法]Daimayuan Online Judge #456. 选数（抽屉原理）

题目链接：选数 - 题目 - Daimayuan Online Judge?

题解链接：【算法Camp】【每日一题】Namomo Spring Camp 2022 Div1 第4天题解（前缀和、抽屉原理）_哔哩哔哩_bilibili

看到这道题第一反应是完全背包，看到n最大1e5，想到用bitset优化，但是发现要输出路径，就没法做了。（正常dp可记录路径但是复杂度太高）

这道题的重点是模数是n，而%n后的前缀和就是有n个。那么可以这样考虑：

1. 假如至少有一个前缀和是0，那么结果就有了。
2. 假如所有前缀和都不是0，即全都是1~n-1，那么根据抽屉原理，必有至少一个重复，那么就找到一段连续子序列满足题意，把它输出即可。

#include <bits/stdc++.h>
using namespace std;
#define int long long
#define FOR(i, a, b) for (int i = (a); i <= (b); i++)
const int N = 1e5+5;
int n,sum[N], pre[N];
signed main() {
    ios::sync_with_stdio(false), cin.tie(0), cout.tie(0);
	cin>>n; int x;
	FOR(i,1,n)
		cin>>x, sum[i]=(sum[i-1]+x+n)%n;
	FOR(i,1,n)
		if(sum[i]==0){
			cout<<i<<'\n';
			FOR(j,1,i) cout<<j<<' ';
			return 0;
		}
	FOR(i,1,n){
		if(!pre[sum[i]]) pre[sum[i]]=i;
		else{
			cout<<i-pre[sum[i]]<<'\n';
			FOR(j,pre[sum[i]]+1,i) cout<<j<<' ';
			return 0;
		}
	}
}