70分最小生成树求助
查看原帖
70分最小生成树求助
764773
AstaVenti_楼主2023/11/17 22:16
#include<bits/stdc++.h>
using namespace std;
struct edge{
	int to,nxt,val;
}e[400500];
int h[400500],dis[400500],vis[400500],cnt,tot,ans,n,m;
void add(int from,int to,int value){
	e[cnt].nxt=h[from],e[cnt].to=to,e[cnt].val=value,h[from]=cnt++;
}
void prim(){
	memset(dis,0x3f,sizeof(dis)),dis[0]=2147483647,dis[1]=0;
	while(1){
		int u=0;for(int i=1;i<=n;i++)if(!vis[i]&&dis[i]<dis[u])u=i;
		if(!u)break;tot++,vis[u]=1,ans+=dis[u];
		for(int i=h[u];i;i=e[i].nxt)dis[e[i].to]=min(dis[e[i].to],e[i].val);
	}
}
int main(){
	cin>>n>>m;
	for(int i=1,u,v,w;i<=m;i++){
		cin>>u>>v>>w;
		add(u,v,w),add(v,u,w);
	}
	prim();
	if(tot==n)cout<<ans;
	else cout<<"orz";
	return 0;
}

2023/11/17 22:16
加载中...