837. 连通块中点的数量

最新推荐文章于 2025-05-30 16:33:56 发布

幽殇默

最新推荐文章于 2025-05-30 16:33:56 发布

阅读量113

点赞数

CC 4.0 BY-SA版权

分类专栏： acwing 文章标签：算法

本文链接：https://blog.csdn.net/bettle_king/article/details/116852114

acwing 专栏收录该内容

107 篇文章

订阅专栏

这篇博客介绍了如何使用C++实现并查集数据结构，并处理两种操作：合并集合（C操作）和查询元素是否属于同一集合（Q1操作）。同时，程序还能够查询集合中元素的数量（Q2操作）。通过示例代码展示了如何进行路径压缩以优化查找效率。

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

在这里插入图片描述
https://www.acwing.com/problem/content/description/839/

s[N] 数组的含义是根节点所在集合的点的个数

#include<cstdio>
#include<iostream>
using namespace std;
const int N=1e5+10;
int p[N],s[N];
int find(int x)
{
    if(x!=p[x]) p[x]=find(p[x]);
    return p[x];
}
int main(void)
{
	int n,m; cin>>n>>m;
	for(int i=1;i<=n;i++) p[i]=i,s[i]=1;
	while(m--)
	{
		string op; cin>>op;
		if(op=="C")
		{
			int a,b; cin>>a>>b;
			if(find(a)==find(b)) continue;
			s[find(b)]+=s[find(a)];
			p[find(a)]=find(b);
		}
		if(op=="Q1")
		{
			int a,b; cin>>a>>b;
			if(find(a)==find(b)) cout<<"Yes"<<endl;
			else cout<<"No"<<endl;
		}
		if(op=="Q2")
		{
			int a; cin>>a;
			cout<<s[find(a)]<<endl;
		}
	}
	return 0;
}