并查集:是一种用来管理元素分组情况的数据结构,并查集可以高效地进行如下操作:
1)查询元素a和元素b是否属于同一组。
2)合并元素a和元素b所在的组。
并查集的功能 | 并查集的结构 |
---|---|
图例 | 说明 |
---|---|
初始化:我们准备n个结点来表示n个元素。最开始时没有边。 | |
合并:从一组的根向另一组根的连边,这样两棵树就变成了一棵树,也就把两个组合并为一个组了。 | |
合并:进一步合并。 | |
查询:为了查询两个节点是否属于同一组,我们需要沿着树向上走,来查询包含这个元素的树的根是谁。如果两个节点走到了同一个根,那么就可以知道它们属于同一组。 | |
在树形数据结构里,如果发生了退化的情况,那么复杂度就会变得很高。因此,有必要想办法避免退化的发生。在并查集中, 只需按照如下方法就可以避免退化:1)对于每棵树,记录这棵树的高度(rank);2)合并时如果两棵树的rank不同,那么从rank小的向rank大的连边。 | |
此外,通过路径压缩可以使得并查集更加高效。对于每个节点,一旦向上走到了一次根结点,就把这个点的父亲的边改为直接连向根。 | |
在此之上,不仅仅是所查的节点,在查询过程中向上经过的所有节点,都改为直接连到根上。这样再次查询这些节点时,就可以很快就知道根是谁了。 |
//用编号代表每个元素
int par[MAX_N];//表示的是父亲的编号
int rank[MAX_N];//树的高度
//初始化n个元素
void init(int n){
for(int i=0;i<n;i++){
par[i]=i;
rank[i]=0;
}
}
//查询树的根
int find(int x){
if(par[x]==x)
return x; //x是所在树的根
else
return par[x]=find(par[x]);//递归:路径压缩
}
//合并x和y所属的集合
void unite(int x,int y){
x=find(x);
y=find(y);
if(x==y) return;
//为了避免退化,设置树高rank[]
if(rank[x] < rank[y])
par[x]=y;
else{
par[y]=x;
if(rank[x]==rank[y])
rank[x]++;
}
}