大话数据结构-普里姆算法（Prim）和克鲁斯卡尔算法（Kruskal）-Toy模板网

这篇具有很好参考价值的文章主要介绍了大话数据结构-普里姆算法（Prim）和克鲁斯卡尔算法（Kruskal）。希望对大家有所帮助。如果存在错误或未考虑完全的地方，请大家不吝赐教，您也可以点击"举报违法"按钮提交疑问。

5 最小生成树

构造连通网的最小代价生成树称为最小生成树，即Minimum Cost Spanning Tree，最小生成树通常是基于无向网/有向网构造的。

找连通网的最小生成树，经典的有两种算法，普里姆算法和克鲁斯卡尔算法。

5.1 普里姆（Prim）算法

普里姆算法，即Prim算法，大致实现过程如下：

(1) 新建数组adjVex[n]，初始值均为0；新建数组lowCost[n]，初始值均为infinity；

(2) 从第一个顶点X（下标为0）开始，把它与各顶点连接的权记录下来，放到lowCost数组里面，然后找到权最小的那个顶点Y，得到最小生成树的第一条边(X,Y)，然后把lowCost数组里面Y对应的下标的元素设置为0；

(3) 然后处理顶点Y，把它与除X外的其他各顶点连接的权，与lowCost数组下标相同的权比较，将小的放入到lowCost里面，并把较小的权对应的顶点的下标记录在adjVex数组里面，也即，adjVex[j]要么是Y，要么是除X外的其他顶点；

(4) 找到lowCost数组中权最小的那个（显然不会是X，也不会是Y），得到最小生成树的第二条边(adjVex[j],j)，然后把lowCost[j]设置为0；

(5) 然后按(3)、(4)的规则，处理第j个顶点，直到所有顶点都被连接起来（注意，最小连通树是针对连通网的）；

下面我们会根据各种存储方式进行举例。

5.1.1 邻接矩阵的最小生成树

假设有以下无向网：

prim和克鲁斯卡尔,数据结构,算法,数据结构,图论

我们定义两个数组，一个X={}，Y={V0、v1、v2、v3、v4、v5、v6、v7、v8}，其中X表示已连通的顶点，Y表示未连通的顶点。

先从第一个顶点v0开始，把它加入X，表示已连通，这时，X={v0}，Y={v1、v2、v3、v4、v5、v6、v7、v8}。

接下来，看X中的V0与其他顶点关联时权值情况，发现(v0,v1)的权值最小，因此认为，v0和v1是最小生成树的一个边，此时X={v0、v1}，Y={v2、v3、v4、v5、v6、v7、v8}：

prim和克鲁斯卡尔,数据结构,算法,数据结构,图论

然后，再看X中的所有顶点与Y中的所有顶点的权值，发现边现(V0,v5)之间的权值最小，因此认为(V0,v5)是最小生成树的一条边，此时X={v0、v1、v5}，Y={v2、v3、v4、v6、v7、v8}：

prim和克鲁斯卡尔,数据结构,算法,数据结构,图论

然后，再看X中的所有顶点与Y中的所有顶点的权值，发现边现(V1,v8)之间的权值最小，因此认为(V1,v8)是最小生成树的一条边，此时X={v0、v1、v5、v8}，Y={v2、v3、v4、v6、v7}：

prim和克鲁斯卡尔,数据结构,算法,数据结构,图论

然后，再看X中的所有顶点与Y中的所有顶点的权值，发现边现(V8,v2)之间的权值最小，因此认为(V8,v2)是最小生成树的一条边，此时X={v0、v1、v5、v8、v2}，Y={v3、v4、v6、v7}：

prim和克鲁斯卡尔,数据结构,算法,数据结构,图论

然后，再看X中的所有顶点与Y中的所有顶点的权值，发现边现(V1,v6)之间的权值最小，因此认为(V1,v6)是最小生成树的一条边，此时X={v0、v1、v5、v8、v2、v6}，Y={v3、v4、v7}：

prim和克鲁斯卡尔,数据结构,算法,数据结构,图论

然后，再看X中的所有顶点与Y中的所有顶点的权值，发现边现(V6,v7)之间的权值最小，因此认为(V6,v7)是最小生成树的一条边，此时X={v0、v1、v5、v8、v2、v6、v7}，Y={v3、v4}：

prim和克鲁斯卡尔,数据结构,算法,数据结构,图论

然后，再看X中的所有顶点与Y中的所有顶点的权值，发现边现(V7,v4)之间的权值最小，因此认为(V7,v4)是最小生成树的一条边，此时X={v0、v1、v5、v8、v2、v6、v7、v4}，Y={v3}：

prim和克鲁斯卡尔,数据结构,算法,数据结构,图论

然后，再看X中的所有顶点与Y中的所有顶点的权值，发现边现(V7,v3)之间的权值最小，因此认为(V7,v3)是最小生成树的一条边，此时X={v0、v1、v5、v8、v2、v6、v7、v4、v3}，Y={}：

