边带权并查集 - Dragon Balls（HDU3635）

版权声明：转载请注明出处 https://blog.csdn.net/weixin_42557561/article/details/83788210

传送门

题意

有n个点，一开始每个点构成一个集合
之后有两个操作：
操作1 M A B：将和A属于一个集合中的所有点都转移到B所在的集合
操作2 ＱＡ：询问Ａ所在的集合，以及该集合中的点数，以及Ａ这个点被转移的次数

Analysis

询问操作中也就只有点被转移的次数需要动一下脑子（并查集好喵喵妙啊）
每一个点被转移的时候就在它所在集合的顶端更新转移次数
最后查询的时候就直接将这个点到其祖先的路径上所有的转移次数相加就好了

一开始以为这样会算多，就不敢写，看了一下题解，发现确实是这样做
仔细思考了一下，原来不会出错啊（原因？？口胡不清，，大家模拟一下就好咯。重点是你每次更新的时候都是和另一个点的 $fa$合并）

Code

#include<bits/stdc++.h>
#define in read()
#define N 10009
using namespace std;
inline int read(){
	char ch;int f=1,res=0;
	while((ch=getchar())<'0'||ch>'9') if(ch=='-') f=-1;
	while(ch>='0'&&ch<='9') {
		res=(res<<3)+(res<<1)+ch-'0';
		ch=getchar();
	}
	return f==1?res:-res;
}
int T,fa[N],sze[N],trans[N],n,q;
char st[5];
inline int getfa(int x){
	if(x==fa[x]) return x;
	int root=getfa(fa[x]);
	trans[x]+=trans[fa[x]];
	return fa[x]=root;
}
int main(){
	T=in;
	for(int t=1;t<=T;++t){
		printf("Case %d:\n",t);
		n=in;q=in;int x,y;
		for(int i=1;i<=n;++i) fa[i]=i,sze[i]=1,trans[i]=0;
		for(int i=1;i<=q;++i){
			scanf("%s",st);
			if(st[0]=='T'){
				x=in;y=in;
				int fx=getfa(x),fy=getfa(y);
				fa[fx]=fy;
				sze[fy]+=sze[fx];
				trans[fx]++;
			}
			else{
				x=in;
				int fx=getfa(x);
				printf("%d %d %d\n",fx,sze[fx],trans[x]);
			}
		}
	}
	return 0;
}