Prim堆优化

最新推荐文章于 2025-04-24 21:21:17 发布

lcy19260817

最新推荐文章于 2025-04-24 21:21:17 发布

阅读量974

点赞数

分类专栏：图论——最小生成树——Prim 文章标签： prim

本文链接：https://blog.youkuaiyun.com/lcy19260817/article/details/90209183

版权

图论——最小生成树——Prim 专栏收录该内容

1 篇文章

订阅专栏

本文介绍了一种使用优先队列优化的Prim算法实现最小生成树(MST)的方法。通过C++代码详细展示了如何构建图的数据结构，以及如何利用优先队列进行节点选择，以达到算法效率的提升。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Prim堆优化

跟 $d^i_jk^s_tr^a$ 一样啦。。。

#include<cstdio>
#include<iostream>
#include<queue>
using namespace std;
int read(){
	int x=0,pos=1;char ch=getchar();
	for(;!isdigit(ch);ch=getchar()) if(ch=='-') pos=0;
	for(;isdigit(ch);ch=getchar()) x=x*10+ch-'0';
	return pos?x:-x;
}
struct node{
	int v,nex,w;
}edge[2000001];
struct pt{
	int po,d;
	bool operator < (const pt &a) const{
		return d>a.d;
	}
};
int top,head[1000001];
void add(int u,int v,int w){
	edge[++top].v=v;
	edge[top].nex=head[u];
	edge[top].w=w;
	head[u]=top;
}
int n,m;
queue<pt>q;
int vis[1000001];
int main(){
	n=read(),m=read();
	for(int i=1;i<=m;i++){
		int u=read(),v=read(),w=read();
		add(u,v,w);
		add(v,u,w);
	}
	int ans=0;
	priority_queue<pt>q;
	pt fi;
	fi.po=1;
	fi.d=0;
	q.push(fi);
	int cnt=0;
	while(!q.empty()){
		pt now=q.top();q.pop();
		if(vis[now.po]) continue;
		cnt++;
		vis[now.po]=1;
		ans+=now.d;
		for(int i=head[now.po];i;i=edge[i].nex){
			int v=edge[i].v;
			if(!vis[v]){
				pt ne;
				ne.d=edge[i].w;
				ne.po=v;
				q.push(ne);
			}
		}
	}
	if(cnt==n) printf("%d",ans);
	else printf("orz");
	return 0;
}