Prim (堆优化)求调
查看原帖
Prim (堆优化)求调
767295
hfjqwq楼主2023/5/30 18:06
#include<bits/stdc++.h>
using namespace std;
const int maxn=5e3+10,maxm=2e5+10;
int n,m,cnt,tot,sum;
int head[maxn],to[maxm],weight[maxm],nxt[maxm],dis[maxn];
bool inq[maxn];
priority_queue<pair<int,int> > q;
void add(int u,int v,int w){
	to[++cnt]=v; weight[cnt]=w; nxt[cnt]=head[u]; head[u]=cnt;
}
void Prim(){
	memset(dis,0x3f,sizeof(dis));
	dis[1]=0; q.push(make_pair(0,1));
	while(!q.empty()){
		int d=q.top().first,now=q.top().second; q.pop();//now 编号,d 边权 
		if(inq[now]) continue; inq[now]=1;
		cnt++; sum+=d;
		for(int i=head[now];i;i=nxt[i]){
			int v=to[i];
			if(weight[i]<dis[v]){
				dis[v]=weight[i];
				q.push(make_pair(dis[v],v));
			}	
		}
	}
}
int main(){
	scanf("%d%d",&n,&m);
	for(int i=1;i<=m;i++){
		int u,v,w;
		scanf("%d%d%d",&u,&v,&w);
		add(u,v,w); add(v,u,w);
	}
	Prim();
//	cout<<cnt<<" "<<sum;
	if(cnt==n){
		cout<<sum;
	}else{
		cout<<"orz";
	}
	return 0;
}
2023/5/30 18:06
加载中...