prim和克鲁斯卡尔,数据结构,算法,数据结构,图论

这时，Y已处理完毕，所有顶点都连起来了，形成了最小生成树。

观察一下，我们在获取最小生成树的边时，第一步是从arc[0][j]取权最小的，取到了(v0,v1)，第二步，是从arc[0][j]和arc[1][j]中取权值最小的，取到了(v0,v5)，第三步，是从arc[0][j]、arc[1][j]和arc[5][j]中取权最小的，取到了(v1,v8)，也即，规则是：

(1) 从arc[0][j]中取最小权对应的边(v0,v1)，此时X={v0}，把v1加入到X中；

(2) 从arc[0][j]、arc[1][j]中取最小权对应的边(v0,v5)，此时X={v0,v1}，把v5加入到X中；

(3) 从arc[0][j]、arc[1][j]、arc[5][j]中取最小权对应的边(v1,v8)，此时X={v0,v1,v5}，把v8加入到X中；

(4) 从arc[0][j]、arc[1][j]、arc[5][j]…arc[x][j]中取最小权对应的边(vi,vk)，然后判断vi和vk是否在X中，不在则加入到X中；

当在arc[0][j]、arc[1][j]、arc[5][j]…arc[x][j]中取最小的权时，我们要比较x个一元数组的值，我们再做一下优化：

(1) 从arc[0][j]中取最小权对应的边(v0,v1)，此时X={v0}，把v1加入到X中；

(2) 从arc[0][j]、arc[1][j]中取最小权对应的边(v0,v5)，此时X={v0,v1}，把v5加入到X中，然后我们把arc[0][j]、arc[1][j]组合一下，取出arc[0][j]、arc[1][j]中较小的权放到lowCost[j]里面，同时使用一个数组adjVex[j]记录lowCost[j]对应的起始顶点下标，同时标注lowCost[0]、lowCost[1]的值为负无穷大，如下：

prim和克鲁斯卡尔,数据结构,算法,数据结构,图论

(3) 这时，只需要从lowCost[j]、arc[5][j]中取最小权对应的边就行了，不用再迭代arc[0][j]和arc[1][j]；

因此，我们定义两个数组，lowCost[n]表示已处理过的顶点跟其他顶点之间的权值最小值列表，其中已处理过的顶点之间的权值设置为负无穷大，adjVex[n]表示最小权值对应的起始顶点，我们重新来看看上述无向网。

第一步，定义lowCost[n]和adjVex[n]，adjVex[n]默认值为0，lowCost[n]默认值为负无穷大：

prim和克鲁斯卡尔,数据结构,算法,数据结构,图论

接下来处理第一个顶点，找到第一个顶点与其他顶点中权值最小的那条边（自身除外），具体做法是，令adjVex[n]的元素均为0，令lowCost[n]的元素值为arc[0][j]，然后找到权值最小的arc[0][x]，取边(adjVex[x],x)，即(v0,v1)：

prim和克鲁斯卡尔,数据结构,算法,数据结构,图论

接下来，要比较的应是arc[0][j]和arc[1][j]中，除arc[0][0]、arc[0][1]、arc[1][0]和arc[1][1]外的其他值，取最小值，因为我们要取的是“其他顶点与顶点v0、v1中权值最小的边”，因此我们把arc[1][j]与lowCost[n]（这时，lowCost即为arc[0][j]）比较，把较小的写入到lowCost中，同时把较小权对应对应的顶点下标写入到adjVex[n]中，如下：

prim和克鲁斯卡尔,数据结构,算法,数据结构,图论

如上可知，arc[1][2]小于lowCost[2]，因此令lowCost[2]=arc[1][2]、adjVex[2]=1，arc[1][6]和arc[1][8]也相似处理。

然后，我们取出其中的最小权，得到下一条最小生成树的边，即(v0,v5)：

prim和克鲁斯卡尔,数据结构,算法,数据结构,图论

接下来，是取得顶点v0、v1、v5与其他顶点的权中的最小值，生成下一条边，整体处理方式与上文类似：

prim和克鲁斯卡尔,数据结构,算法,数据结构,图论

后续操作也类似，直到所有顶点处理完毕，得到最小生成树。

让我们来看有向网的处理，按上述过程处理，得到以下结果：

prim和克鲁斯卡尔,数据结构,算法,数据结构,图论

即：

prim和克鲁斯卡尔,数据结构,算法,数据结构,图论

但注意观察，同样是连通B顶点，弧<B,A>实际上比<C,B>权小，所以最小生成树应该为：

prim和克鲁斯卡尔,数据结构,算法,数据结构,图论

因此，有向树的最小生成树生成过程中，不仅要看“顶点X指向的顶点中权最小的”，还要看“顶点X被指向的顶点中权最小的”。

