用途:
1.将两个集合合并
2.询问两个元素是否在一个集合当中
算法原理:
每个集合用一棵树来表示。树根的编号就是整个集合的编号。每个节点储存它的父节点,x的父节点用p[ x ]表示。
模板题目:
一共有 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
相关模块:
1.初始化,让所有节点指向自己(起初,所有节点都是根节点)
//初始化并查集
void init(int n)
{
for (int i = 1; i <= n; i++)
{
p[i] = i; //让所有的节点自己指向自己
}
}
2.查找集合编号(用于判断是否在同一个集合,并用于合并集合)
我们通过递归实现路径压缩来实现;(即我们让同一个集合中的每个节点的父节点指向根节点,使效率提高)
即由下图所示:
代码实现:
int find (int x)
{
//递归结束出口
if(x == p[x])//找到自己指向自己的节点就是根节点
return p[x];
else
{
p[x] = find(p[x]); //递归
}
}
3.读入数据与判断操作;
当为合并2个集合时:我们只需将其中一个集合的根节点的父节点指向另一个集合的根节点;如下图所示:
代码:
p[find(a)] = find(b);
当判断两个数是否在一集合时,只需判断其根节点是否相同。
代码:
if (find(a) == find(b)) cout << "Yes" << endl;
else cout << "No" << endl;
综合上:并查集的代码:
#define _CRT_SECURE_NO_WARNINGS 1
#include<iostream>
using namespace std;
const int N = 100010;
int n, m;
int p[N];
//初始化并查集
void init(int n)
{
for (int i = 1; i <= n; i++)
{
p[i] = i; //让所有的节点自己指向自己
}
}
//返回x所在集合的编号 + 递归实现路径压缩
int find(int x)
{
//递归结束出口
if (p[x] == x) //找到自己指向自己的节点就是根节点
return p[x];
else
{
p[x] = find(p[x]);
return p[x];
}
}
int main()
{
cin >> n >> m;//分别读入点的数量和操作的个数
init(n);
while (m--)
{
char op[2];
int a, b;
scanf("%s %d %d", op, &a, &b);
if (op[0] == 'M')//合并集合
p[find(a)] = find(b);
else //判断是否在集合里
{
if (find(a) == find(b)) cout << "Yes" << endl;
else cout << "No" << endl;
}
}
}