POJ-1988 Cube Stacking 带权并查集

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

丶聂小白

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

阅读量587

点赞数

CC 4.0 BY-SA版权

分类专栏： ACM 并查集文章标签： acm poj

本文链接：https://blog.youkuaiyun.com/q295657451/article/details/21736713

ACM 同时被 2 个专栏收录

129 篇文章

订阅专栏

并查集

8 篇文章

订阅专栏

本文介绍了一种使用带权值的并查集解决特定问题的方法，包括将一组砖块进行叠放的操作及查询每组下方砖块数量的算法实现。通过C++代码详细展示了如何初始化并查集、查找元素所属集合及其权值、合并两个集合的过程。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

给你编号从1到30000块砖，有2种操作：

1.M a b 表示把a放在b的上面。

还有一种情况是：假如1的下面还有一个2,3的下面还有一个4，那么move1,3的意思就是把1所在的全部立方体放在3全部立方体的上面，而且保持原来1和3所在堆的立方体的顺序。移动后从上到下依次为1,2,3,4.且只能是这一种情况

2.count a 表示询问a下面有几块砖

带权值的并查集

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#define maxn 30030
using namespace std;
long long high[maxn],f[maxn],num[maxn];//f记录上一层，high记录下面有几块砖，num记录以i为根的树子节点个数 (深度） 
void init()//初始化 
{
	for(int i=0;i<maxn;i++)
	{
		f[i]=i;
		high[i]=0;//根节点的高度为0 
		num[i]=1;
	}
}
int find(int x)
{
	if(x==f[x]) return f[x];
	int r=f[x];
	f[x]=find(f[x]);
	high[x]+=high[r];//距根节点的距离 
	return f[x];//x的根节点 
}
void unio(int x,int y)
{
	int fx=find(x);
	int fy=find(y);
	if(fx==fy)  return;
	high[fx]=num[fy];//原根节点的高度等于另一树的高度 
	f[fx]=fy;//fx的根节点变为fy 
	num[fy]+=num[fx];//新树的高度等于原两树深度之和 
	num[fx]=0;//旧根删除高度 
}
int main()
{
	int t,n,m;
	char s[5];
	scanf("%d",&t);
	init();
	int maxm=0;
	while(t--)
	{
		scanf("%s",&s);
		if(s[0]=='M')
		{
			scanf("%d%d",&n,&m);
			unio(n,m);
		}
		else
		{
			scanf("%d",&n);
			find(n);//查询中会计算high[x] 
			printf("%lld\n",high[n]);
		}
	}
	return 0;
}