并查集
并查集是一种树型的数据结构,用于处理一些不相交集合(disjoint sets)的合并及查询问题。
基本操作
- 合并两个集合
- 询问两个元素是否在同一个集合中
基本原理
每个集合用一棵树来表示。数根的编号就是整个集合的编号。每个结点存储它的父结点编号,即p[x]
表示x的父结点,那么:
- 如果
p[x] == x
,表示x
是树根 - 求
x
的集合编号:while(p[x] != x) x = p[x];
- 合并两个集合:
p[x]
是x所在集合的编号,p[y]
是y所在集合的编号,p[x] = y
查找优化
求x
的集合编号时,while(p[x] != x) x = p[x];
,可以发现查找次数与树的高度相关。为了能够使后续查找效率更高,在第一次查找过程中,可以将当前路径中的所有结点的父亲都设置为根结点。这样该条路径上所有点到根结点的距离都为1,这种优化方式就称为路径压缩。
正因为有了上述优化方式,并查集在询问两个元素是否在同一个集合时,时间复杂度接近
O
(
1
)
O(1)
O(1)。
模板问题
一共有 n n n 个数,编号是 1 1 1∼ n n n,最开始每个数各自在一个集合中。
现在要进行 m m m 个操作,操作共有两种:
M a b
,将编号为 a 和 b 的两个数所在的集合合并,如果两个数已经在同一个集合中,则忽略这个操作;Q a b
,询问编号为 a 和 b 的两个数是否在同一个集合中;
输入格式
第一行输入整数
n
n
n 和
m
m
m。
接下来
m
m
m 行,每行包含一个操作指令,指令为 M a b
或 Q a b
中的一种。
输出格式
对于每个询问指令 Q a b
,都要输出一个结果,如果 a 和 b 在同一集合内,则输出 Yes,否则输出 No。
每个结果占一行。
数据范围
1
≤
n
,
m
≤
1
0
5
1≤n,m≤10^5
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 <iostream>
#include <cstdio>
using namespace std;
const int N = 100010;
int n, m, p[N];
//返回x的祖宗编号+路径压缩
int find(int 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;
int a, b;
char op[2];
while(m--){
scanf("%s%d%d", op, &a, &b); //读入一个字符串,可以过滤掉多余空格
int u = find(a), v = find(b); //找到a,b的祖宗编号
if(op[0] == 'Q'){
if(u == v) puts("Yes");
else puts("No");
}
else p[u]=v;
}
return 0;
}