Prim【MST】

来源：互联网发布：java 每隔1秒执行一次编辑：程序博客网时间：2024/05/17 08:47

#include <iostream>
#include <string.h>
#define Max 10000
#define INF 1000000
using namespace std;
int n,m;
int edge[Max][Max];
int lowcost[Max];
int nearvex[Max];
void prim(int u0)
{
int sumweight = 0;
for(int i=1;i<=n;i++)
{
lowcost[i]=edge[u0][i];
nearvex[i]=u0;
}
nearvex[u0]=-1;
for(int i=1;i<n;i++)
{
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)
{
nearvex[v]=-1;//插入一条边
sumweight+=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;
}
}
}
}
cout<<sumweight<<endl;
}
int main()
{
int u,v,w;
cin>>n>>m;
memset(edge,0,sizeof(edge));
for(int i=1;i<=m;i++)
{
cin>>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;
else if(edge[i][j]==0) edge[i][j]=INF;
}
}
prim(1);
return 0;

}

//prim与边的数目无关，适合稠密图，时间复杂度O(n^2)

//采用邻接矩阵储存

//最近要搞一发图论的完全模板

0 0