马上要回家了,复习复习我学习过的知识,
prim写的最小生成树,其实prim和dijkstra是非常相似的,只不过是dijkstra更新的是和而已,,,
所以两个算法很相似,,只不过是吧dis[]=map[][]+dis[] 变成了 dis[]=map[][],当然 同样变化的还有判断条件;
贴出代码:
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <math.h>
#define inf 0x3fffffff
int N;
int map[105][105];
int visit[105];
int dis[105];
int prim()
{
for(int i=1;i<=N;i++)
dis[i]=inf;
dis[1]=0;
for(int j=1;j<=N;j++)
{
int t=inf;
int pos;
for(int i=1;i<=N;i++)
{
if(!visit[i]&&t>dis[i])
{
t=dis[i];
pos=i;
}
}
visit[pos]=1;
for(i=1;i<=N;i++)
{
if(!visit[i]&&map[pos][i]!=0x3f3f3f3f&&dis[i]>map[pos][i])
{
dis[i]=map[pos][i];
}
}
}
int sum=0;
for(i=1;i<=N;i++)
{
sum+=dis[i];
}
return sum;
}
int main()
{
while(scanf("%d",&N),N)
{
int M=(N-1)*N>>1;
int a,b,val;
memset(map,0x3f,sizeof(map));
memset(visit,0,sizeof(visit));
for(int i=1;i<=M;i++)
{
scanf("%d%d%d",&a,&b,&val);
if(map[a][b]>val)
{
map[a][b]=val;
map[b][a]=val;
}
}
int ans=prim();
printf("%d\n",ans);
}
return 0;
}