思路来源:飘过的小牛
题目大意:
初始时,有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;
}