Good Bye 2019 : G. Subset with Zero Sum

ちょっと入力が恣意的な気もする問題。
http://codeforces.com/contest/1270/problem/G

問題

N要素の整数列A[i]が与えられる。
なお、(1-indexedで)i-N≦A[i]≦i-1である。

Aの部分集合のうち和が0のものを1つ構築せよ。

解法

条件を入れ替えると1≦i-A[i]≦Nとなる。
そこでN頂点の有効グラフを考え、頂点iから頂点i-A[i]に辺を張ろう。
N点が1つずつ辺を持つので、このグラフは必ず閉路を持ち、頂点番号vに対するA[v]の和を取ると0になる。

int T;
int N;
ll A[1010101];

class SCC {
public:
	static const int MV = 2020000;
	vector<vector<int> > SC; int NV,GR[MV];
private:
	vector<int> E[MV], RE[MV], NUM; int vis[MV];
public:
	void init(int NV) { this->NV=NV; for(int i=0;i<NV;i++) { E[i].clear(); RE[i].clear();}}
	void add_edge(int x,int y) { E[x].push_back(y); RE[y].push_back(x); }
	void dfs(int cu) { vis[cu]=1; for(int i=0;i<E[cu].size();i++) if(!vis[E[cu][i]]) dfs(E[cu][i]); NUM.push_back(cu); }
	void revdfs(int cu, int ind) { int i; vis[cu]=1; GR[cu]=ind; SC[ind].push_back(cu);
		FOR(i,RE[cu].size()) if(!vis[RE[cu][i]]) revdfs(RE[cu][i],ind);}
	void scc() {
		int c=0,i; SC.clear(); SC.resize(NV); NUM.clear();
		assert(NV);
		FOR(i,NV) vis[i]=0; FOR(i,NV) if(!vis[i]) dfs(i); FOR(i,NV) vis[i]=0;
		for(int i=NUM.size()-1;i>=0;i--) if(!vis[NUM[i]]){
			SC[c].clear(); revdfs(NUM[i],c); sort(SC[c].begin(),SC[c].end()); c++;
		}
		SC.resize(c);
	}
};

SCC scc;
void solve() {
	int i,j,k,l,r,x,y; string s;
	
	scanf("%d",&T);
	while(T--) {
		scanf("%d",&N);
		scc.init(N);
		j=-1;
		FOR(i,N) {
			scanf("%d",&A[i]);
			scc.add_edge(i,i-A[i]);
			if(A[i]==0) j=i;
		}
		if(j!=-1) {
			_P("1\n%d\n",j+1);
			continue;
		}
		scc.scc();
		FORR(sc,scc.SC) if(sc.size()>1) {
			_P("%d\n",sc.size());
			FORR(c,sc) _P("%d ",(c+1));
			_P("\n");
			break;
		}
		
	}
}

まとめ

ようやく2019年分が終わったと思ったらすでに半年ビハインドだ…。