#并查集#poj 1988 Cube Stacking

最新推荐文章于 2022-03-03 13:43:07 发布

原创最新推荐文章于 2022-03-03 13:43:07 发布 · 199 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#poj 1988 #Cube Stacking

并查集专栏收录该内容

47 篇文章

订阅专栏

本文介绍了一种使用加权并查集来处理积木堆操作的方法，包括将编号为x的积木堆移动到编号为y的位置上方，以及查询编号为x的积木下方的积木数量。通过维护每个节点到根节点的距离和堆中积木的数量，实现了高效的查询和更新。

题目大意

两种操作，把编号为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;
}