并查集模板

最新推荐文章于 2022-02-17 15:39:22 发布

早睡早起的猫

最新推荐文章于 2022-02-17 15:39:22 发布

阅读量96

点赞数 1

分类专栏： # 基本数据结构

本文链接：https://blog.youkuaiyun.com/weixin_51310066/article/details/119729050

版权

基本数据结构专栏收录该内容

9 篇文章

订阅专栏

这篇博客介绍了并查集这种数据结构的基础操作，包括如何合并集合和判断元素是否属于同一集合。通过示例展示了在处理一系列合并和查询操作中的应用，数据范围限定在1到10^5之间。

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

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

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

1.M a b，将编号为 a 和 b 的两个数所在的集合合并，如果两个数已经在同一个集合中，则忽略这个操作；
2.Q a b，询问编号为 a 和 b 的两个数是否在同一个集合中；
输入格式
第一行输入整数 n 和 m。

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

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

每个结果占一行。

数据范围
1≤n,m≤105
输入样例：
4 5
M 1 2
M 3 4
Q 1 2
Q 1 3
Q 3 4
输出样例：
Yes
No
Yes

#include<bits/stdc++.h>
using namespace std;
const int N=1e5+10;
int m,n,arr[N];
char ch;
int find(int x)//找到下标为x的节点的根节点
{
    if(arr[x]!=x)
        arr[x]=find(arr[x]);//找arr[x]的父节点，找爸爸
    else
        return arr[x];//返回根节点的取值（可用此时下标和取值相等，进行合并）
}
int main()
{
    cin>>m>>n;
    for(int i=1;i<=m;i++)
        arr[i]=i;
    while(n--)
    {
        int a,b;
        cin>>ch>>a>>b;
        if(ch=='M')
        {
            arr[find(a)]=find(b);
        }
        if(ch=='Q')
        {
            if(find(a)==find(b))
                cout<<"Yes"<<endl;
            else
                cout<<"No"<<endl;
        }
    }
}