AcWing 3210. 最优灌溉 (kruskal 求最小生成树 模板题)

雷雷承包了很多片麦田，为了灌溉这些麦田，雷雷在第一个麦田挖了一口很深的水井，所有的麦田都从这口井来引水灌溉。

为了灌溉，雷雷需要建立一些水渠，以连接水井和麦田，雷雷也可以利用部分麦田作为“中转站”，利用水渠连接不同的麦田，这样只要一片麦田能被灌溉，则与其连接的麦田也能被灌溉。

现在雷雷知道哪些麦田之间可以建设水渠和建设每个水渠所需要的费用（注意不是所有麦田之间都可以建立水渠）。请问灌溉所有麦田最少需要多少费用来修建水渠。

输入格式

输入的第一行包含两个正整数 n,mn,m，分别表示麦田的片数和雷雷可以建立的水渠的数量。麦田使用 1,2,3,……1,2,3,…… 依次标号。

接下来 mm 行，每行包含三个整数 ai,bi,ciai,bi,ci，表示第 aiai 片麦田与第 bibi 片麦田之间可以建立一条水渠，所需要的费用为 cici。

输出格式

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

数据范围

前 20%20% 的评测用例满足：n≤5n≤5。
前 40%40% 的评测用例满足：n≤20n≤20。
前 60%60% 的评测用例满足：n≤100n≤100。
所有评测用例都满足：1≤n≤1000，1≤m≤100,000，0≤ci≤10,0001≤n≤1000，1≤m≤100,000，0≤ci≤10,000。
保证一定可以灌溉到所有麦田，并且无重边和自环（补充这个条件是为了和官方数据保持一致）。
注意，关于 cici 的取值范围，官网标注的是 ci≥1ci≥1，但是经过实际测试，官方数据中包含 ci=0ci=0 的数据，因此在这里予以修正，并添加相应数据。

输入样例：

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

输出样例：

6

样例解释

建立以下三条水渠：麦田 11 与麦田 22、麦田 22 与麦田 44、麦田 44 与麦田 33。

代码如下：

#include<iostream>
#include<algorithm>

using namespace std;
int n, m, res = 0;
const int maxn = 1e5 + 10;
int a[maxn];
struct Edges{
    int a, b, cost;
} edges[maxn];

void init() {
    for (int i = 1; i <= n; i ++)
        a[i] = i;
}

bool cmp(Edges a, Edges b) {
    return a.cost < b.cost;
}

int find(int x) {
    if(x != a[x])
        a[x] = find(a[x]);
    return a[x];
}

void kruskal() {
    sort(edges + 1, edges + 1 + m, cmp);
    init();
    for (int i = 1; i <= m; i ++) {
        int x = edges[i].a, y = edges[i].b, z = edges[i].cost;
        x = find(x);
        y = find(y);
        if(x != y) {
            a[x] = y;
            res += z;
        }
    }
    cout << res;
}

int main() {
    ios::sync_with_stdio(false);
    cout.tie(nullptr);

    cin >> n >> m;

    for (int i = 1; i <= m; i ++) {
        int x, y, z;
        cin >> x >> y >> z;
        edges[i].a = x;
        edges[i].b = y;
        edges[i].cost = z;
    }

    kruskal();
    return 0;
}