【数据结构】并查集基础模板

 一、并查集概念

        定义：并查集作为一种树形的数据结构，主要用来合并、查找不相交集合（即并、查）。使用并查集时还可以维护其他信息，例如将集合内元素的个数保存在根节点中。

        构成：并查集通过数组f[]保存，通过find()函数完成查找功能，通过join()实现合并功能。如果需要维护额外信息，可以新建一个size[]保存。

二、例题：acwing合并集合

合并集合题干：

这是一道并查集模板题，只要实现了并查集基本功能就能够做出来。

输入输出和数据范围：

可以看到输入有n、m两个整数，数据范围在0~1e5之间，可以用int n,m;存下。

接下来m行指令，指令包括一个字符和两个范围在1~n之间的编号，总共3*m个变量，但是因为每一行指令可以单独处理，所以这里我们通过int a,b;和char s;就可以完成存储，每遇到新一行指令更新一次变量。

int n,m；//n编号个数，m指令个数
char s;//指令，用来确定进行什么操作
int a,b;//指令中的元素，确定参与指令的两个元素

同时我们需要一个数组来保存编号为1~n的各个元素。

f[]保存了数组下标对应节点的根节点编号。

const int N = 1e5 + 10;
int f[N];

f[N]数组的初始化：

需要注意的一点是，f[N]保存了各个结点的根节点，在初始时根节点就是他们自身。因此需要对f[N]数组进行初始化，将其指向自身。

for(int i=1;i<=n;i++)f[i] = i;//初始化

find（）函数的实现：

完成了变量的定义，就可以开始实现基本并查集功能了。

查找功能：find（）函数需要查找一个节点元素的根节点，因此我们对它传入一个节点编号，并反回根节点的编号。

如果所有结点全部头尾相接，那么并查集就会退化成单链表，效率降低。因此我们需要通过压缩路径来解决这个问题，即：每次查找之后之间将该结点指向它的根结点。

int find(int x)//搜索根节点 
{
	if(f[x]!=x) f[x] = find(f[x]);//路径压缩
	return f[x];
}

join（）函数的实现：

合并功能：如果两个结点不在一个集合内，则将其中一个结点的根节点指向另一个结点的根节点，实现集合合并。

void join(int a,int b)//合并集合 
{
	if(find(a)!=find(b))
	{
	    f[find(a)] = find(b);
	}
}

题解代码：

至此我们就完成了全部核心功能的设计，根据题意编写代码即可。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N = 1e5 + 10;

int f[N];
int n,m,a,b;
char s;

int find(int x)//搜索根节点 
{
	if(f[x]!=x) f[x] = find(f[x]);
	return f[x];
}

void join(int a,int b)//合并集合 
{
	if(find(a)!=find(b))
	{
	    f[find(a)] = find(b);
	}
}

int main()
{

	cin>>n>>m;
	
	for(int i=1;i<=n;i++)f[i] = i;//初始化
	
	for(int i=0;i<m;i++)
	{
		cin>>s>>a>>b;
		if(s=='M')
		{
			join(a,b);
	    }
		else if(s=='Q')
		{
			if(find(a)==find(b)) cout<<"Yes\n";
			else cout<<"No\n";
		}
	 } 
	
	return 0;
}