算法基础-并查集

思路

并查集主要实现两个功能
1.将两个集合合并
2.询问两个元素是否在一个集合中

题目链接：https://www.acwing.com/problem/content/838/

现在假设一共只有10个数，分别是1~10，我想让1，2合并，2，3合并，3，4合并，4，5合并，6-8同上合并，9，10合并，那么我会先创建1个容纳10个元素的数组

让数组的第x位置存储x（下标从1开始），以此代表起始时每个数各自在一个集合的情况。接着我们想一想该如何把不同的集合合并呢？我们不妨这样做，当我们想把1，2合并时，将数组第2个元素的下标改为1，把2，3合并时，将第三个元素的下标改为2······这样我们就可以把同集合的元素串在一棵树上

现在我想知道3和5是否在同一个集合中，我们只需要知道他们的根（即顶上的数据）是否相同，那么当我检查5的根节点时，我只要通过第5个元素的数据不断向上找到一个下标与数据的值相等的元素即可，3亦然

最后发现3和5所指向的根节点是相同的，所以他们在同一集合。这是答题思路，但是我们会发现每次都这样逐个向上寻找根节点也是比较耗时的，我们可不可以想办法让下面的元素直接指向根节点呢？即这样的结构

此操作可以让代码的时间复杂度降为O(1)，当我们想知道3和5是否在同一集合中时，只需要比较p[3]==p[5]即可，具体操作我会在代码中详细解释

代码

完整代码

#include<iostream>
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]);
	return p[x];
}

int main()
{
	scanf("%d%d", &n, &m);
	for (int i = 1; i <= n; i++)
	{
		p[i] = i;
	}
	while (m--)//进行m次操作
	{
		char op[2];//操作类型 M或Q
		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)) printf("Yes\n");
			else printf("No\n");
		}
	}
	return 0;
}

代码解释

代码的核心是find函数

int find(int x)
{
	if (p[x] != x) p[x] = find(p[x]);
	return p[x];
}

回头看看上面的图，我们会发现数组的根节点下标和数据一定是相同的，所以当我们想得到根节点时就可以用到这个特性

当我们找到根节点时，将其下标或者值赋给其所有子孙节点，这里使用了递归，这样我们只要搜索过一次就可以把其子孙节点全部赋值