并查集
并查集是一种树形结构,又叫“不相交集合”,保持了一组不相交的动态集合,每个集合通过一个代表来识别,代表即集合中的某个成员,通常选择根做这个代表。
其实并查集很常用,比如果刚进入大学还没有分班,这个时候你并不知道你的同学,此时可以看做每个人都是独立的。而分班后,则你与同班的同学有了关系,同班同学的关系。在并查集中,若同学a与同学b是同班同学,则他们都会有一个共同点——有着相同的root(根),这个根也是同班的某一个同学,只是借助它来区分a与b的关系。比如说a与root同学c是同班同学,b与root同学c也是同班同学,则a与b也是同班同学。
以下是带路径压缩的一个并查集模板:
find(int p)表示找这个元素的根
connect(int p,int q)将所处不同根的两个合并 比如将一个班与另一个班合并组成一个更大的班
图例:(union与connect相同)
此时每个元素所在“根系”中都只有自己,每元素都是root (i=id[i])
结果:
此时 如果find(3)结果为4 代表3的root为4
//Union Find class
class UF {
public:
int* id; //父链接数组
int* sz; //各个节点所对应的分量的大小
int count; //连通分量的个数
UF(int n) {
count = n; //初始化 任意两个元素不连通因此由n个连通分量
id = new int[n];
sz = new int[n];
fill(sz, sz + n, 1); // 初始每个点都只有一个分量就是自己
for (int i = 0; i < n; i++) id[i] = i;
}
int connect(int p, int q) {
int i = find(p);
int j = find(q);
if (i == j) return 0;
if (sz[i] < sz[j]) { id[i] = j; sz[j] += sz[i]; }
else { id[j] = i; sz[i] += sz[j]; }
count--;
}
int find(int p) {
while (p != id[p]) {
id[p] = id[id[p]]; //路径压缩 让父亲指向其祖父
p = id[p];
}
return p;
}
};
路径压缩:
树的深度越低,查找根的速度就越慢。
因此我们要做的是将深度降低
int find(int p) {
while (p != id[p]) {
id[p] = id[id[p]]; //路径压缩 让父亲指向其祖父
p = id[p];
}
return p;
}
每次查找的时候 都让其父节点指向祖父节点
但[算法第四版]提到在实际运用中 两种展开差异不大
还一种方式是每次查根都让其指向root 如下图所示
首先让p指向root 再让p的父节点6指向root 再让6的父节点指向root 依次类推
注:本文配图来自算法第四版的公开课