还是畅通工程

题意：某省调查乡村交通状况，得到的统计表中列出了任意两村庄间的距离。省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通（但不一定有直接的公路相连，只要能间接通过公路可达即可），并要求铺设的公路总长度为最小。请计算最小的公路总长度。

思路：prim算法，从任一i点开始，找出最短相邻边，将k点加入自己，遍历该点有关的所有边，跟新i点与其他点的距离，寻找下一个k点直到遍历结束。

#include<iostream>
#include<string.h>
using namespace std;
const int N=105;
const int INF=0x3f3f3f3f;
int n,m;//n个点，m条边 
int map[N][N];//存储每条边 
int vis[N],dis[N];
int ans,k;
void prim(){
    //点1到每条边的距离 
    for(int i=1;i<=n;++i){
        dis[i]=map[1][i];
    }
    vis[1]=1;
    for(int i=2;i<=n;++i){
        int min=INF;
        for(int j=1;j<=n;++j){
            if(!vis[j]&&min>dis[j]){
                min=dis[j];
                k=j;
            } 
        }
        vis[k]=1;        
        ans+=min;        
        for(int j=1;j<=n;++j){
            if(!vis[j]&&dis[j]>map[k][j]){
                dis[j]=map[k][j];
            }
        }
    }
}
int main() {
    ios::sync_with_stdio(false);
    while(scanf("%d",&n)&&n){
        m=n*(n-1)/2;
        memset(vis,0,sizeof(vis));
        for(int i=1;i<=100;++i){
            for(int j=1;j<=100;++j){
                if(i==j){
                    map[i][j]=0;                    
                }
                else{
                    map[i][j]=INF;
                }
            }
        }
        for(int i=1;i<=m;++i){
            int u,v,w;
            scanf("%d %d %d",&u,&v,&w);
            if(map[u][v]>w){
                map[u][v]=w;
                map[v][u]=w;
            }            
        }
        ans=0;
        prim();
        printf("%I64d\n",ans);
    }
    
    return 0;
}