Algs4 union-find算法 带路径压缩的加权Quick-Union实现

Code

package sed.algorithm.chapter1.section5;

public class PathCompressWeightedQuickUnionUF {

private int[] id;
private int[] sz;
private int count;

public PathCompressWeightedQuickUnionUF(int max) {
id = new int[max];
sz = new int[max];

for (int i = 0; i < id.length; i++) {
id[i]=i;
}
for (int i = 0; i < id.length; i++) {
sz[i]=1;
}
count = max;
}

public int count(){
return count;
}

public int find(int p){
int pParent = p;
while(id[pParent]!=pParent){
pParent=id[pParent];
}

while(id[p]!= p){
p=id[p];
id[p]=pParent;
}

return pParent;
}

public boolean connected(int p, int q){
return find(p)== find(q);
}

public void union(int p, int q){
int pRoot = find(p);
int qRoot = find(q);

if (pRoot!=qRoot) {
if (sz[pRoot] < sz[qRoot]) {
id[pRoot] = qRoot;
sz[qRoot] += sz[pRoot];
}else {
id[qRoot] = pRoot;
sz[pRoot] += sz[qRoot];
}
count--;
}
}

}

Input

100 million points , 200million links

Notes：

速度不是一般的快，之前用quick union 算法 跑了一个多小时。。。
(其实我也不想的，因为测试代码里面类引用错了，一直以为用的是这个算法，其实用的是最简单的quick union)。TT 改了以后还没要一分钟。