HDU 3635 并查集+路径压缩+记录每个点移动次数

题意：

给定n个点 oper个操作

每个点有1个龙珠

下面2种操作：

T u v 把u点所有龙珠搬到v

Q u  问u点当前所在城市 u点所在城市有几个龙珠 u点被移动几次

 

思路：

并查集可以求出 u 点所在城市，记录每个点的 son（子节点数）可以求出 某城市的龙珠数量

用step 记录每个点被移动了几次

#include<stdio.h>

#include<string.h>



inline int Max(int a,int b){return a>b?a:b;}

#define N 100001

struct node{

    int step, parent, son;

}a[N];

char s[2];

int find(int x){ 

    if(x==a[x].parent)return x;

    int temp = a[x].parent;

    a[x].parent = find(a[x].parent);

    a[x].step += a[temp].step;

    return a[x].parent;

}

void Union(int x, int y){

    int fx = find(x), fy = find(y);

    if(fx == fy)return;

    a[fx].parent = fy;

    a[fx].step ++;

    a[fy].son += a[fx].son;

    a[fx].son = 0;

}

    int main(){

        int T, Cas = 1, n, oper, i, j, u, v; scanf("%d",&T);

        while(T--){

            scanf("%d %d",&n,&oper);

            printf("Case %d:\n", Cas++);

            for(i = 1; i<= n; i++)

            {

                a[i].son = 1;

                a[i].parent = i;

                a[i].step = 0;

            }

            while(oper--){

                scanf("%s",s);

                if(s[0] == 'T')

                {

                    scanf("%d %d",&u,&v);

                    Union(u, v);

                }

                else

                {

                    scanf("%d",&u);

                    int ans = find(u);

                    printf("%d %d %d\n",ans, a[ ans ].son, a[u].step);

                }

            }

        }

        return 0;

    }