AcWing 836. 合并集合(朴素并查集)

题目链接 ： 点击查看

题目描述 ：

一共有 n 个数，编号是 1∼n，最开始每个数各自在一个集合中。

现在要进行 m 个操作，操作共有两种：

1. M a b，将编号为 a 和 b 的两个数所在的集合合并，如果两个数已经在同一个集合中，则忽略这个操作；
2. Q a b，询问编号为 a 和 b 的两个数是否在同一个集合中；

输入输出格式 ：

输入

第一行输入整数 n 和 m。

接下来 m 行，每行包含一个操作指令，指令为 M a b 或 Q a b 中的一种。

输出

对于每个询问指令 Q a b，都要输出一个结果，如果 a 和 b 在同一集合内，则输出 Yes，否则输出 No。

每个结果占一行。

输入输出样例 ：

输入

4 5
M 1 2
M 3 4
Q 1 2
Q 1 3
Q 3 4

输出

Yes
No
Yes

题目分析 ：

并查集是一种树形结构，常用来处理一些不相交集合的分组与查询问题，其通常支持两种操作，1 . 合并 ： 把两个不相交集合合并为1个集合  2. 查询 ： 查询两个元素是否在同一个集合中。并查集的每一个集合都是一个树形结构，其根节点元素的下标即为整个集合的标号(下标)。要想建立并查集，我们需要一个辅助数组p[N[来记录每个节点的父节点是多少，特别地，根元素的父节点就是他本身。在初始化并查集时，我们把每个元素都当成一个单独的集合，也就是每个数都是由其自己构成集合的根节点，即初始化过程为p[i] = i。建立完毕后，我们要想查询a，b两个数是否在同一集合，就要判断他们相应集合的根节点是否为同一个，因此要 不断进行 a = p[a] , b = p[a],顺着父节点不断向上爬，直到遇到p[a] = a, p[b] = b, 如此整个代码的时间复杂度还是相当高的。这里我们可以采取路径压缩算法优化，即我们让集合内每个节点的父节点直接为根元素，这样便可在近似O(1)的时间复杂度中查询两个数是否为同一个集合，这个可以在find函数中实现，if(p[x] != x) (还没遍历到根节点) p[x] = find(p[x]) (这里是递归find函数使其不断向上移动，直到找到根节点，之后开始回溯，使得每个节点的父元素p[x]就等于(指向)根节点),find()函数返回值便是当前元素所在的集合编号(下标)。关于两个集合合并问题，我们直接将其中一个集合的根节点接到另外一个集合的根节点，即如px的根节点值为x，py根节点值为py，则令p[px] = y, 在此配合find函数的递归过程，便可令x集合中所有元素所在的集合编号为y，即p[find(a)] = find(b),相当于一棵树作为了另一棵树的分支。详见如下代码.

代码 ：

#include<iostream>
#include<cstdio>
using namespace std;
const int N = 1e5 + 7;
int p[N];//储存节点n的父节点

int find(int x) {
	if (p[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;
	while (m -- ) {
		char op;
		int a, b;
		cin >> op >> a >> b;
		if (op == 'M') p[find(a)] = find(b);
		else {
			if (find(a) == find(b)) cout << "Yes" << endl;
			else cout << "No" << endl;
		}
	}
	return 0;
}

---------------------------------------------------

在此我们给出朴素并查集的相关模板

朴素并查集：

    int p[N]; //存储每个点的祖宗节点

    // 返回x的祖宗节点
    int find(int x)
    {
        if (p[x] != x) p[x] = find(p[x]);
        return p[x];
    }

    // 初始化，假定节点编号是1~n
    for (int i = 1; i <= n; i ++ ) p[i] = i;

    // 合并a和b所在的两个集合：
    p[find(a)] = find(b);