Prim算法:最小生成树

时间:2024-08-10 23:33:14
 #define _CRT_SECURE_NO_WARNINGS
/*
7 10
0 1 5
0 2 2
1 2 4
1 3 2
2 3 6
2 4 10
3 5 1
4 5 3
4 6 5
5 6 9
*/
#include <iostream>
#include <vector>
#include <algorithm>
using namespace std; const int maxn = + ;
const int INF = ;
int cost[maxn][maxn]; //cost[u][v]表示边e=(u,v)的权值(不存在时候设为INF)
int mincost[maxn]; //从集合X出发的边到每个顶点的最小权值
bool used[maxn]; //顶点i是否包含在集合X中
int V, E; //顶点数
void init();
void input();
int prim(); int prim()
{
for (int i = ; i < V; ++i) {
mincost[i] = INF;
used[i] = false;
}
mincost[] = ;
int res = ; while (true) {
int v = -;
//从不属于X的顶点中选取从X到其权值最小的顶点
for (int u = ; u < V; u++) {
if (!used[u] && (v == - || mincost[u] < mincost[v]))
v = u;
} if (v == -) break;
used[v] = true; //把顶点v加入到X
res += mincost[v]; //把长度加到结果里 for (int u = ; u < V; u++) {
//下一次到新节点u最短 = min(已知中最短(或者还未知),尚未探索的最短).
mincost[u] = min(mincost[u], cost[v][u]);
}
}
return res;
} void init()
{
for (int i = ; i < V; i++) {
for (int j = ; j < V; j++) {
if (i == j) {
cost[i][j] = ;
}
else {
cost[i][j] = INF;
}
} }
} void input()
{
int s, t, ct;
for (int i = ; i < E; i++) {
cin >> s >> t >> ct;
cost[s][t] = cost[t][s] = ct;
}
} int main()
{
cin >> V >> E;
init();
input();
int res = prim();
cout << res << endl;
return ;
}