POJ 1988 Cube Stacking（并查集）

传送门

题目大意

两个人在玩叠箱子游戏，这个游戏有两个操作。
$1.Mxy$代表把$x$所在的一列整体放倒$y$所在一列之上
$2.Cx$代表数一下$x$下方有几个箱子

思路

并查集
$fa[i]$代表i的父亲结点
$num[i]$代表包括$i$结点在内的其子树结点的个数
$dis[i]$表示$i$点到根结点的距离
在合并函数过程中不断进行路径压缩,并不断更新$dis$值，$dis[i]+=dis[fa[i]]$
查找某一个点$x$下边有几个点时，只用求出$x$根结点所在子树的结点个数-$x$到根结点的距离-$1$
即$num[Find(x)]-dis[x]-1$.

代码

int fa[maxn],num[maxn],dis[maxn];

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

void Union(int x,int y){//将x移动到y上 
	int fx=find(x);
	int fy=find(y);
	if(fx!=fy){
		fa[fy]=fx;
		dis[fy]=num[fx];
		num[fx]+=num[fy];
	}
}

int main(){
	int p;
	scanf("%d",&p);
	for(int i=1;i<=30010;i++){
		fa[i]=i;
		dis[i]=0;
		num[i]=1;
	}
	int x,y;
	char op[10];
	//getchar();
	for(int i=1;i<=p;i++){
		scanf("%s%d",op,&x);
		if(op[0]=='M'){
			scanf("%d",&y);
			Union(x,y);
		}
		else{
			printf("%d\n",num[find(x)]-dis[x]-1);
		}
	}
}