并查集之合并集合

最新推荐文章于 2023-01-10 09:00:00 发布

qq154532295

最新推荐文章于 2023-01-10 09:00:00 发布

阅读量76

点赞数

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

本文链接：https://blog.csdn.net/qq154532295/article/details/124027514

版权

算法同时被 2 个专栏收录

11 篇文章 0 订阅

订阅专栏

并查集

1 篇文章 0 订阅

订阅专栏

该博客主要介绍了如何使用并查集解决集合合并与查询问题。通过ACwing算法基础课中的示例，展示了如何实现合并两个集合以及判断两个元素是否属于同一集合的函数，并提供了对应的C++代码实现。并查集是一种用于维护集合动态合并操作的数据结构，在解决连通性问题时非常有效。

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

合并集合

题目来源Acwing算法基础课

题目

一共有 nn 个数，编号是 1∼n1∼n，最开始每个数各自在一个集合中。

现在要进行 mm 个操作，操作共有两种：

M a b，将编号为 aa 和 bb 的两个数所在的集合合并，如果两个数已经在同一个集合中，则忽略这个操作；
Q a b，询问编号为 aa 和 bb 的两个数是否在同一个集合中；

输入格式

第一行输入整数 nn 和 mm。

接下来 mm 行，每行包含一个操作指令，指令为 M a b 或 Q a b 中的一种。

输出格式

对于每个询问指令 Q a b，都要输出一个结果，如果 aa 和 bb 在同一集合内，则输出 Yes，否则输出 No。

每个结果占一行。

数据范围

1≤n,m≤1051≤n,m≤105

输入样例：
4 5
M 1 2
M 3 4
Q 1 2
Q 1 3
Q 3 4
输出样例：
Yes
No
Yes

#include <iostream>
#include <cstring>
#include <algorithm>

using namespace std;
const int N = 100010;
int p[N];
int find(int x){  //并查集找x
    if(x != p[x]) p[x] = find(p[x]);
    return p[x];
}
int main()
{
    for(int i = 0; i < N; i++)p[i] = i; //每个节点的祖先是它自己
    int n, m;
    cin >> n >> m;
    while(m--){
        string op;
        int a, b;
        cin >> op >> a >> b;
        if(op == "M"){
            if(p[a] != p[b])
                p[find(a)] = find(b);  //把a集合里面的数集中到b集合里面去;
        }
        else{
            if(find(a) == find(b))cout << "Yes" << endl;
            else cout << "No" << endl;
        }
        
    }
    return 0;
}