并查集模板题:合并集合
题目
一共有 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
输入样例:
4 5
M 1 2
M 3 4
Q 1 2
Q 1 3
Q 3 4
输出样例:
Yes
No
Yes
核心代码:
//查找x的祖宗结点
int find(int x){
if(p[x]!=x)p[x]=find(p[x]);//表示当前这个数不是祖宗结点
return p[x];//找到祖宗结点返回
}
思路分析:
合并之后,我们该如何去表示合并后的一个集合内元素间的相互关系呢?以及每个集合之间不同的关系呢?
我们可以设立一个祖宗关系,通过数组去表示这样的对应关系:数组下标表示是哪一个数,数组值表示当前这个数对应的父结点(可能是直接指向,也可能是间接指向祖宗结点,看指向的那个数的值是不是和其下标相同)。这样对于相同的集合内的元素的祖宗结点只有一个且相同,对于不同的集合祖宗结点各不相同。如果一个数对应的值等于其自己,表示祖宗结点就是自己,可以看出其自己就是一个单独的集合。
什么是祖宗结点?
祖宗结点就好比一个集合中的领头,集合中的所有元素都听命与这个领头。如果两个数的祖宗结点一样,那就是说明这两个数在同一集合中。如果两个集合要合并,那么就必须要统一只能有一个领头。
选取祖宗结点是任意的,在插入过程中,如果你让前一个数插到(加入)后一个数中,那后一个数就作为该集合的祖宗结点,反之就是前者作文祖宗结点。两个集合合并可以分成两种情况,如果集合2加入集合1(集合2的祖宗结点变为集合1的祖宗结点,集合2元素的祖宗结点都变成集合1的祖宗结点,两者祖宗结点相同,从而实现集合的合并),那么集合2内所有元素的祖宗结点就变成集合1的祖宗结点;集合1加入了集合2(集合1的祖宗结点变为集合2的祖宗结点,从而实现集合的合并),那么集合1内所有元素的祖宗结点就变成集合2的祖宗结点。同样对于两个元素的合并也适用,因为这两个元素可以看做是都只有一个元素的单独集合。
总结:
通过设立祖宗结点去表示一个集合,不同的祖宗结点表示不同的集合,并且我们让同一集合下的数都有一个共性,那就是指向同一个祖宗结点。
ACwing上别人的讲解:
并查集大致讲解
开始时每个集合都是一个独立的集合,并且都是等于自己本身下标的数
例如:
p[5]=5,p[3]=3;
如果是M操作的话那么就将集合进行合并,合并的操作是:
p[3]=p[5]=5;
所以3的祖宗节点便成为了5
此时以5为祖宗节点的集合为{5,3}
如果要将p[9]=9插入到p[3]当中,应该找到3的祖宗节点,
然后再把p[9]=9插入其中,所以p[9]=find(3);(find()函数用于查找祖宗节点)
也可以是p[find(9)]=find(3),因为9的节点本身就是9
此时以5为祖宗节点的集合为{5,3,9};
如果碰到多个数的集合插入另一个集合当中其原理是相同的
例如:
上述中以5为祖宗节点的是p[5],p[3],p[9];(即p[5]=5,p[3]=5,p[9]=5)
再构造一个以6为祖宗节点的集合为{6,4,7,10}
如果要将以6为祖宗节点的集合插入到以5为祖宗节点的集合,则该操作可以是
p[6]=find(3)(或者find(9),find(5))
此时p[6]=5
当然如果是以6为祖宗节点集合中的4,7,10则可以这样
p[find(4)]=find(3)
或者p[find(7)]=find(3)均可以
此时以6为祖宗节点的集合的祖宗节点都成为了5
AC代码:
#include <iostream>
#include <algorithm>
#include <cstring>
#include <cmath>
using namespace std;
typedef long long ll;
typedef pair<int, int> PII;
const int INF = 1e9;
const int N = 1e5 + 5;
int p[N];
int n,m,a,b;
char op;
//查找x的祖宗结点
int find(int 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;//初始,n个集合,每个集合内的元素都是自己的祖宗结点
while(m--){
cin>>op>>a>>b;
if(op=='M'){
p[find(b)]=find(a);//将b所在的集合和a所在集合合并,即b的祖宗结点更新为a的祖宗结点
}
else{
if(find(a)==find(b))cout<<"Yes"<<endl;//祖宗结点相同表示在同一个集合中
else cout<<"No"<<endl;//祖宗结点不同表示不在同一集合
}
}
return 0;
}
注:
通过find操作,会让那些间接指向祖宗结点变成直接指向祖宗结点。当查询某个节点 i 时,如果 i 的父节点不为根节点(祖宗结点)的话,就会进行递归调用,将 i 节点沿途路径上所有节点均指向父节点,这也是find函数进行路径压缩的功能。
// 1 2 3 4 5
// 1 2 3 4 5
// M 1 2
// 1 2 3 4 5
// 1 1 3 4 5
//M 3 2
// 1 2 3 4 5
// 3 1 3 4 5
//Q 2 3
// 1 2 3 4 5
// 3 3 3 4 5