并查集代码

最新推荐文章于 2024-07-08 21:23:20 发布

玖忱i

最新推荐文章于 2024-07-08 21:23:20 发布

阅读量345

点赞数 11

文章标签： c++ 数据结构

本文链接：https://blog.csdn.net/weixin_51406370/article/details/136162850

版权

文章介绍了如何使用树结构表示集合，并通过C++代码展示了如何合并集合、查找元素所属集合以及判断两个元素是否在同一个集合中的方法，包括压缩路径优化.

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

https://www.acwing.com/problem/content/838/

作用：
- 将两个集合合并
- 查询两个元素是否在一个集合当中
基本原理：
- 每个集合用一棵树来表示，树根的编号就是集合的编号
- 每个结点都存储父结点的下标
- 1. 求某结点的集合编号：重复查找父结点，直到父结点是根
- 1. 判断是根结点的方法：p[x]==x //除根结点外的结点，父结点一定是另一结点
- 1. 如何合并两个集合：p[x]=y; //y是另一个集合的编号，这句实现了x集合并入y集合
- 优化（压缩路径）：把查找路径上的所有结点的父结点设为根

#include <iostream>  
  
using namespace std;  
  
const int N=100010;  
  
int n,m;  //n是点的数量，m是操作的数量  
int p[N];  //存储父结点  
  
int find(int x)  //返回x的祖宗结点（所在集合的编号） + 压缩路径  
{  
    if(p[x]!=x) p[x]=find(p[x]);  //如果不是根结点，就让该结点的父结点等于其祖宗结点  
    return p[x];  
}  
  
int main()  
{  
    scanf("%d%d",&n,&m);  //输入  
    for(int i=1;i<=n;i++) p[i]=i;  
    while(m--)  
    {  
        char op[2];  
        int a,b;  
        scanf("%s%d%d",op,&a,&b);  
        if(op[0]=='M') p[find(a)]=find(b);  //让a的祖宗结点的父亲等于b的祖宗结点  
        else  
        {  
            if(find(a)==find(b)) puts("Yes");  
            else puts("No");  
        }  
    }  
    return 0;  
}