带权并查集 [USACO04OPEN]Cube Stacking G（洛谷 P2342）

[USACO04OPEN]Cube Stacking G

题目描述

约翰和贝西在叠积木。共有30000块积木，编号为1到30000。一开始，这些积木放在地上，自然地分成N堆。贝西接受约翰的指示，把一些积木叠在另一些积木的上面。一旦两块积木相叠， 彼此就再也不会分开了，所以最后叠在一起的积木会越来越高。约翰让贝西依次执行P条操作，操作分为两种：

· 第一种是移动操作，格式为“移动X到Y的上面”。X和Y代表两块积木的编号，意思是将X所的那堆积木，整体叠放到Y所在的那堆积木之上；

· 第二种是统计操作，格式为“统计Z下方的积木数量”。Z代表一块积木的编号，意思是贝西需要报告在编号为Z的积木之下还有多少块积木

请编写一个程序，帮助贝西回答每条统计问题。

---

带权并查集基础题，size[i] 表示以 i 为最低层的连通块大小，ans[i] 表示第 i 块积木下的积木个数，因为是以最低层的积木作为根，所以合并过程中的ans[]合并也特别关键；

代码：

#include<bits/stdc++.h>
#define LL long long
#define pa pair<int,LL>
#define ls k<<1
#define rs k<<1|1
#define inf 0x3f3f3f3f
using namespace std;
const int N=100100;
const int M=50100;
const int mod=2015;
int size[N],ans[N],fa[N];
int find(int p){
	if(p==fa[p]) return p;
	int ff=find(fa[p]);
	ans[p]+=ans[fa[p]];
	return fa[p]=ff;
}
int main(){
	for(int i=1;i<N;i++) fa[i]=i,size[i]=1;
	int p;cin>>p;
	while(p--){
		char s[10];scanf("%s",s);
		if(s[0]=='M'){
			int x,y;scanf("%d%d",&x,&y);
			x=find(x),y=find(y);
			fa[x]=y;
			ans[x]=size[y];
			size[y]+=size[x];
		}
		else{
			int z;scanf("%d",&z);
			find(z);
			printf("%d\n",ans[z]);
		}
	}
	return 0;
}