最小生成树 Prim算法 + 链式前向星+ 堆优化板子

最新推荐文章于 2026-04-27 02:37:28 发布

原创最新推荐文章于 2026-04-27 02:37:28 发布 · 657 阅读

1 ·

本内容遵循CC 4.0 BY-SA版权协议

板子代码同时被 2 个专栏收录

30 篇文章

订阅专栏

acm一些基本算法

26 篇文章

订阅专栏

本文分享了一种简洁且易读的Prim算法实现代码，该算法用于求解最小生成树问题，时间复杂度为O(NlogN)。通过使用优先队列进行优化，代码结构清晰，适合初学者学习和理解。

初学，听大佬说时间复杂度为O(NlogN)
主要是感觉这个板子太简洁了，看起来舒服。。。(在Prim算法中算不错的了)

板子题目链接//~~大佬就不要嘲讽了。我就写着玩的。。。~~
AC代码
如果看不懂，可以去题目里面的题解去看，里面的大佬杠杠滴。

#include <bits/stdc++.h>
using namespace std;
#define NewNode (ListNode *)malloc(sizeof(ListNode))
#define Mem(a,b) memset(a,b,sizeof(a))
const int N = 2e5 + 50;
const int INF = 0x3f3f3f3f;
const double EPS = 1e-10;
const unsigned long long mod = 998244353;
const int II = 3.1415926535;
typedef long long ll;
typedef unsigned long long ull;
typedef pair <int,int> pii;
priority_queue<pii , vector <pii> , greater <pii> > q;
struct node
{
    int z;
    int val;
    int Next;
}Node[N*2];//链式前向星，又叫做邻接表。
int Head[N],vis[N],m = 0,dis[N],ans,sum,n,k;
void Add(int a,int b,int c)
{
    m++;
    Node[m].z = b;
    Node[m].val = c;
    Node[m].Next = Head[a];
    Head[a] = m;
}
void Prim()
{
    q.push(make_pair(0,1));//用优先队列来 优化
    while(!q.empty() && ans < n)
    {
        int d = q.top().first,u = q.top().second;
        q.pop();
        if(vis[u])
            continue;
        sum += d;
        ans++;
        vis[u] = 1;
        for(int i = Head[u];i;i = Node[i].Next)
            if(Node[i].val < dis[Node[i].z])
                dis[Node[i].z] = Node[i].val,q.push(make_pair(Node[i].val,Node[i].z));
    }
}
int main()
{
    std::ios::sync_with_stdio(false);
    cin.tie(0),cout.tie(0);
    fill(dis+1,dis+N,INF);
    cin >> n >> k;
    for(int i = 0;i < k;i++)
    {
        int a,b,c;
        cin >> a >> b >> c;
        Add(a,b,c);
        Add(b,a,c);//无向图的关系
    }
    Prim();
    n == ans ? cout << sum << endl : cout << "orz" << endl;
}