【模板】最小生成树（Prim算法——优先队列实现形式）：适用于稀疏图

green qwq

于 2024-06-29 10:00:00 发布

阅读量1.3k

点赞数 42

分类专栏：新知识文章标签：算法

本文链接：https://blog.csdn.net/m0_60738889/article/details/140053115

版权

新知识专栏收录该内容

31 篇文章 0 订阅

订阅专栏

【模板】最小生成树

题目描述

如题，给出一个无向图，求出最小生成树，如果该图不连通，则输出 orz。

输入格式

第一行包含两个整数 $N, M$ ，表示该图共有 $N$ 个结点和 $M$ 条无向边。

接下来 $M$ 行每行包含三个整数 $X_i,Y_i,Z_i$ ，表示有一条长度为 $Z_i$ 的无向边连接结点 $X_i,Y_i$ 。

输出格式

如果该图连通，则输出一个整数表示最小生成树的各边的长度之和。如果该图不连通则输出 orz。

样例 #1

样例输入 #1

样例输出 #1

提示

数据规模：

对于 $20\%$ 的数据， $N\le 5$ ， $M\le 20$ 。

对于 $40\%$ 的数据， $N\le 50$ ， $M\le 2500$ 。

对于 $70\%$ 的数据， $N\le 500$ ， $M\le 10^4$ 。

对于 $100\%$ 的数据： $1\le N\le 5000$ ， $1\le M\le 2\times 10^5$ ， $1\le Z_i \le 10^4$ 。

样例解释：

所以最小生成树的总边权为 $2 + 2 + 3 = 7$ 。

模板

说明：本代码采用Prim算法，如果是稠密图，我们用暴力法，时间复杂度为 $O(n^2)$ ，如果是优先队列的话，时间复杂度是 $O (m l o g m)$ 的（ $m$ 是边数），因为 push 和 pop 的时间复杂度是 $O (l o g m)$ 的。

//优先队列实现最小生成树——prim

#include<iostream>
#include<algorithm>
#include<cstring>
#include<queue>

using namespace std;

const int N = 2e5+10,M = 2*N;//稀疏图

typedef pair<int,int>PII;

int e[M],ne[M],w[M],h[N],idx;
int dist[N];
bool st[N];
int ans;
int n,m;

void add(int a,int b,int c){
    e[idx]=b,w[idx]=c,ne[idx]=h[a],h[a]=idx++;
}

bool prim(){
    priority_queue<PII,vector<PII>,greater<PII>>q;
    
    memset(dist,0x3f,sizeof dist);
    
    dist[1]=0;
    
    int cnt=0;
    
    q.push({0,1});
    
    while(q.size()){
        PII t=q.top();
        q.pop();
        
        int ver=t.second;
        
        if(st[ver])continue;
        st[ver]=true;
        
        ans+=dist[ver];
        cnt++;
        
        for(int i=h[ver];~i;i=ne[i]){
            int j=e[i];
            
            if(dist[j]>w[i]){
                dist[j]=w[i];
                q.push({dist[j],j});
            }
        }
    }
    
    return cnt==n;
}

int main(){
    cin>>n>>m;
    
    memset(h,-1,sizeof h);
    
    for(int i=1;i<=m;i++){
        int a,b,c;
        cin>>a>>b>>c;
        add(a,b,c);
        add(b,a,c);
    }
    
    if(!prim()){
        puts("orz");
    }else{
        cout<<ans;
    }
    
    return 0;
}