小记:熟练了下prim,其实是照着dijsktra来写的,就一个判断条件的不同而已
思路:prim算法
代码:
#include <iostream>
#include <stdio.h>
#include <string.h>
#include <math.h>
#include <stdlib.h>
#include <map>
#include <set>
#include <vector>
#include <stack>
#include <queue>
#include <algorithm>
#include <string>
using namespace std;
#define mst(a,b) memset(a,b,sizeof(a))
#define REP(a,b,c) for(int a = b; a < c; ++a)
#define eps 10e-8
const int MAX_ = 155;
const int N = 100010;
const int INF = 0x7fffffff;
int n;
struct node{
int s, t;
};
int g[MAX_][MAX_];
int d[MAX_];
bool vis[MAX_];
int m, cnt;
int prim(int start)
{
REP(i, 0, n){
d[i] = INF;
vis[i] = 0;
}
int sum = 0;
d[start] = 0;
REP(i, 0, n){
int mmin = INF, k;
REP(j, 0, n){
if(!vis[j] && d[j] < mmin){
mmin = d[j];
k = j;
}
}
vis[k] = 1;
sum += mmin;
REP(j, 0, n){
if(!vis[j] && d[j] > g[k][j]){
d[j] = g[k][j];
}
}
}
return sum;
}
int main(){
int T, ss, tt;
while(~scanf("%d", &n)){
REP(i, 0, n)REP(j, 0, n){
scanf("%d", &g[i][j]);
}
int ans = prim(0);
printf("%d\n", ans);
}
return 0;
}
本文通过一个简洁的C++实现示例介绍了Prim最小生成树算法的基本原理和应用。详细讲解了如何初始化距离数组和标记已访问顶点的过程,并展示了如何通过迭代找到最小权重边,最终计算出最小生成树的总权重。

被折叠的 条评论
为什么被折叠?



