/*prim模板 水过*/
#include <stdio.h>
#include <string.h>
#define MAX 105
#define MAXCOST 0x7fffffff
int sum,n,graph[MAX][MAX];
int adjvex[MAX];//记录顶点是否被访问
int lowcost[MAX];//只管记录的是已是最小生成树集合到各个点的最小值
void Prim()
{
int i,j,k,min;
for(i=1;i<=n;i++)
lowcost[i]=graph[1][i]; //从第一个的顶点开始
memset(adjvex,0,sizeof(adjvex));
adjvex[1]=true;
for(i=1;i<=n;i++)
{
min=MAXCOST;
for(j=1;j<=n;j++)//选出当前最小生成树集合到非集合点的最小的边
if(adjvex[j]==false && lowcost[j]<min)
{
min=lowcost[j];
k=j; //记下最小的点
}
adjvex[k]=true; //为true表示该权值已经是最小,为flase是还不确定,应继续更新
for(j=1;j<=n;j++)
{
if(adjvex[j]==false && lowcost[j]>graph[k][j]) //更新lowcost
lowcost[j]=graph[k][j];
}
}
for(i=2;i<=n;i++)
{
if(lowcost[i]!=MAXCOST)
{
sum+=lowcost[i]; //求最小生成树路径长度
}
}
}
int main()
{
freopen("test.txt","r",stdin);
int n2,x,y,z;
while(scanf("%d",&n)&&n)
{
sum=0;
n2=n*(n-1)/2;
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
graph[i][j]=MAXCOST;
for(int i=1;i<=n2;i++)
{
scanf("%d%d%d",&x,&y,&z);
if(graph[x][y]>z)
{
graph[x][y]=z;
graph[y][x]=z;
}
}
Prim();
printf("%d\n",sum);
}
}
hdu1233 还是畅通工程 (prim)
最新推荐文章于 2021-05-19 05:53:55 发布