题解 | #最小生成树#

#include <vector>
class Solution {
public:
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     *
     * 返回最小的花费代价使得这n户人家连接起来
     * @param n int整型 n户人家的村庄
     * @param m int整型 m条路
     * @param cost int整型vector<vector<>> 一维3个参数，表示连接1个村庄到另外1个村庄的花费的代价
     * @return int整型
     */
    static const int N = 10010;
    int find(vector<int>& parent, int x){
        if(parent[x] != x)
            parent[x] = find(parent, parent[x]);
        return parent[x];
    }
    static bool cmp(vector<int>& x, vector<int>& y){
        return x[2] < y[2];
    }
    int miniSpanningTree(int n, int m, vector<vector<int> >& cost) {
        vector<int> p(n + 1);
        for(int i = 0; i < n; i ++) p[i] = i;
        sort(cost.begin(), cost.end(), cmp);
        int res = 0;
        for(int i = 0; i < m; i ++){
            int a = cost[i][0];
            int b = cost[i][1];
            int w = cost[i][2];
            a = find(p,a), b = find(p, b);
            if(a != b){
                p[a] = b;
                res += w;
            }
        }
        return res;
    }

};