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

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

#include<stdio.h>
#include<string.h>
#define inf 99999

using 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;
}