因此，寻找最低代码的边/弧的逻辑应是“找到该顶点指向的即该顶点被指向的顶点中，代价最小的边或弧”，对于无向网当然这两个概念是一样的，但对于有向网，则要进行双向处理，因此上述有向网的处理逻辑有所不同。

以以上有向网为例，第一步，定义三个数组：lowCost与无向网相同，tailVex和headVex代替adjVex，分别表示箭头的尾巴和头（若为无向网则尾巴和头可以任意），初始化tailVex和headVex为0，lowCost为无穷大：

prim和克鲁斯卡尔,数据结构,算法,数据结构,图论

然后使用arc[0][j]和arc[j][0]与lowCost比较，取小的权，并记录箭头的尾巴和头，可取出最小的权对应的尾巴和头，得到最小生成树的第一条弧<0,3>，然后设置lowCost[0]和lowCost[3]为负无穷大，表示这两个顶点已连通：

prim和克鲁斯卡尔,数据结构,算法,数据结构,图论

然后取arc[3][j]和arc[j][3]，重复上述处理，得到下一条弧<3,5>，然后设置lowCost[3]和lowCost[5]为负无穷大，表示这两个顶点已连通：

prim和克鲁斯卡尔,数据结构,算法,数据结构,图论

然后对arc[5][j]和arc[j][5]做类似处理，一直到所有顶点都连通。

代码实现如下所示：

/**
 * 生成最小生成树
 *
 * @param visitedVal 已访问的顶点被标记的值
 * @return 最小生成树的边或弧列表
 * @author Korbin
 * @date 2023-02-07 15:23:14
 **/
@SuppressWarnings("unchecked")
public List<String> minimumCostSpanningTree(W visitedVal) {

    List<String> result = new ArrayList<>();

    if (type.equals(BusinessConstants.GRAPH_TYPE.UNDIRECTED) ||
            type.equals(BusinessConstants.GRAPH_TYPE.DIRECTED)) {
        // 最小生成树只针对网
        return null;
    }

    // 用于存储“当前检查的顶点X与顶点j的权W(X,j)”及“顶点tailVex[j]到顶点headVex[j]的权W(adj[j],j)”中较优（如果是数值的话就是较小）的那个权
    W[] lowCost = (W[]) Array.newInstance(infinity.getClass(), vertexNum);
    for (int i = 0; i < vertexNum; i++) {
        lowCost[i] = infinity;
    }

    // 用于存储箭头的尾和头
    int[] tailVex = new int[vertexNum];
    int[] headVex = new int[vertexNum];

    // 取权最小的，即第一个顶点与其他顶点之间权最小的顶点
    int index = 0;
    int connectedNum = 0;
    boolean[] visited = new boolean[vertexNum];
    while (connectedNum < vertexNum) {

        // 其他顶点是与lowCost比较，取小的
        W minWeight = null;
        int newIndex = index;
        for (int i = 0; i < vertexNum; i++) {
            if (i != index && !lowCost[i].equals(visitedVal) && arc[index][i].compareTo(lowCost[i]) < 0) {
                lowCost[i] = arc[index][i];
                tailVex[i] = index;
                headVex[i] = i;
            }
            if (type.equals(BusinessConstants.GRAPH_TYPE.DIRECTED_NETWORK)) {
                // 有向网还需要处理指向本顶点的顶点
                if (i != index && !lowCost[i].equals(visitedVal) && arc[i][index].compareTo(lowCost[i]) < 0) {
                    lowCost[i] = arc[i][index];
                    tailVex[i] = i;
                    headVex[i] = index;
                }

            }

            // 取出最小的权
            // 忽略自身
            // 忽略lowCost[j]为已访问过的顶点
            if (i != index && !lowCost[i].equals(visitedVal)) {
                if (null == minWeight || minWeight.compareTo(lowCost[i]) > 0) {
                    minWeight = lowCost[i];
                    newIndex = i;
                }
            }
        }

        index = newIndex;

        // 打印取到的边
        switch (type) {
            case DIRECTED_NETWORK: {
                String val = "<" + tailVex[index] + "," + headVex[index] + ">";
                result.add(val);
                break;
            }
            case UNDIRECTED_NETWORK: {
                String val = "(" + tailVex[index] + "," + headVex[index] + ")";
                result.add(val);
                break;
            }
        }

        // 设置已连通的顶点数
        if (!visited[tailVex[index]]) {
            visited[tailVex[index]] = true;
            connectedNum++;
        }
        // 设置该顶点已处理
        if (!visited[headVex[index]]) {
            visited[headVex[index]] = true;
            connectedNum++;
        }

        // 设置lowCost[index]为已访问
        lowCost[tailVex[index]] = visitedVal;
        lowCost[headVex[index]] = visitedVal;

    }

    return result;

}

5.1.2 邻接表的最小生成树

邻接表和逆邻接表的处理方式也是类似，只是需要找到指向的和被指向的顶点的权进行比较，另外，邻接表和逆邻接表的处理方式也会有差异：

