CodeForces

题目链接：http://codeforces.com/problemset/problem/1249/B2

思路：用并查集模拟链表，把关系串联起来，如果成环，则满足题意。之后再用并查集合并一个链，一个链代表

一个集合，一个集合有共同的祖先，一个集合答案相同，则输出答案时候只需要输出该元素属于哪一个集合，那个集合有几个元素

就可以了。

#include <stdio.h>
#include <iostream>
using namespace std;

const int N = (int)1e5+100;
int fa[N];
int ans[N];
int root[N];

int search(int x,int& deep){

    ++deep;
    if(fa[x] == x) return root[x] = x;
    else{
        return root[x] = search(fa[x],deep);
    }
}

int main(){

    int q;
    scanf("%d",&q);

    int n,in;
    while(q--){
        scanf("%d",&n);

        for(int i = 1; i <= n; i++){
            fa[i] = root[i] = i;
            ans[i] = 0;
        }

        int deep = 0;
        for(int i = 1; i <= n; i++){
            scanf("%d",&in);
            deep = 0;
            if(search(in,deep) == i){
                ans[i] = deep;
            }
            else fa[i] = in;
        }

        printf("%d",ans[root[1]]);
        for(int i = 2; i <= n; i++) printf(" %d",ans[root[i]]);
        printf("
");
    }

    return 0;
}