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;
}