#并查集#poj 1988 Cube Stacking

题目大意

两种操作，把编号为x的那一堆移到编号为y的上方，或求编号为x的积木下方有多少块积木。

---

分析

类似于银河星球传说
加权并查集，son表示这一堆的积木数量，dis表示该点到堆顶的距离。
很显然可以求出
answer[x]=son[getf(x)]-dis[x]-1

---

代码

#include <cstdio>
#include <cctype>
#define maxn 30000
using namespace std;
struct ufind{int f,dis,son;}uf[maxn+1];
int getf(int u){
	if (uf[u].f==u) return u;
	int fu=getf(uf[u].f);
	uf[u].dis+=uf[uf[u].f].dis;
	return uf[u].f=fu;//压缩祖先
}
void combi(int x,int y){
	int fa=getf(x),fb=getf(y);
	uf[fb].f=fa; uf[fb].dis=uf[fa].son; //到堆顶的距离更新uf[fa].son+=uf[fb].son;
}
char alp(); int dig();
int main(){
	int m=dig(); for (int i=1;i<=maxn;i++) uf[i].f=i,uf[i].son=1;
	while (m--){
		char c=alp(); int x=dig();
		if (c=='C') printf("%d\n",uf[getf(x)].son-uf[x].dis-1);//输出
		else combi(x,dig());
	}
	return 0;
}
char alp(){
	char c=getchar();
	while (!isalpha(c)) c=getchar();
	return c;
}
int dig(){
	int ans=0; char c=getchar();
	while (!isdigit(c)) c=getchar();
	while (isdigit(c)) ans=ans*10+c-48,c=getchar();
	return ans;
}