AcWing
  • 首页
  • 题库
  • 题解
  • 分享
  • 问答
  • 活动
  • 应用
  • 吐槽
  • 登录/注册

AcWing 858. Prim算法求最小生成树    原题链接    简单

作者: 作者的头像   TaoZex ,  2019-08-05 14:17:13 ,  阅读 428


2


#include<bits/stdc++.h>
using namespace std;

const int N=510,INF=0x3f3f3f3f;
int g[N][N];
int d[N];
bool st[N];
int n,m;

int prim(){
    memset(d,INF,sizeof(d));
    int res=0;

    for(int i=0;i<n;i++){                
        int t=-1;
        for(int j=1;j<=n;j++){
            if(!st[j]&&(t==-1||d[j]<d[t])){    //找出不在集合中离集合最近的点
                t=j;
            }
        }

        if(i&&d[t]==INF) return INF;       //判断是否连通,具有最小生成树   
                                           //(有些图有负权边,最小生成树权值和恰好为-1,返回-1就会出错)

        if(i) res+=d[t];
        st[t]=true;                        //将t加入集合,并更新最小生成树权值

        for(int j=1;j<=n;j++) d[j]=min(d[j],g[t][j]);     //更新所连点到集合的最短距离
    }
    return res;
}

int main(){
    cin>>n>>m;

    for(int i=1;i<=n;i++){
        for(int j=1;j<=n;j++){
            if(i==j) g[i][j]=0;
            else g[i][j]=INF;
        }
    }

    while(m--){
        int a,b,c;
        cin>>a>>b>>c;
        g[a][b]=g[b][a]=min(g[a][b],c);
    }

    int ans=prim();
    if(ans==INF) puts("impossible");
    else cout<<ans<<endl;
}

0 评论

你确定删除吗?

© 2018-2021 AcWing 版权所有  |  京ICP备17053197号-1
联系我们  |  常见问题
AcWing
请输入登录信息
更多登录方式: 微信图标 qq图标
请输入绑定的邮箱地址
请输入注册信息