/**
 * 生成最小生成树
 *
 * @param visitedVal 已访问的顶点被标记的值
 * @return 最小生成树的边或弧列表
 * @author Korbin
 * @date 2023-02-07 15:23:14
 **/
@SuppressWarnings("unchecked")
public List<String> minimumCostSpanningTree(W visitedVal) {

    List<String> result = new ArrayList<>();

    if (type.equals(BusinessConstants.GRAPH_TYPE.UNDIRECTED) ||
            type.equals(BusinessConstants.GRAPH_TYPE.DIRECTED)) {
        // 最小生成树只针对网
        return null;
    }

    // 用于存储“当前检查的顶点X与顶点j的权W(X,j)”及“顶点adjVex[j]到顶点j的权W(adj[j],j)”中较优（如果是数值的话就是较小）的那个权
    W[] lowCost = (W[]) Array.newInstance(infinity.getClass(), vertexNum);
    for (int i = 0; i < vertexNum; i++) {
        lowCost[i] = infinity;
    }

    // 用于存储箭头的尾和头
    int[] tailVex = new int[vertexNum];
    int[] headVex = new int[vertexNum];

    boolean[] visited = new boolean[vertexNum];
    int connectedNum = 0;
    int index = 0;

    while (connectedNum < vertexNum) {

        for (int i = 0; i < vertexNum; i++) {
            VertexNode<T, W> vertexNode = vertexes[i];

            EdgeNode<W> edgeNode = vertexNode.getFirstEdge();
            while (null != edgeNode) {

                int refIndex = edgeNode.getIndex();
                W weight = edgeNode.getWeight();

                // 邻接表
                if (i == index && !lowCost[refIndex].equals(visitedVal) &&
                        weight.compareTo(lowCost[refIndex]) < 0) {
                    // 本顶点指向的
                    if (!reverseAdjacency) {
                        tailVex[refIndex] = i;
                        headVex[refIndex] = refIndex;
                        lowCost[refIndex] = weight;
                    } else {
                        tailVex[refIndex] = refIndex;
                        headVex[refIndex] = i;
                        lowCost[refIndex] = weight;
                    }
                } else if (refIndex == index && !lowCost[i].equals(visitedVal) &&
                        weight.compareTo(lowCost[i]) < 0) {
                    if (!reverseAdjacency) {
                        // 指向本顶点的
                        tailVex[i] = i;
                        headVex[i] = refIndex;
                        lowCost[i] = weight;
                    } else {
                        tailVex[i] = refIndex;
                        headVex[i] = i;
                        lowCost[i] = weight;
                    }
                }

                edgeNode = edgeNode.getNext();

            }

        }

        // 取lowCost中最小的那个
        W minWeight = null;
        for (int i = 0; i < vertexNum; i++) {
            // 忽略自身
            // 忽略lowCost[j]为已访问过的顶点
            if (i != index && !lowCost[i].equals(visitedVal)) {

                if (null == minWeight || minWeight.compareTo(lowCost[i]) > 0) {
                    minWeight = lowCost[i];
                    index = i;
                }
            }
        }

        // 打印取到的边
        switch (type) {
            case DIRECTED_NETWORK: {
                String val = "<" + tailVex[index] + "," + headVex[index] + ">";
                result.add(val);
                // 用于测试
                System.out.println(val);
                break;
            }
            case UNDIRECTED_NETWORK: {
                String val = "(" + tailVex[index] + "," + headVex[index] + ")";
                result.add(val);
                // 用于测试
                System.out.println(val);
                break;
            }
        }

        // 设置已连通的顶点数
        if (!visited[tailVex[index]]) {
            visited[tailVex[index]] = true;
            connectedNum++;
        }
        // 设置该顶点已处理
        if (!visited[headVex[index]]) {
            visited[headVex[index]] = true;
            connectedNum++;
        }

        // 设置lowCost[index]为已访问
        lowCost[tailVex[index]] = visitedVal;
        lowCost[headVex[index]] = visitedVal;

        // 用于测试
        StringBuilder builder = new StringBuilder("lowCost is [");
        StringBuilder builder2 = new StringBuilder("tailVex is [");
        StringBuilder builder3 = new StringBuilder("headVex is [");
        for (int i = 0; i < vertexNum; i++) {
            builder.append(lowCost[i]).append(",");
            builder2.append(tailVex[i]).append(",");
            builder3.append(headVex[i]).append(",");
        }
        builder.append("]");
        builder2.append("]");
        System.out.println(builder);
        System.out.println(builder2);
        System.out.println(builder3);

    }

    return result;

}

5.1.3 十字链表的最小生成树

十字链表记录了每个顶点的in和out，因此十字链表的处理较为简单。

/**
 * 生成最小生成树
 *
 * @param visitedVal 已访问的顶点被标记的值
 * @return 最小生成树的边或弧列表
 * @author Korbin
 * @date 2023-02-07 15:23:14
 **/
