并查集算法

并查集，在一些有N个元素的集合应用问题中，我们通常是在开始时让每个元素构成一个单元素的集合，然后按一定顺序将属于同一组的元素所在的集合合并，其间要反复查找一个元素在哪个集合中。

一、并查集两个操作：
1、将两个集合合并
2、询问两个数是否在一个集合中

二、基本原理： 每个集合利用一棵树来表示。树根的编号就是整个集合的编号，每个结点存储它的父节点，p[x]表示x的父结点

我们在这里将每个集合的编号存储在树根结点中，也称为祖宗结点，易知，在刚开始时每个元素都是各自在一个集合内，此时每个元素都满足p[x]=x ，但在不断的合并操作后，只有每棵树的祖宗结点的p[x]是等于x的（即根）。

上图绿色字体就是每个结点的父结点

三、明确3个问题：
1、如何判断树根： if(p[x] == x)
2、如何求x的集合编号： while(p[x] != x) x=p[x];

沿着x的父结点向上走，直到走到祖宗结点，此时x就赋值为祖宗结点的编号，即集合的编号

图中红色路径就是找4号结点的祖宗结点，即找4号元素的集合编号

3、如何合并两个集合： p[x]=y

即将x集合的根结点直接接到y集合的根结点上，也就是x集合的根结点变成y了，即p[x]=y（也可以py接到px上，即p[y]=x）

四、路径压缩优化

还有一点就是在查找一个结点的祖宗结点（查找集合编号）时有一个路径压缩优化，就是当查找x结点的祖宗结点时，让这一条路径上的所有经过结点都指向祖宗结点

例如下图： 在查找6号结点的祖宗结点后，将沿这条路径的所有结点（2、4结点）直接指向祖宗结点，这样，如果需要再查找2、4结点的祖宗结点，就可以很快了

五、重点代码实现
利用递归，实现寻找祖宗结点+路径压缩，整个代码的重点就在find函数上了

int find(int x)  //返回x的祖宗结点 + 路径优化 
{
	if(p[x]!=x) p[x] = find(p[x]);  // p[x]!=x，说明不是祖宗结点
	return p[x];
}

如果不理解这个，看到别人将这个模板拆分了，会方便理解些

int find (int x)
{
    if(p[x] != x)
    {
        int t = find(p[x]);  //寻找祖宗结点
        p[x] = t;  //路径压缩
    }
    return p[x];
}

六、例题
AcWing 836. 合并集合
题目描述
一共有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

#include<iostream>
using namespace std;
const int N=1e5+10;
int n,m;
int p[N]; //p[N]表示每个结点的父结点

int find(int x) //返回x的祖宗结点 + 路径优化 
{
	if(p[x]!=x) p[x] = find(p[x]);
	return p[x];
}

int main()
{
	cin>>n>>m;
	for(int i=1; i<=n; i++) p[i] = i; 
	//刚开始每个元素都是各自在一个集合内，此时所有元素的p[x]=x 
	
	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;	
		}  
	}
}

AcWing 837. 连通块中点的数量
题目描述
给定一个包含n个点（编号为1~n）的无向图，初始时图中没有边。

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

“C a b”，在点a和点b之间连一条边，a和b可能相等；
“Q1 a b”，询问点a和点b是否在同一个连通块中，a和b可能相等；
“Q2 a”，询问点a所在连通块中点的数量；
输入格式
第一行输入整数n和m。

接下来m行，每行包含一个操作指令，指令为“C a b”，“Q1 a b”或“Q2 a”中的一种。

输出格式
对于每个询问指令”Q1 a b”，如果a和b在同一个连通块中，则输出“Yes”，否则输出“No”。

对于每个询问指令“Q2 a”，输出一个整数表示点a所在连通块中点的数量

每个结果占一行。

数据范围
1≤n,m≤105

#include<iostream>
using namespace std;
const int N=1e5+10;

int n,m;
int p[N],size[N]; //p[N]存每个结点的父结点，size[N]存每个集合的元素个数
//只有祖宗结点的size才有意义 

int find(int x) //返回x的祖宗结点+路径优化 
{ 
	if(p[x]!=x) p[x]=find(p[x]); //p[x]!=x说明不是祖宗结点 
	return p[x];
}

int main()
{
	cin>>n>>m;
	for(int i=1; i<=n; i++)
	{
		p[i]=i;    //刚开始每个元素都是各自在一个集合内，此时所有元素的p[x]=x 
		size[i]=1; // 刚开始每个集合都只有一个元素 
	}
	
	while(m--)
	{
		char op[5];
		int a,b;
		cin>>op;
		
		if(op[0] == 'C') //合并 
		{
			cin>>a>>b;
			if(find(a) == find(b)) continue; //a和b已经在一个集合了
			size[find(b)] += size[find(a)]; //合并后b(size)=b(size)+a(size)
			p[find(a)] = find(b); 
		}
		else if(op[1] == '1') //查询是否在同一集合 
		{
			cin>>a>>b;
			if(find(a) == find(b)) cout<<"Yes"<<endl;
			else cout<<"No"<<endl; 
		}
		else //查询集合内元素个数
		{
			cin>>a;
			cout<<size[find(a)]<<endl;
		}
	}
	return 0;
}

还有一道关于并查集的题：L2-010 排座位 (25分) 并查集