概念:
并查集是一种树型的高级数据结构,主要用于处理不相交集合的合并及查询问题。顾名思义,并查集有两个重要的操作:合并和查询。一般用来判断两个元素是否属于同一个集合,我们只需让同一个集合中的所有元素都指向同一个祖先,后面判断这两个元素的祖先是否是同一个即可判断它们是否属于同一集合。
过程:
- 初始化:将每个元素的父亲都设为其本身。
- 合并:先查询这两个元素的父亲,如果不属于同一个集合,将这两个集合中的任一集合的祖先指向另一个集合的祖先,那么这两个集合就完成了合并。
- 查询:查找指定元素所在的集合,即查找不断查找其父亲直到找到祖先(根节点)。
以并查集的模板题(家族)为例子,代码如下:
#include <iostream>
using namespace std;
const int maxn=1e5+10;
int fa[maxn];
void init(int n){//初始化,每个元素的父亲都是自己
for(int i=1;i<=n;i++){
fa[i]=i;
}
}
int find(int x){//查询
if(fa[x]!=x){//路径压缩
fa[x]=find(fa[x]);
}
return fa[x];
}
void merge(int x,int y){//合并
int fx=find(x);
int fy=find(y);
if(fx!=fy) fa[fx]=fy;
}
int main(){
int n,m,p;//n个元素,m个关系(合并),p次查询
cin >> n >> m >> p;
init(n);
int x,y;
for(int i=0;i<m;i++){
cin >> x >> y;
merge(x,y);
}
for(int i=0;i<p;i++){
cin >> x >> y;
if(find(x)==find(y)) cout << "Yes" << endl;
else cout << "No" << endl;
}
return 0;
}
关于上面的路径压缩:一个集合就是一棵树,父亲即为根节点,因为我们只关心根节点而不关心父子关系,所以我们在每次在find()的过程中,让路径上的元素都指向根节点。这样树就从细长型变成了扁平型,提高了查找效率。