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;
}