@SuppressWarnings("unchecked")
public List<String> minimumCostSpanningTree(W visitedVal) {

    List<String> result = new ArrayList<>();

    if (type.equals(BusinessConstants.GRAPH_TYPE.UNDIRECTED) ||
            type.equals(BusinessConstants.GRAPH_TYPE.DIRECTED)) {
        // 最小生成树只针对网
        return null;
    }

    // 用于存储“当前检查的顶点X与顶点j的权W(X,j)”及“顶点adjVex[j]到顶点j的权W(adj[j],j)”中较优（如果是数值的话就是较小）的那个权
    W[] lowCost = (W[]) Array.newInstance(infinity.getClass(), vertexNum);
    for (int i = 0; i < vertexNum; i++) {
        lowCost[i] = infinity;
    }

    // 用于存储箭头的尾和头
    int[] tailVex = new int[vertexNum];
    int[] headVex = new int[vertexNum];

    boolean[] visited = new boolean[vertexNum];
    int connectedNum = 0;
    int index = 0;

    while (connectedNum < vertexNum) {

        AcrossLinkVertexNode<T, W> vertexNode = vertexes[index];

        // 处理指向自己的
        AcrossLinkEdgeNode<W> inEdge = vertexNode.getFirstIn();
        while (null != inEdge) {
            // 自身是head
            int tailIndex = inEdge.getTailIndex();
            W weight = inEdge.getWeight();

            if (!lowCost[tailIndex].equals(visitedVal) && weight.compareTo(lowCost[tailIndex]) < 0) {
                lowCost[tailIndex] = weight;
                tailVex[tailIndex] = tailIndex;
                headVex[tailIndex] = index;
            }
            inEdge = inEdge.getNextTail();
        }

        // 处理指向的
        AcrossLinkEdgeNode<W> outEdge = vertexNode.getFirstOut();

        while (null != outEdge) {

            // 自身是tail
            int headIndex = outEdge.getHeadIndex();
            W weight = outEdge.getWeight();

            if (!lowCost[headIndex].equals(visitedVal) && weight.compareTo(lowCost[headIndex]) < 0) {
                lowCost[headIndex] = weight;
                tailVex[headIndex] = index;
                headVex[headIndex] = headIndex;
            }

            outEdge = outEdge.getNextHead();
        }

        // 取lowCost中最小的那个
        W minWeight = null;
        for (int i = 0; i < vertexNum; i++) {
            // 忽略自身
            // 忽略lowCost[j]为已访问过的顶点
            if (i != index && !lowCost[i].equals(visitedVal)) {

                if (null == minWeight || minWeight.compareTo(lowCost[i]) > 0) {
                    minWeight = lowCost[i];
                    index = i;
                }
            }
        }

        // 打印取到的边
        switch (type) {
            case DIRECTED_NETWORK: {
                String val = "<" + tailVex[index] + "," + headVex[index] + ">";
                result.add(val);
                // 用于测试
                System.out.println(val);
                break;
            }
            case UNDIRECTED_NETWORK: {
                String val = "(" + tailVex[index] + "," + headVex[index] + ")";
                result.add(val);
                // 用于测试
                System.out.println(val);
                break;
            }
        }

        // 设置已连通的顶点数
        if (!visited[tailVex[index]]) {
            visited[tailVex[index]] = true;
            connectedNum++;
        }
        // 设置该顶点已处理
        if (!visited[headVex[index]]) {
            visited[headVex[index]] = true;
            connectedNum++;
        }

        // 设置lowCost[index]为已访问
        lowCost[tailVex[index]] = visitedVal;
        lowCost[headVex[index]] = visitedVal;

        // 用于测试
        StringBuilder builder = new StringBuilder("lowCost is [");
        StringBuilder builder2 = new StringBuilder("tailVex is [");
        StringBuilder builder3 = new StringBuilder("headVex is [");
        for (int i = 0; i < vertexNum; i++) {
            builder.append(lowCost[i]).append(",");
            builder2.append(tailVex[i]).append(",");
            builder3.append(headVex[i]).append(",");
        }
        builder.append("]");
        builder2.append("]");
        System.out.println(builder);
        System.out.println(builder2);
        System.out.println(builder3);

    }

    return result;

}

5.1.4 邻接多重表的最小生成树

邻接多重表是对无向网的优化，因此我们不考虑有向网，而在无向网中，通过iVex、iLink、jVex、jLink，可以直接定位到某顶点关联的所有顶点，因此代码实现如下所示：

/**
 * 生成最小生成树
 *
 * @param visitedVal 已访问的顶点被标记的值
 * @return 最小生成树的边或弧列表
 * @author Korbin
 * @date 2023-02-07 15:23:14
 **/
