G - Subset with Zero Sum

link
题意：要求在数组中选一些数，使得他们的和为0，i从1到n ，i-n<=a[i]<=i-1；
思路：
我们把等式变形一下，可以发现i-a[i]>=1 && i-a[i]<=n，然后我们按照这个建图，从i指向i-a[i]，每个点都会有出度，这不是基环树嘛，那么图中的环可以发现把环上这些点加起来就是0。
复杂度O(n)
代码:

#include<bits/stdc++.h>
using namespace std;
const int N=1000010;
int e[N],ne[N],idx,h[N]; 
void add(int a,int b)
{
	e[idx]=b,ne[idx]=h[a],h[a]=idx++;
}
vector<int>res;
bool flag=0;
int stk[N];
int to[N];
int a[N];
bool vis[N];
int top;
int main()
{
	int t;
	cin>>t;
	while(t--)
	{
	top=0;
	int n;
	scanf("%d",&n);
	for(int i=1;i<=n;i++)	scanf("%d",&a[i]),to[i]=i-a[i] ;
	for(int i=1;i<=n;i++)	vis[i]=0;
	int pos=1;
	while(!vis[pos])
	{
		vis[pos]=1;
		stk[++top]=pos;
		pos=to[pos];
	}
	res.clear();
	do
	{
		res.push_back(stk[top--]);
		if(res.back()==pos)	break;
	}while(1);
	printf("%d\n",(int)res.size());
	for(auto x:res)
	printf("%d ",x);
	puts("");
	}
	return 0;
}