AcWing 836.合并集合

假设现在有集合A(1,2,3)，B(5,6,7)

朴素算法：用数组存储数据，如果要合并，就将其中一个数组，加到另一个数组之后。

并查集:我们可以用树来存储元素(祖先节点为集合名)，

A是元素1，2，3的祖先节点，B是元素5，6，7的祖先节点

现在将集合B合并成集合A时，只需要将5，6，7的祖先节点等于A即可，就是p[5]=p[6]=p[7]=A;

find()：查找祖先节点函数。

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

祖先节点定义就是：节点的父节点是本身，p[x]==x。因为2的父节点是1，1的父节点就是本身（在图中是A，但是实际程序中，就是以第一个元素来命名集合的，find(1)==1）。

但这样查找的话还是很麻烦，需要从底下依次向上遍历，此时需要对于find函数（查找祖先节点）进行路径压缩优化。那么，怎么压缩呢？

就是在不断递归中，直接将元素指向他的祖先节点。

 836. 合并集合 - AcWing题库

#include<iostream>
using namespace std;
const int N=10e5+10;
int n,m;
int p[N];
int find(int x)//查找根节点，并完成路径压缩！
{
    if(p[x]!=x) p[x]=find(p[x]);//p[x]==x 是根节点 不停向父节点递归 直到找到 
    return p[x];
}
int main()
{
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++) p[i]=i;
	while(m--)
	{
		char op[2];
		int a,b;
		scanf("%s%d%d",op,&a,&b);
	    if(*op=='M') p[find(a)]=find(b);//合并 a的根节点的父节点等于b的根节点
	    else if(find(a)==find(b)) puts("Yes");//根节点相同 在同一集合 不同 在不同集合
	    else puts("No");
	}
	return 0;
}

更细致的讲解可以看（AcWing 836. 并查集(保姆级讲解) - AcWing）：