#include <bits/stdc++.h>
using namespace std;
const int INF = 0x3f3f3f3f;
const int maxn = 1005;
bool vis[maxn];
int cost[maxn][maxn], lowc[maxn];
vector<int> demo;
int prim(int cost[][maxn], int n)
{
int ans = 0;
memset(vis, false, sizeof(vis));
vis[0] = true;
for(int i = 1; i < n; ++i)
lowc[i] = cost[0][i];
for(int i = 1; i < n; ++i)
{
int minc = INF;
int p = -1;
for(int j = 0; j < n; ++j)
{
if(!vis[j] && minc > lowc[j])
{
minc = lowc[j];
p = j;
}
}
if(minc == INF)
return -1;
ans += minc;
demo.push_back(minc);
vis[p] = true;
for(int j = 0; j < n; ++j)
{
if(!vis[j] && lowc[j] > cost[p][j])
lowc[j] = cost[p][j];
}
}
return ans;
}
int main()
{
int n, m, u, v, w; //n是顶点数,m是边数
scanf("%d %d", &m, &n);
memset(cost, INF, sizeof(cost));
//memset(vis, 0 ,sizeof(vis));
for(int i = 0; i < m; ++i)
{
for(int j=0;j<m;j++)
{
cin>>cost[i][j];
cost[j][i]=cost[i][j];
}
}
//也可以通过边的数量输入数据 x y之间有 z长度的边,则连cost[x][y]=z
prim(cost, m);
for(int i=0;i<m-1;i++) cout<<demo[i]<<" ";
return 0;
}
图论--最小生成树--Prim算法(带边输出)模板
转载本文章为转载内容,我们尊重原作者对文章享有的著作权。如有内容错误或侵权问题,欢迎原作者联系我们进行内容更正或删除文章。
提问和评论都可以,用心的回复会被更多人看到
评论
发布评论
相关文章
-
prim算法模板 (最小生成树)
求无向图的最小生成树的主要算法有:prim算法和Kruskal算法
prim算法模板 最小生成树 #define prim算法 -
java 图论三 带权图的最小生成树 Prim算法
ime算法实现 prim算法的总思路 /** * 生成最小生成树 * 将顶点放到树集合中,重复
java 图论三 带权图 最小生成树 Prim算法