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

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

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

#include 
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;
}

你可能感兴趣的:(板子代码,acm一些基本算法)