并查集的作用:
1、将两个集合合并
2、判断两个元素是否在一个集合当中
基本原理: 每个集合用一棵树来表示,树根的编号就是整个集合的编号,每个节点存储它的父节点,p[x]表示x的父节点。
问题1: 如何判断是树根: if(p[x]==x)
问题2: 如何求x的集合编号: while(p[x] != x) x = p[x];
问题3: 如何合并两个集合 : px是x的集合编号,py是y的集合编号。p[x]=y
存在的一个复杂度问题: 在求x编号的时候,每一次都要从底部进行遍历到根节点,每次遍历的复杂度都是树的深度。所以采用路径压缩的方式,在第一次遍历之后呢,把除了根节点之外的所有节点都直接连接到根节点上上面,下一次遍历找x编号的时候复杂度就是O(1)
优化: 采用路径压缩
#include <bits/stdc++.h>
using namespace std;
const int N = 100010;
int n,m;
int p[N]; //用来存放每个节点的父节点
int find(int x){
if(p[x]!=x) p[x] = find(p[x]); //如果x不是根节点,递归,直到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); //将b集合的祖宗节点,也就是find(b) 复制给集合a的祖宗节点的父亲
else{ //查看两个集合是否在一个大集合内
if(find(a) == find(b)) puts("yes"); //如果两个根节点的父节点相等
else puts("no");
}
}
return 0;
}