最小生成树之Prime算法（基于优先队列）

最新推荐文章于 2022-04-30 21:15:25 发布

Think_Idea

最新推荐文章于 2022-04-30 21:15:25 发布

阅读量979

点赞数 1

分类专栏：模板图论

本文链接：https://blog.csdn.net/mengxingyuanlove/article/details/48695887

版权

本文介绍了基于优先队列实现的Prime算法，用于解决稠密图的最小生成树问题。通过贪心策略，算法的时间复杂度为O(NlogN+M)。在主函数中，读取图的边权重并调用Prime函数，输出从特定顶点出发的最小生成树总权重。

摘要由CSDN通过智能技术生成

#include <cstdio>
#include <queue>
#include <cstring>
#define MAXN 1010
#define mem(a,b) memset(a,b,sizeof(a))
using namespace std;
/*
 * Prme算法：贪心，适用于稠密图
 * 时间复杂度：O（NlogN+M）
*/
struct node{
    int v,dist;
    node(int v,int dist):v(v),dist(dist){}
    bool operator<(const node& b)const{
        return dist>b.dist;
    }
};
int n, g[MAXN][MAXN],dis[MAXN];
bool vis[MAXN];
priority_queue<node> q;
int Prime(int u){
    int ans=0;
    mem(vis,false);
    while(!q.empty()) q.pop();
    for(int i=1;i<=n;++i) if(i!=u){dis[i]=g[u][i]; q.push(node(i,dis[i]));}
    vis[u]=true;
    dis[u]=0;
    int cnt=1;
    while(!q.empty()){
        node tmp=q.top(); q.pop();
        u=tmp.v;
        if(vis[u]) continue;
        vis[u]=true;
        ans+=tmp.dist; cnt++;
        if(cnt>=n) break;
        for(int i=1;i<&#