最小生成树prim算法示例程序

n，m表示顶点数和边数。

lowcost[]存放顶点集合T'内各顶点到顶点集合T内各顶点权值最小的边的权。

nearvex[]记录顶点集合T'内各顶点距离顶点集合T内那个顶点最近。nearvex[i]为-1时，表示顶点i属于集合T.

#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
#include<string>
#include<cmath>
#include<map>
using namespace std;

const int INF=1000000;
const int maxn=21;
int n,m;
int edge[maxn][maxn];
int lowcost[maxn];
int nearvex[maxn];

void prim(int u)
{
int sumw=0;
for(int i=1;i<=n;i++)//初始化两个数组
{
lowcost[i]=edge[u][i];
nearvex[i]=u;
}
nearvex[u]=-1;
for(int i=1;i<n;i++)//把剩余n-1个节点扩展进来
{
int min=INF;
int v=-1;
for(int j=1;j<=n;j++)
if(nearvex[j]!=-1 && lowcost[j]<min)
{
v=j;
min=lowcost[j];
}
if(v!=-1)//没有找到权值最小的边
{
printf("%d %d %d\n",nearvex[v],v,lowcost[v]);
nearvex[v]=-1;
sumw+=lowcost[v];
for(int j=1;j<=n;j++)
if(nearvex[j]!=-1 && edge[v][j]<lowcost[j])
{
lowcost[j]=edge[v][j];
nearvex[j]=v;
}
}
}
printf("weight of MST is %d\n",sumw);
}
int main()
{
int u,v,w;
while(~scanf("%d%d",&n,&m))
{
memset(edge,0,sizeof(edge));
for(int i=1;i<=m;i++)
{
scanf("%d%d%d",&u,&v,&w);
edge[u][v]=edge[v][u]=w;
}
for(int i=1;i<=n;i++)//对邻接矩阵中其他元素赋值
for(int j=1;j<=n;j++)
{
if(i!=j && edge[i][j]==0)
edge[i][j]=INF;
}
prim(1);//从顶点1出发构造最小生成树
}
return 0;
}