@SuppressWarnings("unchecked")
public List<String> minimumCostSpanningTree(W visitedVal) {

    List<String> result = new ArrayList<>();

    if (type.equals(BusinessConstants.GRAPH_TYPE.UNDIRECTED) ||
            type.equals(BusinessConstants.GRAPH_TYPE.DIRECTED) ||
            type.equals(BusinessConstants.GRAPH_TYPE.DIRECTED_NETWORK)) {
        // 最小生成树只针对无向网
        return null;
    }

    // 用于存储“当前检查的顶点X与顶点j的权W(X,j)”及“顶点adjVex[j]到顶点j的权W(adj[j],j)”中较优（如果是数值的话就是较小）的那个权
    W[] lowCost = (W[]) Array.newInstance(infinity.getClass(), vertexNum);
    for (int i = 0; i < vertexNum; i++) {
        lowCost[i] = infinity;
    }

    // 用于存储箭头的尾和头
    int[] tailVex = new int[vertexNum];
    int[] headVex = new int[vertexNum];

    boolean[] visited = new boolean[vertexNum];
    int connectedNum = 0;
    int index = 0;

    while (connectedNum < vertexNum) {

        AdjacencyMultiVertexNode<T, W> vertexNode = vertexes[index];

        AdjacencyMultiEdgeNode<W> edgeNode = vertexNode.getFirstEdge();
        boolean firstEdge = true;
        while (null != edgeNode) {

            int refIndex = index;
            int iVex = edgeNode.getIVex();
            int jVex = edgeNode.getJVex();
            W weight = edgeNode.getWeight();

            if (firstEdge) {
                refIndex = jVex;
                edgeNode = edgeNode.getILink();
                firstEdge = false;
            } else {
                if (iVex == index) {
                    refIndex = jVex;
                } else if (jVex == index) {
                    refIndex = iVex;
                }
                edgeNode = edgeNode.getJLink();
            }

            if (!lowCost[refIndex].equals(visitedVal) && weight.compareTo(lowCost[refIndex]) < 0) {
                lowCost[refIndex] = weight;
                tailVex[refIndex] = iVex;
                headVex[refIndex] = jVex;
            }

        }

        // 取lowCost中最小的那个
        W minWeight = null;
        for (int i = 0; i < vertexNum; i++) {
            // 忽略自身
            // 忽略lowCost[j]为已访问过的顶点
            if (i != index && !lowCost[i].equals(visitedVal)) {

                if (null == minWeight || minWeight.compareTo(lowCost[i]) > 0) {
                    minWeight = lowCost[i];
                    index = i;
                }
            }
        }

        // 打印取到的边
        switch (type) {
            case DIRECTED_NETWORK: {
                String val = "<" + tailVex[index] + "," + headVex[index] + ">";
                result.add(val);
                // 用于测试
                System.out.println(val);
                break;
            }
            case UNDIRECTED_NETWORK: {
                String val = "(" + tailVex[index] + "," + headVex[index] + ")";
                result.add(val);
                // 用于测试
                System.out.println(val);
                break;
            }
        }

        // 设置已连通的顶点数
        if (!visited[tailVex[index]]) {
            visited[tailVex[index]] = true;
            connectedNum++;
        }
        // 设置该顶点已处理
        if (!visited[headVex[index]]) {
            visited[headVex[index]] = true;
            connectedNum++;
        }

        // 设置lowCost[index]为已访问
        lowCost[tailVex[index]] = visitedVal;
        lowCost[headVex[index]] = visitedVal;

        // 用于测试
        StringBuilder builder = new StringBuilder("lowCost is [");
        StringBuilder builder2 = new StringBuilder("tailVex is [");
        StringBuilder builder3 = new StringBuilder("headVex is [");
        for (int i = 0; i < vertexNum; i++) {
            builder.append(lowCost[i]).append(",");
            builder2.append(tailVex[i]).append(",");
            builder3.append(headVex[i]).append(",");
        }
        builder.append("]");
        builder2.append("]");
        System.out.println(builder);
        System.out.println(builder2);
        System.out.println(builder3);

    }

    return result;

}

5.5.5 边集数组的最小生成树

边集数组的最小生成树实现代码如下所示：

/**
 * 生成最小生成树
 *
 * @param visitedVal 已访问的顶点被标记的值
 * @return 最小生成树的边或弧列表
 * @author Korbin
 * @date 2023-02-07 15:23:14
 **/
