Codeforces #794 : Div1 D1. Permutation Weight (Easy Version)

Easyは割とシンプル。
https://codeforces.com/contest/1685/problem/D1

問題

1～NのPermutation Pが与えられる。
以下を満たすPermutation Qを構築せよ。

• が最小。

解法

i→P[i]に辺を張った無向グラフを考える。
ここに閉路がK個あるとき、解は2(K-1)となる。
すなわち、各サイクル、サイクル中に現れる頂点毎にQを埋めて行こう。
サイクル内の要素間については、絶対値の部分が0になり、サイクルの境界部分では絶対値の部分が2となる。

int T,N,P[202];
int Q[202];

template<int um> class UF {
	public:
	vector<int> par,rank,cnt;
	UF() {par=rank=vector<int>(um,0); cnt=vector<int>(um,1); for(int i=0;i<um;i++) par[i]=i;}
	void reinit(int num=um) {int i; FOR(i,num) rank[i]=0,cnt[i]=1,par[i]=i;}
	int operator[](int x) {return (par[x]==x)?(x):(par[x] = operator[](par[x]));}
	int count(int x) { return cnt[operator[](x)];}
	int operator()(int x,int y) {
		if((x=operator[](x))==(y=operator[](y))) return x;
		cnt[y]=cnt[x]=cnt[x]+cnt[y];
		if(rank[x]>rank[y]) return par[x]=y;
		rank[x]+=rank[x]==rank[y]; return par[y]=x;
	}
};

void solve() {
	int i,j,k,l,r,x,y; string s;
	
	cin>>T;
	while(T--) {
		UF<202> uf;
		cin>>N;
		for(i=1;i<=N;i++) {
			cin>>P[i];
			uf(i,P[i]);
		}
		x=0;
		for(i=1;i<N;i++) {
			if(uf[i]!=uf[i+1]) {
				uf(i,i+1);
				FOR(x,N) if(P[x+1]==i) break;
				FOR(y,N) if(P[y+1]==i+1) break;
				swap(P[x+1],P[y+1]);
				
			}
		}
		
		Q[1]=1;
		cout<<1;
		for(i=2;i<=N;i++) {
			for(j=1;j<=N;j++) if(P[j]==Q[i-1]) Q[i]=j;
			cout<<" "<<Q[i];
		}
		cout<<endl;
	}
}

まとめ

こういう添え字がごちゃごちゃした問題、状態遷移が割と混乱する。