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 改了以后还没要一分钟。