CodeForces - 1214D B2. Books Exchange (hard version)

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

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

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

就可以了。

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

 const int N = (int)1e5+;
 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 = ; i <= n; i++){
             fa[i] = root[i] = i;
             ans[i] = ;
         }

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

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

     return ;
 }