算法板子
算法板子

#include <bits/stdc++.h>
using namespace std;
const int N = 505;
int a[N][N], d[N], v[N];
int n, m, ans = 0;
void prim() {
    memset(d, 0x3f, sizeof d);
    memset(v, 0, sizeof v);
    d[1] = 0;
    for (int i = 1; i < n; ++i) {
        int x = 0;
        for (int j = 1; j <= n; ++j) {
            if (!v[j] && (x == 0 || d[j] < d[x])) {
                x = j;
            }
        }
        v[x] = 1;
        for (int j = 1; j <= n; ++j) {
            if (!v[j]) d[j] = min(d[j], a[x][j]);
        }
    }
};
int main() {
    cin >> n >> m; 
    memset(a, 0x3f, sizeof a);
    for (int i = 1; i <= n; ++i) a[i][i] = 0;
    for (int i = 1;i <= m; ++i) {
        int u, v, w;
        scanf("%d%d%d", &u, &v, &w);
        a[u][v] = a[v][u] = min(a[v][u], w);
    }

    prim();
    for (int i = 2; i <= n; ++i) {
        if (d[i] == 0x3f3f3f3f) {
            puts("impossible");
            return 0;
        }
        ans += d[i];
    }
    printf("%d\n", ans);
    return 0;
}
本文为原创文章,欢迎分享,勿全文转载,如果内容你实在喜欢,可以加入收藏夹,说不定哪天故事又继续更新了呢。
本文地址:https://qoogle.top/minimum-spanning-tree-by-prim/
最后修改日期:2020年8月17日

作者

留言

撰写回覆或留言

发布留言必须填写的电子邮件地址不会公开。