简单的最小生成树——Kruskal（1）

简单的最小生成树——Kruskal（1）

题目来源：洛谷 P3366 【模板】最小生成树

原题

共十个测试点

题解

最小生成树（MST）两大代码：Kruskal（加边），Prim（加点）

// kruskal算法模板
#include <bits/stdc++.h>
using namespace std;

const int maxn = 2e5 + 5;
const int maxnp = 5e3 + 5;
#define inf 0x3f3f3f3f
// 存图结构体
struct Node
{
    int u, v, w;
    // 排序
    bool operator<(const Node &b) const
    {
        return w < b.w;
    }
} e[maxn];
int n, m, ans = 0, cnt = 0;
// 并查集
int fa[maxnp];
// 并查集的初始化
void init()
{
    for (int i = 1; i <= n; ++i)
        fa[i] = i;
}
// 并查集判断环
// 如果 find(x) == find(y) 则再加入一边形成环
int find(int x)
{
    if (fa[x] == x)
        return x;
    else
    {
        fa[x] = find(fa[x]);
        return fa[x];
    }
}
// 维护并查集，连接两个节点
void Union(int x, int y)
{
    fa[find(x)] = find(y);
}
// kruskal算法
// 按照权值加边，并用并查集维护连通性
void kruskal()
{
    // 按照边权排序
    sort(e + 1, e + m + 1);
    // 遍历图中边
    for (int i = 1; i <= m; ++i)
    {
        int u = e[i].u, v = e[i].v;
        int fau = find(u), fav = find(v);
        // 如果加边不形成环，则加边
        if (fau != fav)
        {
            ++cnt;
            Union(fau, fav);
            ans += e[i].w;
        }
    }
}
// 输出函数
void out()
{
    if (cnt != n - 1)
        printf("orz");
    else
        printf("%d", ans);
}

int main()
{
    scanf("%d%d", &n, &m);
    init();
    for (int i = 1; i <= m; ++i)
    {
        scanf("%d%d%d", &e[i].u, &e[i].v, &e[i].w);
    }
    kruskal();
    out();
    return 0;
}