最优灌溉_最小生成树Kruskal

问题描述

　　雷雷承包了很多片麦田，为了灌溉这些麦田，雷雷在第一个麦田挖了一口很深的水井，所有的麦田都从这口井来引水灌溉。
　　为了灌溉，雷雷需要建立一些水渠，以连接水井和麦田，雷雷也可以利用部分麦田作为“中转站”，利用水渠连接不同的麦田，这样只要一片麦田能被灌溉，则与其连接的麦田也能被灌溉。
　　现在雷雷知道哪些麦田之间可以建设水渠和建设每个水渠所需要的费用（注意不是所有麦田之间都可以建立水渠）。请问灌溉所有麦田最少需要多少费用来修建水渠。

输入格式

　　输入的第一行包含两个正整数n, m，分别表示麦田的片数和雷雷可以建立的水渠的数量。麦田使用1, 2, 3, ……依次标号。
　　接下来m行，每行包含三个整数a_i, b_i, c_i，表示第a_i片麦田与第b_i片麦田之间可以建立一条水渠，所需要的费用为c_i。

输出格式

　　输出一行，包含一个整数，表示灌溉所有麦田所需要的最小费用。

样例输入

4 4
1 2 1
2 3 4
2 4 2
3 4 3

样例输出

6

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <vector>

using namespace std;

const int maxn = 1000 + 5;
const int maxe = 100000 + 5;
const int inf = 0x3f3f3f3f;

struct Edge{
    int u, v, cost;
    Edge(){}
    Edge(int u, int v, int c) :u(u), v(v), cost(c){}
    //    bool operator < (const Edge& rhs){
    //        return cost < rhs.cost;
    //    }
};

int  n, m;
int  p[maxn];
Edge e[maxe];

int find(int x){ return x == p[x] ? x : p[x] = find(p[x]); }
int cmp(const Edge& e1, const Edge& e2){ return e1.cost < e2.cost; }

void Kruskal(){
    for (int i = 0; i <= n; ++i) p[i] = i;
    sort(e, e + m, cmp);

    int ans = 0;
    for (int i = 0; i < m; ++i){
        int u = find(e[i].u), v = find(e[i].v), w = e[i].cost;
        if (u != v){
            p[u] = v;
            ans += w;
        }
    }

    cout << ans << endl;
}


int main(){
    cin >> n >> m;
    for (int i = 0; i < m; ++i){
        int u, v, w;
        cin >> u >> v >> w;
        e[i] = Edge(u, v, w);
    }
    Kruskal();
    return 0;
}