AcWing 858. Prim算法求最小生成树
原题链接
简单
作者:
C___
,
2021-12-05 21:20:45
,
所有人可见
,
阅读 167
#include<iostream>
#include<cstdio>
#include<cstring>
#include<vector>
#include<algorithm>
#include<queue>
using namespace std;
using LL = long long;
const int N = 5 * 1e2 + 10, INF = 0x3f3f3f3f;
int d[N][N], dist[N], n, m;
bool st[N];
void Prim(){
memset(dist, 0x3f, sizeof dist);
int res = 0;
// 生成 n 个节点的树
for(int i = 0; i < n; i++){
int t = -1;
// 找到当前距离树最短距离的点
for(int j = 1; j <= n; j++)
if(!st[j] && (t == -1 || dist[j] < dist[t])) t = j;
// 标记已经入树
st[t] = true;
// 如果当前不是第一个点并且找不到能够更新到树中的点则无法生成最小生成树
if(i && dist[t] == INF) {printf("impossible\n"); return;}
// 如果不是第一个节点则记录最小生成树的最短路径
if(i) res += dist[t];
// 更新每个节点到树的最短距离
for(int j = 1; j <= n; j++) dist[j] = min(dist[j], d[t][j]);
}
printf("%d\n", res);
}
int main(void){
scanf("%d%d", &n, &m);
// 默认任意两点之间的距离无限大
memset(d, 0x3f, sizeof d);
for(int i = 1; i <= m; i++){
int a, b, c; scanf("%d%d%d", &a, &b, &c);
// 取两点之间的最短路径
d[a][b] = d[b][a] = min(d[a][b], c);
}
Prim();
return 0;
}