连通块中点的数量

最新推荐文章于 2022-05-17 14:22:08 发布

撒浪嘿呦x

最新推荐文章于 2022-05-17 14:22:08 发布

阅读量391

点赞数

分类专栏：算法 # 并查集文章标签：算法 c++ 并查集

本文链接：https://blog.csdn.net/qq_45806470/article/details/122847178

版权

算法同时被 2 个专栏收录

101 篇文章 3 订阅

订阅专栏

并查集

4 篇文章 0 订阅

订阅专栏

这篇博客介绍了一种使用并查集数据结构来处理无向图中连通块的问题。通过建立并查集，实现了连边、判断两点是否在同一连通块以及查询连通块点数的功能。代码展示了如何实现并查集的操作，并给出了具体的输入输出示例。

摘要由CSDN通过智能技术生成

连通块中点的数量

给定一个包含 n 个点（编号为 1∼n）的无向图，初始时图中没有边。

现在要进行 m 个操作，操作共有三种：

C a b，在点 a 和点 b 之间连一条边，a 和 b 可能相等；
Q1 a b，询问点 a 和点 b 是否在同一个连通块中，a 和 b 可能相等；
Q2 a，询问点 a 所在连通块中点的数量；

输入格式

第一行输入整数 n 和 m。

接下来 m 行，每行包含一个操作指令，指令为 C a b，Q1 a b 或 Q2 a 中的一种。

输出格式
对于每个询问指令 Q1 a b，如果 a 和 b 在同一个连通块中，则输出 Yes，否则输出 No。

对于每个询问指令 Q2 a，输出一个整数表示点 a 所在连通块中点的数量

每个结果占一行。

数据范围
1≤n,m≤10⁵
输入样例：

输出样例：

Yes
2
3

分析：
并查集题目，设置一个记录根节点下有多少节点数的数组，在merge操作时对相应的根节点进行操作。

代码：

#include<bits/stdc++.h>
using namespace std;
const int N = 100010;
int n,m,fa[N],cnt[N]; //cnt[i]记录第i个根节点连接节点数（连通块中点的数量）
int get(int x)
{
    if(fa[x]==x) return x;
    return fa[x]=get(fa[x]);
}
void merge(int a,int b)
{
    int x=get(a),y=get(b);
    fa[y]=x;
    cnt[x]+=cnt[y];
}
int main()
{
    cin>>n>>m;
    for(int i=1;i<=n;i++)
    {
        fa[i]=i;
        cnt[i]=1; //初始化每个根都只有一个节点。
    }
    while(m--)
    {
        string q;
        int a,b;
        cin>>q;
        if(q=="C")
        {
            cin>>a>>b;
            if(get(a)!=get(b))
            {
                merge(a,b);
            }
        }
        else if(q=="Q1")
        {
            cin>>a>>b;
            if(get(a)==get(b)) cout<<"Yes"<<endl;
            else cout<<"No"<<endl;
        }
        else
        {
            cin>>a;
            cout<<cnt[get(a)]<<endl;
        }
    }
    return 0;
}