洛谷 P1546 最短网络 Agri-Net（最小生成树_Prim）

最新推荐文章于 2024-06-01 09:00:18 发布

原创最新推荐文章于 2024-06-01 09:00:18 发布 · 395 阅读

0 ·

CC 4.0 BY-SA版权

最小生成树专栏收录该内容

5 篇文章

订阅专栏

本文介绍使用Prim算法求解最小生成树问题的一种实现方法。通过初始化距离数组并不断更新顶点状态，最终求得最小生成树的总权重。

传送门

最小生成树模板，大家都说是Kruskal，但brz大神说是稠密图要用Prim。
由于大神很强我听大神的

关于Prim算法和Kruskal看这里，我觉得他写得很好

Code:

#include<cstdio>
#include<cstdlib>
#include<cstring>

int n,ans=0;
int a[110][110],d[110];
bool v[110];

int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
        for(int j=1;j<=n;j++)
            scanf("%d",&a[i][j]);
    memset(d,63,sizeof(d));
    for(int i=1;i<=n;i++) d[i]=a[1][i];
    memset(v,false,sizeof(v));v[1]=true;
    for(int i=1;i<=n;i++)
    {
        int k=0;
        for(int j=1;j<=n;j++)
            if(!v[j] && d[k]>d[j]) k=j;
        v[k]=true;
        for(int j=1;j<=n;j++)
            if(!v[j] && a[k][j]<d[j]) d[j]=a[k][j];
    }
    for(int i=1;i<=n;i++) ans+=d[i];
    printf("%d",ans);
}