10. 并查集

并查集是一种树型的数据结构，并查集可以高效地进行如下操作：

查询元素p和元素q是否属于同一组

合并元素p和元素q所在的组

10.1 并查集结构

并查集也是一种树型结构，但这棵树跟我们之前讲的二叉树、红黑树、B树等都不一样，这种树的要求比较简单：

每个元素都唯一的对应一个结点；
每一组数据中的多个元素都在同一颗树中；
一个组中的数据对应的树和另外一个组中的数据对应的树之间没有任何联系；
元素在树中并没有子父级关系的硬性要求；

10.2 并查集API设计

10.3 并查集的实现

10.3.1 UF(int N)构造方法实现

初始情况下，每个元素都在一个独立的分组中，所以，初始情况下，并查集中的数据默认分为N个组；
初始化数组eleAndGroup；
把eleAndGroup数组的索引看做是每个结点存储的元素，把eleAndGroup数组每个索引处的值看做是该结点

所在的分组，那么初始化情况下，i索引处存储的值就是i

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-0aCzadvl-1682237162968)(null)]

10.3.2 union(int p,int q)合并方法实现

如果p和q已经在同一个分组中，则无需合并
如果p和q不在同一个分组，则只需要将p元素所在组的所有的元素的组标识符修改为q元素所在组的标识符即

可
分组数量-1

10.3.3 代码

package com.ynu.Java版算法.U10_并查集.T1_并查集实现;

public class Main {
    public static void main(String[] args) {

        // 创建10个元素的并查集
        UF uf = new UF(10);
        System.out.println(uf.count());

        // 把1,9合并到4
        uf.union(1,4);
        uf.union(9,4);

        // 合并2，3两组  把5合并到3
        uf.union(2,3);
        System.out.println(uf.count());

        // 合并5，3两组  把5合并到3
        uf.union(5,3);
        // 输出3组的所有元素
        System.out.println(uf.findAll(3));

        // 把4合并到3
        uf.union(4,3);
        System.out.println(uf.findAll(3));


    }
}

package com.ynu.Java版数据结构.U10_并查集.T1_并查集实现;

import java.util.ArrayList;
import java.util.List;

public class UF {

    // 记录节点元素和该元素所在分组的标识
    private int[] eleAndGroup;
    // 记录并查集中数据的分组个数
    private int count;

    //初始化并查集
    public UF(int N) {

        //初始情况下，每个元素都在一个独立的分组中，所以，初始情况下，并查集中的数据默认分为N个组
        this.count = N;
        //初始化数组
        eleAndGroup = new int[N];

        // 把eleAndGroup数组的索引看做是每个结点存储的元素，把eleAndGroup数组每个索引处的值看做是该
        // 结点所在的分组，那么初始化情况下，i索引处存储的值就是i
        for (int i = 0; i < eleAndGroup.length; i++) {
            eleAndGroup[i] = i;
        }

    }

    // 获取当前并查集中有多少个分组
    public int count(){
        return count;
    }

    // 判断并查集中元素p和元素q是否在同一分组中
    public boolean connected(int p,int q){
        return find(p)==find(q);
    }

    // 查找元素p所在分组的标识符
    public int find(int p){
        return eleAndGroup[p];
    }

    // 把元素p和元素q所在的分组合并
    // p合并到q上
    public void union(int p,int q){

        if (connected(p,q)){
            System.out.println("p,q本来就在一个分组");
            return;
        }

        int pGroup = find(p);
        int qGroup = find(q);


        for (int i = 0; i < eleAndGroup.length; i++) {
            if (eleAndGroup[i]==pGroup){
                eleAndGroup[i] = qGroup;
            }
        }

        // 分组数量减1
        count--;


    }


    // 查找某一组的所有元素
    public List<Integer> findAll(int g){

        List<Integer> res = new ArrayList<>();
        for (int i = 0; i < eleAndGroup.length; i++) {
            if (eleAndGroup[i]==g){
                res.add(i);
            }
        }

        return res;

    }


}

10.3.4 并查集应用举例

如果我们并查集存储的每一个整数表示的是一个大型计算机网络中的计算机，则我们就可以通过connected(int p,int q)来检测，该网络中的某两台计算机之间是否连通？如果连通，则他们之间可以通信，如果不连通，则不能通信，此时我们又可以调用union(int p,int q)使得p和q之间连通，这样两台计算机之间就可以通信了。一般像计算机这样网络型的数据，我们要求网络中的每两个数据之间都是相连通的，也就是说，我们需要调用很多次union方法，使得网络中所有数据相连，其实我们很容易可以得出，如果要让网络中的数据都相连，则我们至少要调用N-1次union方法才可以，但由于我们的union方法中使用for循环遍历了所有的元素，所以很明显，我们之前实现的合并算法的时间复杂度是O(N^2)，如果要解决大规模问题，它是不合适的，所以我们需要对算法进行优化。

