AcWing 4292. 网络连接 (Prim求最小生成树)

给定一个 nn 个点 mm 条边的无向连通图。

图中点的编号为 1∼n1∼n。

图中可能存在重边。

求该图的最小生成树的树边权重之和。

输入格式

输入包含多组测试数据。

每组数据第一行包含两个整数 n,mn,m。

接下来 mm 行，每行包含三个整数 a,b,ca,b,c，表示点 aa 和点 bb 之间存在一条边长度为 cc。

每组数据输入完毕后，还会输入一个空行。

当输入一行为单个 00 时，表示输入结束。

输出格式

每组数据输出一行答案，一个整数，表示最小生成树的树边权重之和。

数据范围

每个输入最多包含 100100 组数据。
1≤n≤501≤n≤50,
0≤m≤15000≤m≤1500,
1≤a,b≤n1≤a,b≤n,
1≤c≤1001≤c≤100。

输入样例：

1 0

2 3
1 2 37
2 1 17
1 2 68

3 7
1 2 19
2 3 11
3 1 7
1 3 5
2 3 89
3 1 91
1 2 32

5 7
1 2 5
2 3 7
2 4 8
4 5 11
3 5 10
1 5 6
4 2 12

0

输出样例：

0
17
16
26

代码如下：

#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
int n, m;
bool st[55];
int dist[55];
int g[55][55];

const int INF = 0x3f3f3f3f;

void prim() {
    memset(dist, 0x3f, sizeof(dist));

    for (int i = 0; i < 55; i ++)
        st[i] = false;
     int res = 0;

    for (int i = 0; i < n; i ++) {
        int t = -1;
        for (int j = 1; j <= n; j ++) {
            if(!st[j] && (t == -1 || dist[t] > dist[j]))
                t = j;
        }
        if(i){
            res += dist[t];
        }
        st[t] = true;
        for (int j = 1; j <= n; j ++) {
            dist[j] = min(dist[j], g[t][j]);
        }
    }
    cout << res << "\n";
}

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

    while(cin >> n >> m) {
        if(n == 0)  return 0;
//需要注意的是图的初始化需要初始化为INF，每个点到自身的距离设置为0
        memset(g, 0x3f, sizeof(g)); 
        for (int i = 1; i <= n; i ++) {
            for (int j = 1; j <= n; j ++) {
                if(i == j)
                    g[i][j] = 0;
            }
        }

            for (int i = 0; i < m; i++)
            {
                int a, b, c;
                cin >> a >> b >> c;
                if ((g[a][b] > c || g[a][b] == 0) || (g[b][a] > c || g[b][a] == 0))
                {
                    g[a][b] = c; //有重边需要判断后输入
                    g[b][a] = c;
                }
            }
        prim();
    }

    return 0;
}