最小生成树：Kruskal算法

NC159 最小生成树

连通图中的每一棵生成树，都是原图的一个极大无环子图。也就是说，从其中删去任何一条边，生成树就不再连通；反之，在其中引入任意一条新边，都会形成（恰好）一个回路。

对于一个带权图（即网络）而言，不同的生成树所对应的总权值也不尽相同。那么如何找出总权值最小的生成树呢？

构造最小生成树的方法有多种，典型的有两种：Kruskal算法和Prim算法。

Kruskal算法和Prim算法的原理：最小生成树(Kruskal(克鲁斯卡尔)和Prim(普里姆))算法动画演示

/*
 * Kruskal 最小生成树算法
 */

#include<vector>
#include<iostream>
#include<algorithm>
using namespace std;

struct Edge {
    int u, v, w;
    bool operator<(const Edge& a)const { return w < a.w; }
};

class Solution {
public:
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     *
     * 返回最小的花费代价使得这n户人家连接起来
     * @param n int n户人家的村庄
     * @param m int m条路
     * @param cost intvector<vector<>> 一维3个参数，表示连接1个村庄到另外1个村庄的花费的代价
     * @return int
     */
    int miniSpanningTree(int n, int m, vector<vector<int>>& cost) {
        // write code here
        vector<int> parent(n + 1, -1);
        vector<Edge> edges;
        for (auto tmp : cost) {
            Edge e;
            e.u = tmp[0];
            e.v = tmp[1];
            e.w = tmp[2];
            edges.push_back(e);
        }
        int res = kruskal(edges, parent);
    }

    int findRoot(vector<int>& parent, int x) {
        int root = x;
        while (parent[root] != -1) {
            root = parent[root];
        }
        return root;
    }

    void unionNode(vector<int>& parent, int x, int y) {
        int x_root = findRoot(parent, x);
        int y_root = findRoot(parent, y);
        parent[x_root] = y_root;
    }

    int kruskal(vector<Edge>& edges, vector<int>& parent) {
        sort(edges.begin(), edges.end());
        int ans = 0;
        while (!edges.empty()) {
            if (findRoot(parent, edges[0].u) != findRoot(parent, edges[0].v)) {
                ans += edges[0].w;
                unionNode(parent, edges[0].u, edges[0].v);
            }
            edges.erase(edges.begin());
        }
        return ans;
    }
};

参考资料
[1]《数据结构（用面向对象方法与C++语言描述）》殷人昆
[2]最小生成树(Kruskal(克鲁斯卡尔)和Prim(普里姆))算法动画演示