C语言编程练习64：还是畅通工程

#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<cmath>
#include<iostream>
using namespace std;

const int INF=0x3f3f3f3f;
int mp[100][100];//邻接矩阵存边
int n,m,x,y,z;
int dis[10005];//到点的距离
int vis[100];//标记点是否被访问
void prim()
{
    dis[1]=0;
    while(1)
    {
        int k=-1;
        int minn=INF;
        for(int i=1;i<=n;i++)
        {
            if(dis[i]<minn&&vis[i]==0)
            {
                minn=dis[i];
                k=i;//当前距离最短的点
            }
        }
        if(k==-1)
        {
            break;
        }
        vis[k]=1;
        for(int i=1;i<=n;i++)//访问当前距离最短的点所连接的点
        {
            if(dis[i]>mp[k][i]&&vis[i]==0)
            {
                dis[i]=mp[k][i];
            }
        }
    }
    int ans;
    ans=0;
    for(int i=1;i<=n;i++)
    {
        ans+=dis[i];
    }
    cout<<ans<<endl;
}
int main()
{
    while(scanf("%d",&n)!=EOF)
    {
        if(n==0)
        {
            break;
        }
        m=n*(n-1)/2;
        memset(vis,0,sizeof(vis));
        memset(dis,INF,sizeof(dis));
        for(int i=1;i<=n;i++)
        {
            for(int j=1;j<=n;j++)
            {
                if(i==j)
                {
                    mp[i][j]=0;
                }
                else
                {
                    mp[i][j]=INF;
                }
                
            }
        }
        for(int i=0;i<m;i++)
        {
            cin>>x>>y>>z;
            if(mp[x][y]>z)
            {
                mp[x][y]=z;
                mp[y][x]=z;
            }
        }
        prim();
    }
    return 0;
}