并查集是一种树型的数据结构,用于处理一些不相交集合的合并及查询问题。
并查集常用的2中操作:
- 查找(find):查询两个元素是否在同一集合中
- 合并(union):把两个不相交的集合合并为一个集合
并查集的思想是用一个数组表示了整片森林(parent),树的根节点唯一标识了一个集合,我们只要找到了某个元素的的树根,就能确定它在哪个集合里。
并查集相关操作实现
1. 查找—find()
int find(int x){
while(x != p[x])
x = p[x];
return x;
}
2. 合并—union
//将i、j所在的集合进行合并
void union(int i,int j){
int fx = find(i),fy = find(j);
if(fx != fy) p[fx] = fy;
}
3. 路径压缩
并查集里的 find
函数里可以进行路径压缩,是为了更快速的查找一个点的根节点。对于一个集合树来说,它的根节点下面可以依附着许多的节点,因此,我们可以尝试在 find
的过程中,从底向上,如果此时访问的节点不是根节点的话,那么我们可以把这个节点尽量的往上挪一挪,减少数的层数,这个过程就叫做路径压缩。
int find(int x){
int r = x; //记录集合老大的下标,进行后续路径压缩
while(p[r] != r) //如果值和下标不等,说明这不是老大,老大的值就是自己的下标
r = p[r];
int j;
while(x != r){
j = p[x]; //保存x的上级
p[x] = r; //x的上级改为老大
x = j;
}
return r; //返回老大的下标
}