算法设计与分析——并查集

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

其中最重要的函数find函数（利用递归寻找根节点）

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

例题：

题目描述
一共有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

算法流程：

①刚开始每个元素都是各自在一个集合内，将所有的p[x]都赋值为x，即每个结点的根节点都是自己。

②关于合并问题：将a所在集合与b所在集合进行合并，需要将a的根节点连接在b的根节点上，意思就是a的根节点是b了。

③关于查找a,b是否在同一集合中的问题：运用find函数，如果根节点相同，就输出Yes，不同，则输出No。

#include<bits/stdc++.h>
using namespace std;
int p[1000];

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);
        }
        if(op=='Q')
        {
            if(find(a)==find(b))cout<<"Yes"<<endl;
            else
            {
                cout<<"No"<<endl;
            }
        }
    }

}

题目描述
给定一个包含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

需要注意的是，在指令中存在Q1和Q2，为了能够区分它们，需要char op[ ],根据op[1]来判断需要执行哪个操作。

和上一道题相比，多了一个计算连通块中点的数量的问题，需要我们再创建一个数组size[ ],在开始时将每个size[ ]都置为1，如果a,b合并，那么size[find(b)]=size[find(b)]+size[find(a)]。
 

#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]); 
	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)
            size[find(a)]=size[find(b)];//a和b在同一个连通块中，将a中元素的数量进行更新
			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;
}