10.4 UF_Tree算法优化

为了提升union算法的性能，我们需要重新设计fifind方法和union方法的实现，此时我们先需要对我们的之前数据结构中的eleAndGourp数组的含义进行重新设定：

1.我们仍然让eleAndGroup数组的索引作为某个结点的元素；

2.eleAndGroup[i]的值不再是当前结点所在的分组标识，而是该结点的父结点；

10.4.1 UF_Tree API设计

10.4.2 find(int p)查询方法实现

1.判断当前元素p的父结点eleAndGroup[p]是不是自己，如果是自己则证明已经是根结点了；

2.如果当前元素p的父结点不是自己，则让p=eleAndGroup[p]，继续找父结点的父结点,直到找到根结点为止；

10.4.3 union(int p,int q)合并方法实现

找到p元素所在树的根结点
找到q元素所在树的根结点
如果p和q已经在同一个树中，则无需合并；
如果p和q不在同一个分组，则只需要将p元素所在树根结点的父结点设置为q元素的根结点即可；
分组数量-1

10.4.4 代码

package com.ynu.Java版算法.U10_并查集.T2_UF_Tree算法优化;

public class UF_Tree {


    // 记录并查集中数据的分组个数
    private int count;

    // 记录元素和组别
    private int[] eleAndGroup;

    // 初始化元素个数为N的并查集
    public UF_Tree(int N) {

        eleAndGroup = new int[N];
        count = N;
        for (int i = 0; i < eleAndGroup.length; i++) {
            eleAndGroup[i] = i;
        }

    }

    // 获取组的个数
    public int getCount(){
        return count;
    }

    // 判断两个元素是否属于用一个组
    public boolean connected(int p,int q){
        return find(p)==find(q);
    }

    // 查找元素属于的组
    public int find(int p){

        while (true){
            if (p==eleAndGroup[p]){
                return p;
            }
            p = eleAndGroup[p];
        }

    }

    // 合并元素所在的组  把p合并到q所在的组中去
    public void union(int p,int q){

        if (connected(p,q)){
            System.out.println("本来就在一组无需合并");
            return;
        }

        int pRoot = find(p);  // 找到p所在的组
        int qRoot = find(q);  // 找到q所在的组

        eleAndGroup[pRoot] = qRoot;  // 把p所在的组合并到q所在的组
        count--;   // 组数减一

    }



}

10.4.5 优化后的性能分析

我们优化后的算法union，如果要把并查集中所有的数据连通，仍然至少要调用N-1次union方法，但是，我们发现union方法中已经没有了for循环，所以union算法的时间复杂度由O(N^2)变为了O(N)。但是这个算法仍然有问题，因为我们之前不仅修改了union算法，还修改了find算法。我们修改前的find算法的时间复杂度在任何情况下都为O(1)，但修改后的find算法在最坏情况下是O(N)：

10.5 路径压缩

UF_Tree中最坏情况下union算法的时间复杂度也较高，其最主要的问题在于最坏情况下，树的深度和数组的大小一样，如果我们能够通过一些算法让合并时，生成的树的深度尽可能的小，就可以优化find方法。之前我们在union算法中，合并树的时候将任意的一棵树连接到了另外一棵树，这种合并方法是比较暴力的，如果我们把并查集中每一棵树的大小记录下来，然后在每次合并树的时候，把较小的树连接到较大的树上，就可以减小树的深度。

只要我们保证每次合并，都能把小树合并到大树上，就能够压缩合并后新树的路径，这样就能提高find方法的效率。为了完成这个需求，我们需要另外一个数组来记录存储每个根结点对应的树中元素的个数，并且需要一些代码调整数组中的值。

1.5.1 UF_Tree_Weighted API设计

1.5.2 代码

package com.ynu.Java版算法.U10_并查集.T3_路径压缩;

public class UF_Tree_Weighted {
    //记录结点元素和该元素所的父结点
    private int[] eleAndGroup;
    //存储每个根结点对应的树中元素的个数
    private int[] sz;
    //记录并查集中数据的分组个数
    private int count;

    //初始化并查集
    public UF_Tree_Weighted(int N) {
        //初始情况下，每个元素都在一个独立的分组中，所以，初始情况下，并查集中的数据默认分为N个组
        this.count = N;
        //初始化数组
        eleAndGroup = new int[N];
        sz = new int[N];
        //把eleAndGroup数组的索引看做是每个结点存储的元素，把eleAndGroup数组每个索引处的值看做是该结点的父结点，那么初始化情况下，i索引处存储的值就是i
        for (int i = 0; i < N; i++) {
            eleAndGroup[i] = i;
        }
        //把sz数组中所有的元素初始化为1，默认情况下，每个结点都是一个独立的树，每个树中只有一个元素
        for (int i = 0; i < sz.length; i++) {
            sz[i] = 1;
        }
    }

