- 并查集是 n 叉树
- 每次调用 find() 方法都会进行路径压缩,使得树的高度一直保持在不超过3层,不可能有更多的情况。一般的高度应该是2,但当union()的时候,A树接到B树上,如果A树和B树的高度都是2,那么接起来B树的高度就是3了。然后如果对第三层结点调用 find() 方法,第三层就又会被压缩到第二层。
class UF {
private int count;
private int[] parent;
private int[] size;
public UF(int n) {
count = n;
parent = new int[n];
size = new int[n];
for (int i = 0; i < count; i++) {
parent[i] = i;
size[i] = 1;
}
}
public void union(int p, int q) {
int rootP = find(p);
int rootQ = find(q);
if (rootP == rootQ) {
return;
}
if (size[rootP] > size[rootQ]) {
parent[rootQ] = rootP;
size[rootP] += size[rootQ];
} else {
parent[rootP] = rootQ;
size[rootQ] += size[rootP];
}
count--;
}
public boolean connected(int p, int q) {
int rootP = find(p);
int rootQ = find(q);
return rootP == rootQ;
}
public int find(int x) {
while (parent[x] != x) {
parent[x] = parent[parent[x]];
x = parent[x];
}
return x;
}
public int count() {
return count;
}
}