最小生成树(prim算法)

主要是听说prim和dijkstra是孪生兄弟所以才先拿它下手(其实代码就改了一行好像)。
最小生成树模板题
大概意思就是说再n个点之中找出n-1条边使这n-1条边能将每个点连通并且这n-1条边的边权值之和要最小。
prim原理图:
在这里插入图片描述
和dijkstra不同prim并没有源点的概念可以任选一个点出发(反正最后结果都一样),图中是以v1为起点更新并找出当前已知边的最小边并以其终点为下一阶段的起点继续更新并重复上述操作,直到所有顶点都被更新。
附完整代码:

#include <bits/stdc++.h>//最小生成树 prim算法
using namespace std;
#define int long long
#define lll 0x3f3f3f3f
struct node{
    int next,to,w;
}p[200005<<1];
int head[5005],dis[5005]; bool vis[5005]; int cnt;
void add(int a,int b,int c){//链式前向星存图
    p[cnt].to=b; p[cnt].w=c;
    p[cnt].next=head[a];
    head[a]=cnt; cnt++;
}
signed main(){
    int n,m; cin>>n>>m;
    memset(head,-1,sizeof(head));
    for(int i=0;i<m;i++){
        int a,b,c; cin>>a>>b>>c;
        add(a,b,c);
        add(b,a,c);
    }
    memset(dis,lll,sizeof(dis));
    dis[1]=0; vis[0]=true;
    for(int i=head[1];i!=-1;i=p[i].next){//更新起点所连的边 
        dis[p[i].to]=min(dis[p[i].to],p[i].w);
    }
    for(int i=1;i<=n-1;i++){
        int post,mm=lll;
        for(int j=1;j<=n;j++){
            if(!vis[j]&&dis[j]<mm){
                mm=dis[j]; post=j;
            }
        }
        vis[post]=true;
        for(int i=head[post];i!=-1;i=p[i].next){
            if(dis[p[i].to]>p[i].w&&!vis[p[i].to]){//必须将已标记的节点排除确保当前节点不在连通分量之中
                dis[p[i].to]=p[i].w;
            }
        }
    }
    int sum=0;
    for(int i=1;i<=n;i++){
        if(dis[i]==lll) {//表示存在顶点未被更新,即图未连通
            cout<<"orz"<<endl; return 0;
        }
        sum+=dis[i];
    }
    cout<<sum<<endl;
}

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值