蒟蒻求助,悬赏关注!
查看原帖
蒟蒻求助,悬赏关注!
769006
crzcqh楼主2023/8/13 12:36
#include<bits/stdc++.h>
using namespace std;
int n,m,u,v,w,cnt,ans;
int fa[5005],vis[5005],head[5005];
struct node{
	int u,v,w,ne;
}e[200010];
int find(int x){
	return fa[x]==x?x:fa[x]=find(fa[x]);
}
int merge(int x1,int x2){
	if(find(x1)!=find(x2)){
		fa[x1]=x2;
		return 1;
	}
	return 0;
}
void add(int u,int v,int w){
	e[++cnt].v=v;
	e[cnt].u=u;
	e[cnt].w=w;
	e[cnt].ne=head[u];
	head[u]=cnt;
}
void dfs(int x){
	vis[x]=1;
	for(int i=head[x];i;i=e[i].ne){
		if(vis[e[i].v]==0) dfs(e[i].v);
	}
}
bool cmp(node a,node b){
	return a.w<b.w;
}
int main(){
	cin>>n>>m;
	for(int i=1;i<=m;i++){
		cin>>u>>v>>w;
		add(u,v,w);
		add(v,u,w);
	} 
	dfs(1);
	for(int i=1;i<=n;i++)
		if(vis[i]==0){
			cout<<"orz";
			return 0;
		}
	for(int i=1;i<=n;i++) fa[i]=i;
	sort(e+1,e+1+m,cmp);
	for(int i=1;i<=m;i++){
		if(merge(e[i].u,e[i].v)){
			ans+=e[i].w;
		}
	}
	cout<<ans;
	return 0;
}
2023/8/13 12:36
加载中...