并查集实现类

class UF {
        private int[] parent;
        private int count;  // 连通分量的个数

        // n是图中节点的个数
        public UF(int n) {
            this.count = n;
            parent = new int[n];
            for(int i = 0;i < n;i++)
                parent[i] = i;
        }

        // 将节点p和节点q连通
        public void union(int p, int q) {
            int rootP = find(p);
            int rootQ = find(q);
            if(rootP == rootQ) return;

            parent[rootP] = rootQ;
            count--;
        }

        // 判断节点p和节点q是否连通
        public boolean isConnected(int p, int q) {
            return find(p) == find(q);
        }

        // 寻找x节点的根节点, 并进行路径压缩
        public int find(int x) {
            if(parent[x] != x) {
                parent[x] = find(parent[x]);
            }

            return parent[x];
        }

        public int count() {
            return this.count;
        }
    }