@SuppressWarnings("unchecked")
public List<String> minimumCostSpanningTree(W visitedVal) {

    List<String> result = new ArrayList<>();

    if (type.equals(BusinessConstants.GRAPH_TYPE.UNDIRECTED) ||
            type.equals(BusinessConstants.GRAPH_TYPE.DIRECTED)) {
        // 最小生成树只针对网
        return null;
    }

    // 用于存储“当前检查的顶点X与顶点j的权W(X,j)”及“顶点adjVex[j]到顶点j的权W(adj[j],j)”中较优（如果是数值的话就是较小）的那个权
    W[] lowCost = (W[]) Array.newInstance(infinity.getClass(), vertexNum);
    for (int i = 0; i < vertexNum; i++) {
        lowCost[i] = infinity;
    }

    // 用于存储箭头的尾和头
    int[] tailVex = new int[vertexNum];
    int[] headVex = new int[vertexNum];

    boolean[] visited = new boolean[vertexNum];
    int connectedNum = 0;
    int index = 0;

    while (connectedNum < vertexNum) {

        for (int i = 0; i < edgeNum; i++) {
            EdgeListEdgeNode<W> edgeNode = arc[i];
            int begin = edgeNode.getBegin();
            int end = edgeNode.getEnd();
            W weight = edgeNode.getWeight();

            if (begin == index && !lowCost[end].equals(visitedVal) && weight.compareTo(lowCost[end]) < 0) {
                lowCost[end] = weight;
                tailVex[end] = begin;
                headVex[end] = end;
            } else if (end == index && !lowCost[begin].equals(visitedVal) && weight.compareTo(lowCost[begin]) < 0) {
                lowCost[begin] = weight;
                tailVex[begin] = begin;
                headVex[begin] = end;
            }
        }

        // 取lowCost中最小的那个
        W minWeight = null;
        for (int i = 0; i < vertexNum; i++) {
            // 忽略自身
            // 忽略lowCost[j]为已访问过的顶点
            if (i != index && !lowCost[i].equals(visitedVal)) {

                if (null == minWeight || minWeight.compareTo(lowCost[i]) > 0) {
                    minWeight = lowCost[i];
                    index = i;
                }
            }
        }

        // 打印取到的边
        switch (type) {
            case DIRECTED_NETWORK: {
                String val = "<" + tailVex[index] + "," + headVex[index] + ">";
                result.add(val);
                // 用于测试
                System.out.println(val);
                break;
            }
            case UNDIRECTED_NETWORK: {
                String val = "(" + tailVex[index] + "," + headVex[index] + ")";
                result.add(val);
                // 用于测试
                System.out.println(val);
                break;
            }
        }

        // 设置已连通的顶点数
        if (!visited[tailVex[index]]) {
            visited[tailVex[index]] = true;
            connectedNum++;
        }
        // 设置该顶点已处理
        if (!visited[headVex[index]]) {
            visited[headVex[index]] = true;
            connectedNum++;
        }

        // 设置lowCost[index]为已访问
        lowCost[tailVex[index]] = visitedVal;
        lowCost[headVex[index]] = visitedVal;

        // 用于测试
        StringBuilder builder = new StringBuilder("lowCost is [");
        StringBuilder builder2 = new StringBuilder("tailVex is [");
        StringBuilder builder3 = new StringBuilder("headVex is [");
        for (int i = 0; i < vertexNum; i++) {
            builder.append(lowCost[i]).append(",");
            builder2.append(tailVex[i]).append(",");
            builder3.append(headVex[i]).append(",");
        }
        builder.append("]");
        builder2.append("]");
        System.out.println(builder);
        System.out.println(builder2);
        System.out.println(builder3);

    }

    return result;

}

5.2 克鲁斯卡尔(Kruskal)算法

克鲁斯卡尔的官方描述是“假设N=(V,{E})是连通网，则令最小生成树的初始状态为只有n个顶点而无边的非连通图T=(V,{})，图中每个顶点自成一个连通分量。在E中选择代价最小的边，若该边依附的顶点落在T中不同的连通分量上，则将此边加入到T中，否则舍去此边选择下一条代价最小的边。依此类推，直到T中所有顶点都在同一连通分量上为止”。

克鲁斯卡尔算法针对的是边，因此我们用边集数组来进行克鲁斯卡尔算法的实现描述，以以下边集数组为例：

prim和克鲁斯卡尔,数据结构,算法,数据结构,图论

首先我们定义一个数组connected来存储已连通的分量，长度为顶点长度，默认值均为无穷大，表示所有顶点均为连通：

prim和克鲁斯卡尔,数据结构,算法,数据结构,图论

这里面的规则是：如果connected[i]=j,connected[j]=x,connected[x]= $\infty$ ，则表示顶点i、j、x是连通的。

然后我们把边集数组按权由小到大排序：

prim和克鲁斯卡尔,数据结构,算法,数据结构,图论

迭代边集数组，首先是arc[0]，begin=4，end=7，connected[4]= $\infty$ ，connected[7]= $\infty$ ，这两个顶点未连通，我们把它们连通起来，输出第一个连通分量的第一条边(4,7)，同时令connected[4]=7，表示这两个顶点已连通：

prim和克鲁斯卡尔,数据结构,算法,数据结构,图论

然后是arc[1]，begin=2，end=8，connected[2]= $\infty$ ，connected[8]= $\infty$ ，这两个顶点未连通，我们把它们连通起来，输出第二个连通分量的第一条边(2,8)，同时令connected[2]=8，表示这两个顶点已连通：

