最小生成树板子(不加堆优化)

最新推荐文章于 2023-07-17 12:47:48 发布

原创最新推荐文章于 2023-07-17 12:47:48 发布 · 178 阅读

0 ·

CC 4.0 BY-SA版权

板子代码专栏收录该内容

30 篇文章

订阅专栏

本文深入探讨了Prim算法在解决最小生成树问题中的应用，通过详细的代码解析，介绍了如何构造图的数据结构，使用Prim算法寻找加权无向图的最小生成树，以及算法的具体实现过程。文章适用于对图论和算法设计感兴趣的读者。

题目链接(板子题)

#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;
struct node
{
    int z;
    int val;
    int Next;
}Node[2*N];
int Head[N],vis[N],m,n,k,ans,sum,dis[N];
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()
{
    int now = 1;
    fill(dis+1,dis+N,INF);
    for(int i = Head[1];i;i = Node[i].Next)
        dis[Node[i].z] = min(dis[Node[i].z],Node[i].val);
    while(++ans < n)
    {
//        cout << sum << " " << now << endl;
        vis[now] = 1;
        int Min = INF;
        for(int i = 1;i <= n;i++)
            if(!vis[i] && Min > dis[i])
                Min = dis[i],now = i;
        sum += Min;
        for(int i = Head[now];i;i = Node[i].Next)
            if(!vis[Node[i].z])
                dis[Node[i].z] = min(dis[Node[i].z],Node[i].val);
    }
}
int main()
{
    std::ios::sync_with_stdio(false);
    cin.tie(0),cout.tie(0);
    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();
    cout << sum << endl;
}