还是畅通工程

http://acm.hdu.edu.cn/showproblem.php?pid=1233

Prim:

#include<stdio.h>
#include<string.h>
#define MAX 99999
#define M 110
int dis[M] ;
int map[M][M] ;
bool vis[M] ;
void init()
{
    for(int i=0; i<M; i++)
    for(int j=0; j<M; j++)
    {
        if(i==j) map[i][j] = 0 ;
        map[i][j] = MAX ;
    }
}
int prim(int n)
{
    int i, j, min, pos ;
    int sum = 0 ;
    memset(vis, 0, sizeof(vis)) ;
    for(i=1; i<=n; i++)
    dis[i] = map[1][i] ;
    vis[1] = true ;
    dis[1] = MAX ;
    for(i=2; i<=n; i++)
    {
        min = MAX ;
        for(j=1; j<=n; j++)
        {
            if(!vis[j]&&dis[j]<min)
            {
                min = dis[j] ;
                pos = j ;
            }
        }
        sum += dis[pos] ;
        vis[pos] = true ;
        for(j=1; j<=n; j++)
        {
            if(!vis[j]&&dis[j]>map[pos][j])
            {
                dis[j] = map[pos][j] ;
            }
        }
     }
    return sum ;
}
int main()
{
    int a, b, c, n ;
    while(scanf("%d", &n), n)
    {
        init() ;
        for(int i=1; i<=n*(n-1)/2; i++)
        {
            scanf("%d%d%d", &a,&b, &c) ;
            if(map[a][b]>c)
            map[a][b] = map[b][a] = c ;
        }
        printf("%d\n", prim(n)) ;
    }
    return 0 ;
}