(模板) 最小生成树Prim算法

39 篇文章 0 订阅
5 篇文章 0 订阅

算法跟Dijkstra很像 ,但是要特别注意两点:1.最好使用一个额外的数组来保证每个点只被访问一次,否则的话应该保证在35行处使用e.dis>=dis[e.num]而不仅是>,否则会造成一些相等的额外相加。
2.40行的地方要注意与Dijkstra的区别!!不是加而是直接换!!!

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<vector>
#include<cstring>
#include<queue>
using namespace std;
struct edge{
    int to,cost;
};
vector<edge>G[5005];
struct d{
    int num,dis;
};
bool operator <(d a,d b)
{
    return a.dis>b.dis;
}
bool vis[5005];
int main()
{
    int n,m,i,j,k;
    cin>>n>>m;
    for(i=1;i<=m;i++){
        int a,b,c;
        scanf("%d %d %d",&a,&b,&c);
        G[a].push_back(edge{b,c});
        G[b].push_back(edge{a,c});
    }
    int dis[5005];memset(dis,0x7f,sizeof(dis));dis[1]=0;
    priority_queue<d>que;que.push(d{1,0});
    int ans=0;
    while(!que.empty()){
        d t=que.top();que.pop();
        if(vis[t.num])continue;
        ans+=t.dis;vis[t.num]=true;
        for(i=0;i<G[t.num].size();i++){
            edge e=G[t.num][i];
            if(dis[e.to]>e.cost){
                dis[e.to]=e.cost;
                que.push(d{e.to,dis[e.to]});
            }
        }
    }
    cout<<ans<<endl;

    return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值