HDU 1233 prim (first blood)

原创于 2018-06-03 16:22:42 发布 · 128 阅读

CC 4.0 BY-SA版权

本文介绍了一种求解最小生成树(Minimum Spanning Tree, MST)问题的经典算法——Prim算法，并通过C++代码实现了该算法。文章首先定义了一个包含顶点数n的图，并初始化了图的邻接矩阵，随后详细展示了Prim算法的具体步骤，包括如何选择下一个加入最小生成树的顶点，以及更新其他顶点到当前最小生成树的距离。最后，主函数中通过读取输入生成图并调用Prim算法计算最小生成树的总权重。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

http://acm.hdu.edu.cn/showproblem.php?pid=1233

#include<bits/stdc++.h>
using namespace std;
#define inf 0x3f3f3f3f
int n;
int mmp[101][101],dis[101],vis[101];
int prim()
{
    int i,j,pos,result=0;
    memset(vis,0,sizeof(vis));
    vis[1]=1;
    for(i=1; i<=n; i++)
        if(i!=1)
            dis[i]=mmp[1][i];
    for(i=1; i<n; i++)
    {
        int minn=inf;
        for(j=1; j<=n; j++)
        {
            if(!vis[j]&&minn>dis[j])
                minn=dis[pos=j];//注意别写反了
        }
        result+=minn;
        vis[pos]=1;
        for(j=1; j<=n; j++)
        {
            if(!vis[j]&&dis[j]>mmp[pos][j])
                dis[j]=mmp[pos][j];
        }
    }
    return result;
}
int main()
{

    while(scanf("%d",&n)!=EOF)
    {
        if(n==0)
            break;
        int m=n*(n-1)/2,i,j,a,b,c;
        for(i=1; i<=n; i++)
        {
            for(j=1; j<=n; j++)
                mmp[i][j]=inf;
        }
        for(i=1; i<=m; i++)
        {
            scanf("%d%d%d",&a,&b,&c);
            mmp[a][b]=c;
            mmp[b][a]=c;
        }
        int ans=prim();
        printf("%d\n",ans);
    }
    return 0;
}