hdu 3635 Dragon Balls(带权并查集)

思路来源：飘过的小牛

题目大意：

初始时，有n个龙珠，编号从1到n，分别对应的放在编号从1到n的城市中。

现在又2种操作：

T A B，表示把A球所在城市全部的龙珠全部转移到B城市。（第一次时，因为A球所在的城市只有一个球，所以只移动1个，如果有多个，则全部移动）。

Q A，表示查询A。要求得到的信息分别是：A现在所在的城市，A所在城市的龙珠数目，A转移到该城市移动的次数（如果没有移动就输出0）

1.龙珠i在哪个城市 解决：find(i)函数就可以找到i龙珠最后在哪个城市

2.城市find(i)中有多少个龙珠 解决：直接输出num[find(i)]即可

3.龙珠i转移了多少次 解决：？？

解决方法：

通过路径压缩来更新转移的次数，比如每次移动时，我只需要把这个城市的根结点的转移次数+1，等到以后路径压缩时，子结点自己移动的次数加上根结点移动的次数，就是这个结点总共的移动次数，不明白的可以自己动手画画。。

这里，根结点的转移次数不是直接加上根结点的移动次数，而是在路径压缩的过程中逐层累加，比如有4层，根结点为1，然后第二层2，第三层3，第四层4.那么，第一层路径压缩过程中进入第二层，然后第三层，然后第四层，第四层也就是根，找到根结点，返回第三层，第三层+上第四层的转移次数，然后返回第二层，第二层转移次数加上第三层转移次数，返回第一层，第一层转移次数加上第二层转移次数，同时，在这个过程中也把下面3层的结点的父节点直接更新为和根结点，也就是1相连，查找结束后，称为2层结构，根结点1在一层，234都处于第二层。这样的一个过程是通过递归调用本身完成的，跟数据结构（严蔚敏版本）讲解栈那节的汉诺塔过程相似，如果想完全掌握，推荐仔细看教材+自己模拟，这样就可以理解了。。。。。

#include<stdio.h>
#include<iostream>
#include<string>
#include<algorithm>
using namespace std;
#define MAXN 10005
int pre[MAXN];//pre[i]表示i球在哪个城市,其实是并查集中的集合
int num[MAXN];//num[i]表示i号城市有几个球 
int trans[MAXN];//trans[i]表示i球转移了多少次
int n,m;
/*带权并查集*/
void init(){
    for(int i=0;i<MAXN;i++){
        pre[i]=i;//初始i号球在i城市 
        num[i]=1;//初始i城市只有一个球 
        trans[i]=0;//初始i号球没有经过任何转移 
    }
}
int find(int x){
    if(x==pre[x])
    return x;
    int px = find(pre[x]);
    trans[x]+=trans[pre[x]];
    return pre[x]=px;
}
void merge(int x,int y){
    int xx=find(x);
    int yy=find(y);
    if(xx!=yy){
        pre[xx]=yy;
        trans[xx]++;//第一次移动
        num[yy]+=num[xx];
        num[xx]=0;//移动完毕后置0
    	//这里的num其实是并查集的树的高度。 
    }
}
int main(){
    int t;
    int KASE=1;
    scanf("%d",&t);
    while(t--){
    	printf("Case %d:\n",KASE++);
        scanf("%d%d",&n,&m);
        init();
        for(int i=0;i<m;i++){
        	getchar();
        	char c;
        	int a,b;
        	scanf("%c",&c);
        	if(c=='T'){
        		scanf("%d%d",&a,&b);
        		merge(a,b);
			}else if(c=='Q'){
				scanf("%d",&a);
				int x = find(a);			
				printf("%d %d %d\n",x,num[x],trans[a]);
			}
		}    
    }   
    return 0;
}