POJ 1988 Cube Stacking

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

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

0 ·

CC 4.0 BY-SA版权

POJ+百练同时被 2 个专栏收录

120 篇文章

订阅专栏

数据结构：并查集

11 篇文章

订阅专栏

本文介绍了一个使用并查集解决木块堆叠问题的具体案例。通过定义数据结构和实现并查集的基本操作，如查找根节点和合并集合，来跟踪木块堆叠的状态。文章中的代码实现了输入命令来动态更新木块堆叠情况，并输出指定木块下方的木块数量。

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

题目地址

total[x]记录最下方木块是x，这一堆总共有多少木块

under[x]记录x木块下面有几块木块

#include<iostream>
#include<cstdio>
#include<queue>
#include<vector>
#include<cctype>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn=30000 + 5;
int Par[maxn],total[maxn],under[maxn];
int getPar(int x){
	if(Par[x]==x) return x;
	int p=getPar(Par[x]);
	under[x]+=under[Par[x]];  //加上到那一堆之前的所有根节点 
	Par[x]=p;
	return p; 
}
void merge(int a,int b){
	int p1=getPar(a);
	int p2=getPar(b);
	if(p1==p2) return ;
	Par[p2]=p1;
	under[p2]=total[p1];
	total[p1]+=total[p2];
}
int main()
{
	int N;
	cin>>N;
	for(int i=1;i<maxn;i++){
		total[i]=1;
		under[i]=0;
		Par[i]=i;
	}
	string cmd;
	while(N--)
	{
		cin>>cmd;
		if(cmd[0]=='M') {
			int x,y; cin>>x>>y;
			merge(y,x);
		}
		else {
			int x; cin>>x;
			getPar(x);
			cout<<under[x]<<endl;
		}
	}
	return 0;
}