3376: [Usaco2004 Open]Cube Stacking 方块游戏 带权并查集

Description

---

约翰和贝茜在玩一个方块游戏．编号为1到n的n(1≤n≤30000)个方块正放在地上．每个构成一个立方柱．
游戏开始后，约翰会给贝茜发出P(1≤P≤100000)个指令．指令有两种：
1．移动(M)：将包含X的立方柱移动到包含Y的立方柱上．
2．统计©:统计名含X的立方柱中，在X下方的方块数目．
写个程序帮贝茜完成游戏．

Solution

---

一直不会写路径压缩的带权并查集。。

我们记录dis[x]表示x下面有多少个，size[x]表示x所在集合有多少个，find的时候更新dis[]，合并的时候更新最浅的dis和size就可以了

Code

---

#include <stdio.h>
#include <string.h>
#define rep(i,st,ed) for (int i=st;i<=ed;++i)

const int N=60005;

int fa[N],dis[N],size[N];

int read() {
	int x=0,v=1; char ch=getchar();
	for (;ch<'0'||ch>'9';v=(ch=='-')?(-1):(v),ch=getchar());
	for (;ch<='9'&&ch>='0';x=x*10+ch-'0',ch=getchar());
	return x*v;
}

int find(int x) {
	if (!fa[x]) return x;
	int tmp=find(fa[x]);
	dis[x]+=dis[fa[x]];
	return fa[x]=tmp;
}

bool merge(int x,int y) {
	x=find(x),y=find(y);
	if (x==y) return false;
	fa[x]=y;
	dis[x]=size[y];
	size[y]+=size[x];
	return true;
}

int main(void) {
	rep(i,1,N-1) size[i]=1;
	for (int T=read(),opt,x,y;T--;) {
		for (opt=getchar();opt!='M'&&opt!='C';opt=getchar());
		x=read();
		if (opt=='M') {
			y=read();
			merge(x,y);
		} else {
			find(x);
			printf("%d\n", dis[x]);
		}
	}
	return 0;
}