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]灌水

上一篇:Prim算法堆优化


下一篇:bzoj1601 / P1550 [USACO08OCT]打井Watering Hole(堆优化prim)