CodeForces - 1249B2 Books Exchange (hard version)（带权并查集找儿子到祖先的距离）

题目链接：https://vjudge.net/contest/338207#problem/C


翻译：
给定一个整数n，接下来n个数，第i个人的书可以给第ai的人，求每本书回到自己手里需要经过几次。
比如n=5时，1 2 3 4 5每本书都在自己手里，所以输出1，
又如n=6时，4 6 2 1 5 3，对于第一个数4，可以给到第4个位置，而第四个位置是1，是第一个数的位置，所以输出2。对于第二个位置的6，到第6个位置，又到第3个位置，第3个位置的数时2，所以输出3。。。。
分析：可以考虑用并查集，计算每个下标i到其祖先的距离。
代码：

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=2*1e5+10;
int f[N],a[N],people[N];
int n;
void init()
{
    for(int i=1; i<=n; i++)
    {
        f[i]=i;
        people[i]=1;
    }
}
int getf(int u)
{
    if(u==f[u])
        return u;
    else
    {
        f[u]=getf(f[u]);
        return f[u];
    }
}
void gether(int u,int v)
{
    int t1=getf(u);
    int t2=getf(v);
    if(t1!=t2)
    {
        f[t1]=t2;
        people[t2]+=people[t1];
    }
}
int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d",&n);
        init();
        for(int i=1; i<=n; i++)
        {
            scanf("%d",&a[i]);
            gether(i,a[i]);
        }
        for(int i=1; i<=n; i++)
            printf("%d ",people[getf(i)]);/*people表示的就是i到其祖先的距离*/
        printf("\n");
    }
    return 0;
}

每个好女孩都会懂得拒绝