P3366 【模板】最小生成树（堆优化prim）

堆优化prim

#include<cstdio>
#include<cstring>
#include<queue>
using namespace std;
struct data{
    int d,u;
    bool operator < (const data &tmp) const {return d>tmp.d;}
}e[];//注意优先队列默认是大根堆
priority_queue <data> h;
bool vis[];
int n,m,ans,dis[],cnt,hd[],nxt[],ed[];
inline void add(int x,int y,int v){ //邻接表存边
    nxt[ed[x]]=++cnt; hd[x]= hd[x] ? hd[x]:cnt;
    ed[x]=cnt; e[cnt]=(data){v,y};
}
int main(){
    memset(dis,,sizeof(dis));
    scanf("%d%d",&n,&m); int q1,q2,q3,k=;
    for(int i=;i<=m;++i) scanf("%d%d%d",&q1,&q2,&q3),add(q1,q2,q3),add(q2,q1,q3);
    dis[]=; h.push((data){,});
    while(!h.empty()&&k<n){
        data x=h.top(); h.pop();
        if(vis[x.u]) continue;
        vis[x.u]=; ++k; ans+=x.d;
        for(int i=hd[x.u];i;i=nxt[i])
            if(dis[e[i].u]>e[i].d)//更新周围的点
                dis[e[i].u]=e[i].d,h.push((data){dis[e[i].u],e[i].u});
    }
    printf("%d",ans);
    return ;
}

应用：bzoj1601: [Usaco2008 Oct]灌水