Graph Theory NO.7_HDU_1233_还是畅通工程_最小生成树

#include <cstdio>
#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
#define INF 0xffffff
int map[101][101];
int visit[10009];
int low[10009];
int prim(int n)
{
for(int i=1;i<=n;i++)
{
low[i]=map[1][i];
visit[i]=0;
}
visit[1]=1;
int sum=0;
for(int i=1;i<n;i++)
{
int temp=INF,pos;
for(int j=1;j<=n;j++)
{
if(!visit[j]&&low[j]<temp)
{
temp=low[j];
pos=j;
}
}
visit[pos]=1;
sum+=temp;
for(int j=1;j<=n;j++)
{
if(!visit[j]&&low[j]>map[pos][j])
{
low[j]=map[pos][j];
}
}
}
return sum;
}
int main()
{
int n;
while(~scanf("%d",&n),n)
{
int a,b,c;
for(int i=0;i<n*(n-1)/2;i++)
{
scanf("%d%d%d",&a,&b,&c);
map[a][b]=map[b][a]=c;
}
printf("%d\n",prim(n));
}
return 0;
}