定义
并查集是一种数据结构,可以高效的实现对【集合】的【并】与【查】操作
- 查找(Find):确定某个元素处于哪个子集;
- 合并(Union):将两个子集合并成一个集合。
实现思路
-
p[i]为i节点的父节点的值
-
每一个集合都可以用一个树表示
-
p[i]=i表示该节点为所在集合的树的根节点
-
给数组p初始化赋值,设其值为p[i]=i,也即是说,p[1]=1,p[2]=2,以此类推。
-
如果p[2]=1,p[3]=2则有:3节点的父节点为2,2节点的父节点为1,1节点的父节点为1,因此1为根节点。【1、2、3】在同一个集合内
代码
查询
- 查找一个集合的根节点(由于每个集合都可以用一个树形结构来表示,因此该树的根节点可以唯一标志一个集合)。下面代码在查找根节点的过程中附带了路径压缩功能
- 路径压缩前:
- 路径压缩后:
int find(int x)//寻找x的父节点
{
if(p[x]!=x)//说明x不是根节点,因此寻找x的父节点的父节点,并且通过赋值进行路径压缩
p[x]=find(p[x]);
return p[x];
}
合并
合并两个集合只需将一个集合的根节点指向另一个集合的根节点即可
合并a,b两个集合:p[find(a)]=find(b);
#include <iostream>
using namespace std;
const int N=1e5+10;
int p[N];
int find(int x)//寻找x的父节点
{
if(p[x]!=x)//说明x不是根节点,因此寻找x的父节点的父节点,并且通过赋值进行路径压缩
p[x]=find(p[x]);
return p[x];
}
int main()
{
int n,m;
cin>>n>>m;
for(int i=1;i<=n;i++) p[i]=i;
for(int i=0;i<m;i++)
{
char c;
int a,b;
cin>>c>>a>>b;
if(c=='M')//将a,b集合合并
p[find(a)]=find(b);
else if(c=='Q')//查询a,b是否在一个集合内
if(find(a)==find(b)) puts("Yes");
else puts("No");
}
return 0;
}