CF 595 Div3 B2

Books Exchange (hard version)

B1 数据范围很小，直接暴力就可以解决，B2 数据范围很大，就不能暴力了

题意

让求 从 $a[i] $ 到 \(i\) 经过几步

思路

把 \(a[i] \sim i\) 看成一个环，环中的每一个\(a[i]\) 到 \(i\) 需要的步数都是相同的。（有点并查集的思想）

例如

n   5
a[] 5 1 2 4 3

a[1] = 5
a[1] -> a[5] -> a[3] -> a[2] -> a[1]
即 5 3 2 1 5 构成一个环，环内所有数字的步数都相同

代码

#include <bits/stdc++.h>
using namespace std;
int main(){
	int q,n;
	cin >> q;
	while(q--){
		cin >> n;
		vector<int> a(n+1);
		for(int i = 1;i <= n; ++i) cin >> a[i];
		vector<int> used(n+1);
		vector<int> ans(n+1);
		for(int i = 1;i <= n; ++i){
			if(!used[i]){
				vector<int> cur;// 保存该环内所有的数字 
				while(!used[i]){
					cur.push_back(i);
					used[i] = true;
					i = a[i];
				}
				for(auto el : cur) ans[el] = cur.size(); // 环内所有数字的步数 就是 环的大小。
			}
		}
		for(int i = 1;i <= n; ++i) cout << ans[i] <<" ";	
		cout << endl;
	}
	return 0;
}