题解：AcWing 859 Kruskal算法求最小生成树

【题目来源】

AcWing：859. Kruskal算法求最小生成树 - AcWing题库

【题目描述】

给定一个 \(n\) 个点 \(m\) 条边的无向图，图中可能存在重边和自环，边权可能为负数。

求最小生成树的树边权重之和，如果最小生成树不存在则输出 impossible。

给定一张边带权的无向图 \(G=(V,E)\)，其中 \(V\) 表示图中点的集合，\(E\) 表示图中边的集合，\(n=|V|\)，\(m=|E|\)。

由 \(V\) 中的全部 \(n\) 个顶点和 \(E\) 中 \(n-1\) 条边构成的无向连通子图被称为 \(G\) 的一棵生成树，其中边的权值之和最小的生成树被称为无向图 \(G\) 的最小生成树。

【输入】

第一行包含两个整数 \(n\) 和 \(m\)。

接下来 \(m\) 行，每行包含三个整数 \(u,v,w\)，表示点 \(u\) 和点 \(v\) 之间存在一条权值为 \(w\) 的边。

【输出】

共一行，若存在最小生成树，则输出一个整数，表示最小生成树的树边权重之和，如果最小生成树不存在则输出 impossible。

【输入样例】

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

【输出样例】

6

【解题思路】

【算法标签】

《AcWing 859 Kruskal算法求最小生成树》 #最小生成树# #Kruskal#

【代码详解】

#include <bits/stdc++.h>
using namespace std;
const int N=1e5+5, M=2e5+5, INF=1e9;
int n, m, ans, cnt;  //n点数量，m边数量，ans生成树边权和，cnt生成树边数量
int p[N];  //p[i]:点i的根节点
struct Edge {
    int u, v, cost;  //起点u，终点v顶点编号，cost，边v-u的权值
    bool operator < (const Edge &E) const {  //重载<操作符，按cost从小到大排序
        return cost < E.cost;
    }
}edges[M];

int find(int x)  //并查集查找顶点x的根节点
{
    if (p[x]!=x) p[x]=find(p[x]);
    return p[x];
}
int main()
{
    cin >> n >> m;  //n点数量，m边数量
    for (int i=1; i<=m; i++) {
        int u, v, w;
        cin >> u >> v >> w;
        if (u>v) swap(u, v);
        edges[i] = {u, v, w};
    }
    sort(edges+1, edges+m+1);  //边权值从小到大排序
    for (int i=1; i<=n; i++)
        p[i] = i;  //并查集初始化，根是它本身，p[x]指x的父节点编号

    for (int i=1; i<=m; i++) {  //从小到大查找边
        int s = edges[i].u, e = edges[i].v, w = edges[i].cost;
        s = find(s), e = find(e);  //查找边edges[i]的两个顶点u,v的根节点
        if (s!=e) {
            p[e] = s;  //集合合并，根节点相同
            ans += w;
            cnt++;
        }
    }
    if (cnt<n-1) puts("impossible");  //边数少于顶点数-1
    else cout << ans << endl;
    return 0;
}

【运行结果】

4 5
1 2 1
1 3 2
1 4 3
2 3 2
3 4 4
6