prim和克鲁斯卡尔,数据结构,算法,数据结构,图论

后续arc[2]处理方式同上：

prim和克鲁斯卡尔,数据结构,算法,数据结构,图论

然后看arc[3]，begin=0，end=5，注意到，connected[0]=1，表示第0和第1个顶点是已经连通的，第0个顶点已在一个连通分量内，我们继续看这个连通分量里面有哪些顶点：connected[0]=1，connected[1]= $\infty$ ，按上面定的规则，表示这个连通分量里面只有0和1两个顶点；connected[5]= $\infty$ ，表示顶点5没有在任何一个连通分量内，这时，我们令connected[1]=5，表示顶点0、1、5在同一个连通分量内，并输出(0,5)，表示这个边是这个连通分量的一条边：

prim和克鲁斯卡尔,数据结构,算法,数据结构,图论

然后看arc[4]，begin=1，end=8，因connected[1]=5，connected[5]=0，connected[8]= $\infty$ ，表示顶点1和顶点5在同一个连通分量内，但这个连通分量不包含顶点8，因此令connected[5]=8，并输出边(1,8)，将顶点8也加入到这个连通分量内：

prim和克鲁斯卡尔,数据结构,算法,数据结构,图论

接下来arc[5]和arc[6]的处理方式类似：

prim和克鲁斯卡尔,数据结构,算法,数据结构,图论

来看arc[7]，begin=5，end=6，此时，connected[5]=8，connected[8]=6，connected[6]=0，即顶点5、8、6在一个连通分量内，即begin和end都在一个连通分量内，不需要处理，跳过。

arc[8]与arc[7]一致，相关顶点都在同一个连通分量内，不需要处理，跳过。

继续迭代arc，按如上规则处理，能得到最终的最小生成树。

代码实现如下所示：

/**
 * 生成最小生成树，克鲁斯卡尔算法
 *
 * @return 最小生成树的边或弧列表
 * @author Korbin
 * @date 2023-02-13 15:09:58
 **/
public List<String> minimumCostSpanningTree() {
    List<String> result = new ArrayList<>();

    if (type.equals(BusinessConstants.GRAPH_TYPE.UNDIRECTED) ||
            type.equals(BusinessConstants.GRAPH_TYPE.DIRECTED)) {
        // 最小生成树只针对网
        return null;
    }

    // 对边集数组按权从小到大进行排序
    sortArc(0, edgeNum);

    // 定义一个数组，并初始化为infinity，用于表示连通分量
    int[] connected = new int[vertexNum];
    for (int i = 0; i < vertexNum; i++) {
        connected[i] = -1;
    }

    // 迭代边集数组
    for (int i = 0; i < edgeNum; i++) {
        EdgeListEdgeNode<W> edgeNode = arc[i];
        int begin = edgeNode.getBegin();
        int end = edgeNode.getEnd();

        // 寻找begin对应的连通分量
        int beginComponent = begin;
        while (connected[beginComponent] != -1) {
            beginComponent = connected[beginComponent];
        }

        // 寻找end对应的连通分量
        int endComponent = end;
        while (connected[endComponent] != -1) {
            endComponent = connected[endComponent];
        }

        // 如果不在同一个连通分量
        if (beginComponent != endComponent) {
            connected[beginComponent] = end;

            switch (type) {
                case UNDIRECTED_NETWORK: {
                    String val = "(" + begin + "," + end + ")";
                    result.add(val);
                    break;
                }
                case DIRECTED_NETWORK: {
                    String val = "<" + begin + "," + end + ">";
                    result.add(val);
                    break;
                }
            }
        }
    }

    return result;
}

5.3 总结

普里姆（Prim）算法，从一个顶点出发X，取其他所有顶点与X连通的权中最小权对应的那个顶点Y，放入连通分量TE中，然后取其他所有顶点与X和Y连通的权中最小权对应的那个顶点Z，放入连通分量中，直到所有的顶点都在连通分量中时，最小生成树生成。

克鲁斯卡尔（Kruskal）算法，把所有的边放到一个数组中，按权从小到大排序，然后从第一条边开始迭代，判断begin和end是否在同一个连通分量，若不在，则把他们连通起来，直到所有边都处理完毕。

相对来讲，克鲁斯卡算法更适用于边集数组，而普里姆虽然各类存储结构都可以实现，但时间复杂度不同：

存储结构	普里姆算法		克鲁斯卡尔算法
存储结构	适用性	最差时间复杂度	适用性	最差时间复杂度
邻接矩阵	适用	O(n²)	不适用	--
邻接表	适用	O(n³)	不适用	--
十字链表	适用	O(n²)	不适用	--
邻接多重表	适用	O(n²)	不适用	--
边集数组	适用	O(n²)	适用	O(nlog(n))