    //获取当前并查集中的数据有多少个分组
    public int count() {
        return count;
    }

    //元素p所在分组的标识符
    public int find(int p) {
        while (true) {
            //判断当前元素p的父结点eleAndGroup[p]是不是自己，如果是自己则证明已经是根结点了；
            if (p == eleAndGroup[p]) {
                return p;
            }
            //如果当前元素p的父结点不是自己，则让p=eleAndGroup[p]，继续找父结点的父结点,直到找到根结点为止；
            p = eleAndGroup[p];
        }
    }

    //判断并查集中元素p和元素q是否在同一分组中
    public boolean connected(int p, int q) {
        return find(p) == find(q);
    }

    //把p元素所在分组和q元素所在分组合并
    public void union(int p, int q) {
        //找到p元素所在树的根结点
        int pRoot = find(p);
        //找到q元素所在树的根结点
        int qRoot = find(q);
        //如果p和q已经在同一个树中，则无需合并；
        if (pRoot == qRoot) {
            return;
        }
        //如果p和q不在同一个分组，比较p所在树的元素个数和q所在树的元素个数,把较小的树合并到较大的树上
        if (sz[pRoot] <= sz[qRoot]) {
            eleAndGroup[pRoot] = qRoot;
            //重新调整较大树的元素个数
            sz[qRoot] += sz[pRoot];
        } else {
            eleAndGroup[qRoot] = pRoot;
            sz[pRoot] += sz[qRoot];
        }
        //分组数量-1
        count--;
    }

}



package com.ynu.Java版算法.U10_并查集.T3_路径压缩;

public class Main {
    public static void main(String[] args) {

        UF_Tree_Weighted uf_tree_weighted = new UF_Tree_Weighted(10);

        // 合并1,2到3
        uf_tree_weighted.union(1,3);
        uf_tree_weighted.union(2,3);

        System.out.println(uf_tree_weighted.count());

        //合并 4 ,7到8
        uf_tree_weighted.union(4,8);
        uf_tree_weighted.union(7,8);

        System.out.println(uf_tree_weighted.count());

        //合并3组到9
        uf_tree_weighted.union(3,9);

        //查找1所在的组  应该是3
        System.out.println(uf_tree_weighted.find(1));
    }
}

10.6 畅通工程

某省调查城镇交通状况，得到现有城镇道路统计表，表中列出了每条道路直接连通的城镇。省政府“畅通工程”的目标是使全省任何两个城镇间都可以实现交通（但不一定有直接的道路相连，只要互相间接通过道路可达即可）。问最少还需要建设多少条道路？

总共有20个城市，目前已经修改好了7条道路，问还需要修建多少条道路，才能让这20个城市之间全部相通？

解题思路：

1.创建一个并查集UF_Tree_Weighted(20);

2.分别调用union(0,1),union(6,9),union(3,8),union(5,11),union(2,12),union(6,10),union(4,8)，表示已经修建好的

道路把对应的城市连接起来；

3.如果城市全部连接起来，那么并查集中剩余的分组数目为1，所有的城市都在一个树中，所以，只需要获取当前

并查集中剩余的数目，减去1，就是还需要修建的道路数目；每修一条路，就少一个分组。

代码：

package com.ynu.Java版算法.U10_并查集.T4_畅通工程;

import com.ynu.Java版算法.U10_并查集.T3_路径压缩.UF_Tree_Weighted;

import java.util.ArrayList;
import java.util.List;

public class Main {

    public static void main(String[] args) {

        List<List<Integer>> paths = new ArrayList<>();
        List<Integer> path1 = new ArrayList<>();
        path1.add(0);
        path1.add(2);
        List<Integer> path2 = new ArrayList<>();
        path2.add(6);
        path2.add(9);
        List<Integer> path3 = new ArrayList<>();
        path3.add(3);
        path3.add(8);
        List<Integer> path4 = new ArrayList<>();
        path4.add(5);
        path4.add(11);
        List<Integer> path5 = new ArrayList<>();
        path5.add(2);
        path5.add(12);
        List<Integer> path6 = new ArrayList<>();
        path6.add(6);
        path6.add(10);
        List<Integer> path7 = new ArrayList<>();
        path7.add(4);
        path7.add(8);
        paths.add(path1);
        paths.add(path2);
        paths.add(path3);
        paths.add(path4);
        paths.add(path5);
        paths.add(path6);
        paths.add(path7);
        UF_Tree_Weighted uf = new UF_Tree_Weighted(20);

        for (List<Integer> path : paths) {
            Integer v = path.get(0);
            Integer w = path.get(1);
            uf.union(v,w);
        }

        System.out.println("还需要"+ (uf.count() - 1 )+"条道路");

    }
}