社区讨论

求助大佬!!!

P3366【模板】最小生成树参与者 2已保存回复 2

讨论操作

快速查看讨论及其快照的属性,并进行相关操作。

当前回复
2 条
当前快照
1 份
快照标识符
@mibd05iy
此快照首次捕获于
2025/11/23 14:51
3 个月前
此快照最后确认于
2025/11/23 16:26
3 个月前
查看原帖
CPP
#include<bits/stdc++.h>
using namespace std;
const int N=5100,M=2e5+10,INF=0x3f3f3f3f;
typedef pair<int,int> PII;
int n,m;
int h[N],e[M],ne[M],w[M],idx;
int dist[N];
bool st[N];
void add(int a,int b,int c){
	e[idx]=b,w[idx]=c,ne[idx]=h[a],h[a]=idx++;
}
int prim(){
	memset(dist,0x3f,sizeof(dist));
	dist[1]=0;
	priority_queue<PII,vector<PII>,greater<PII> > heap;
	heap.push({0,1});
	int MST=0;
	int cnt=0;

	while(heap.size()){
		PII t=heap.top();
		heap.pop();
		int u=t.second,distince=t.first;
		if(st[u])continue;
		st[u]=true;
		MST+=distince;
		cnt++;
		for(int i=h[u];i!=-1;i=ne[i]){
			int v=e[i];
			if(!st[v]&&dist[v]>w[i]){
				dist[v]=w[i];
				heap.push({dist[v],v});
			}
		}
	}
	return cnt<n?INF:MST;
}
int main(){
	scanf("%d%d",&n,&m);
	memset(h,-1,sizeof(h));
	while(m--){
		int u,v,w;
		scanf("%d%d%d",&u,&v,&w);
		add(u,v,w),add(v,u,w);
	}
	int t=prim();
	if(t==INF)puts("orz");
	else printf("%d\n",t);
	return 0;
}
三个点RE,求助求助!!!

回复

2 条回复,欢迎继续交流。

正在加载回复...