kruskal求助!
  • 板块学术版
  • 楼主windows_fleon
  • 当前回复8
  • 已保存回复8
  • 发布时间2022/6/14 14:05
  • 上次更新2023/10/27 23:19:50
查看原帖
kruskal求助!
531499
windows_fleon楼主2022/6/14 14:05
#include<bits/stdc++.h>
using namespace std;
const int N=5010,M=200010;
int n,m,cnt=1,head[N],fa[N],k;
struct aa{
	int u,v,w,next;
}f[M];
long long ans;
void adde(int u,int v,int w){
	f[cnt].u=u;
	f[cnt].v=v;
	f[cnt].w=w;
	f[cnt].next=head[u];
	head[u]=cnt++;
}
int find(int x){
	if(fa[x]==x) return x;
	return fa[x]=find(fa[x]);
}
bool cmp(aa a, aa b) {return a.w<b.w;}
void kruskal(){
	sort(f+1,f+1+m,cmp);
	for(int i=1;i<=m;i++){
		int eu=find(f[i].u),ev=find(f[i].v);
		if(eu==ev) continue;
	    fa[ev]=eu;
		ans+=f[i].w;
		k++;
		if(k==n-1) break;
	}
}
int main(){
	cin>>n>>m;
	memset(head,-1,sizeof(head));
	for(int i=1;i<=m;i++){
		int u,v,w;
		cin>>u>>v>>w;
		adde(u,v,w);
		adde(v,u,w);
	}
	for(int i=1;i<=n;i++) fa[i]=i;
	kruskal();
	cout<<ans;
	return 0;
}

只过了样例

2022/6/14 14:05
加载中...