HDOJ 1863 畅通工程——普利姆算法

普利姆算法是以一个点为源点，找该源点到下一个点的最短距离

#include<stdio.h>#include<string.h>#define inf 99999using namespace std;int mp[110][110];int dis[110];bool vis[110];//普利姆算法void prim(int m){    int i,j;    int sum=0;    memset(vis,0,sizeof(vis));//标记数组清零    for(i=1;i<=m;i++)    {        dis[i]=mp[1][i];//对dis数组初始化    }    vis[1]=1;    for(i=1;i<m;i++)//m-1次循环，循环遍历完所有的点    {        int k=0,mini=inf;        for(j=1;j<=m;j++)        {            if(!vis[j]&&dis[j]<mini)            {                mini=dis[k=j];            }        }//循环一遍，找到权值最小的边，并记录下所连的点        sum+=mini;        vis[k]=1;        for(j=1;j<=m;j++)        {            if(!vis[j]&&dis[j]>mp[k][j])                dis[j]=mp[k][j];        }//以k点为源点，用k点到各点的距离更新dis数组    }    int flag=0;    for(i=1;i<=m;i++)    {        if(!vis[i])        {            flag=1;            break;        }    }//若有没被标记的点，则不能畅通    if(!flag)        printf("%d\n",sum);    else        printf("?\n");}int main(){    int n,m,i,j;    while(~scanf("%d%d",&n,&m))    {        if(n==0)            break;        for(i=0;i<=m;i++)        {            for(j=0;j<=m;j++)            {                if(i==j)                    mp[i][j]=0;                else                    mp[i][j]=inf;            }        }//mp数组初始化，开始为散落的、无关联的点。        //除自身到自身距离为0外，其余都为无穷。        int a,b,c;        for(i=1;i<=n;i++)        {            scanf("%d%d%d",&a,&b,&c);            if(mp[a][b]>c)//两点间的距离可能有多个，取最小值            {                mp[a][b]=c;                mp[b][a]=c;//无向图            }        }        prim(m);    }    return 0;}