【模板】最小生成树Prim 4heap

基本介绍
模板题目
代码实现

基本介绍

终于来填Prim的坑了 代码实现和最短路的Dijkstra差不多 也用的堆优化
大体意思就是说 现在图中选取一个蓝点染成白色 然后遍历与这个点相连的所有边 选取最短的边然后将另一个端点染成白色 再遍历这个点 从所有白点中找最短的 一直这样下去 stl小根堆比较好 便于每次找短的

模板题目

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

输入输出格式
输入格式：
第一行包含两个整数N、M，表示该图共有N个结点和M条无向边。（N<=5000，M<=200000）
接下来M行每行包含三个整数Xi、Yi、Zi，表示有一条长度为Zi的无向边连接结点Xi、Yi
输出格式：
输出包含一个数，即最小生成树的各边的长度之和；如果该图不连通则输出orz

输入输出样例
输入样例：
4 5
1 2 2
1 3 2
1 4 3
2 3 4
3 4 3
输出样例：
7

代码实现

#include<iostream>
#include<cstdio>
#include<cctype>
#include<queue>

    using namespace std;
    #define in = read()
    typedef long long ll;
    const ll size = 1000000 + 10000;

        struct point{   ll next,to,dis;}edge[size];
        priority_queue<pair<ll , ll> , vector<pair<ll , ll> > , greater<pair<ll , ll> > > q;

        ll n,m;
        ll site,ans,total,u;
        ll head[size];
        bool exist[size];

inline ll read(){
        ll num = 0 , f = 1;   char ch = getchar();

        while(!isdigit(ch)){
                if(ch == '-')   f = -1;
                ch = getchar();
        }

        while(isdigit(ch)){
                num = num*10 + ch - '0';
                ch = getchar();
        }

        return num*f;
}

inline void add(ll x,ll y,ll z){
        edge[++site].next = head[x];
        edge[site].to = y;
        edge[site].dis = z;
        head[x] = site;
}

int main(){
        n in;   m in;
        for(int i=1;i<=m;i++){
                ll x,y,z;   x in;   y in;   z in;
                add(x,y,z);   add(y,x,z);
        }

        pair<ll , ll> x;
        q.push(make_pair(0,1));
        while(!q.empty() && total != n - 1){
                x = q.top();    q.pop();    u = x.second;
                if(exist[u])    continue;
                exist[u] = true;
                ans += x.first;
                total --;
                for(int i=head[u];i;i=edge[i].next)
                        if(!exist[edge[i].to])
                                q.push(make_pair(edge[i].dis,edge[i].to));
        }

        printf("%d",ans);
}


//COYG