最小生成树Prim

#include<cstdio>#include<cstring>#include<algorithm>using namespace std;const int inf=0x3f3f3f3f;int map[105][105];int prim(int n) {    int vis[105]={0},dis[105],ans=0;    memset(dis,inf,sizeof(dis));//初始化为无穷大     dis[1]=0;//随意选定一个点开始     for(int i=1;i<=n;++i)//循环加入n个顶点     {        int k=-1,tp=inf;         for(int j=1;j<=n;++j)        //查找距离已连接的部分最近的顶点         {            if(!vis[j]&&dis[j]<tp)            {                k=j;tp=dis[j];            }        }        vis[k]=1;ans+=tp;//统计和标记         for(int j=1;j<=n;++j)        //更新未选中的点和已经连接的部分之间的最短距离         {            if(!vis[j]&&map[k][j]<dis[j])            {                dis[j]=map[k][j];            }        }    }    return ans;}int main(){    int n,m,a,b,c;    while(scanf("%d",&n)&&n)    {        memset(map,inf,sizeof(map));        m=n*(n-1)/2;//边的数量         for(int i=0;i<m;i++)        {            scanf("%d%d%d",&a,&b,&c);            map[a][b]=map[b][a]=c;//无向边         }        printf("%d\n",prim(n));    }    return 0;}