什么是并查集?
并查集是一种树型的数据结构,用于处理一些不相交集合(disjoint sets)的合并及查询问题。
简单的并查集实现两个功能:
1.两个不同的集合合并
2.查询某两个元素是否在同一个集合
例题:
一共有 n 个数,编号是 1∼n,最开始每个数各自在一个集合中。
现在要进行 m 个操作,操作共有两种:
M a b,将编号为 a 和 b 的两个数所在的集合合并,如果两个数已经在同一个集合中,则忽略这个操作;
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
思路(个人理解):
设置一个p[N]数组,设想n个不同的集合是n颗不同的树型结构,每个非根节点x(做数组下标)的p[x]保存其父节点的值,只有根节点保存自己的下标值,即:root = p[root] ;这个root就是这个集合的编号,因此集合中的每个元素都和他们的根节点保存了一样的集合编号root,只要不断向上找就行了;两个不同的集合合并即从root1和root2中挑选一个作为根节点,另一个挂在下面作为子节点即可,这样另一个向上找,就找到了根节点,即保存了同样的集合编号值。
代码如下:
#include<iostream>
#include<algorithm>
#include<string>
using namespace std;
const int N = 100010;
int p[N];
int n, m;
int find(int x)
//寻找祖宗节点编号,即是该节点所在集合编号
{
if (p[x] != x) p[x] = find(p[x]);
return p[x];
}
int main()
{
cin >> n >> m;
for (int i = 1 ; i <= n; i ++ ) p[i] = i;
while ( m -- )
{
int x , y;
char ch;
cin >> ch >> x >> y;
if ( ch == 'M' ) p[find(x)] = find(y);
//集合合并
//y不一定是根节点
//将x祖宗节点(编号)的父节点置为y祖宗节点(编号)
else
{
if ( find(x) == find(y) ) cout << "Yes" << endl;
else cout << "No" << endl;
}
}
return 0;
}```