牛客（最小生成树）

最新推荐文章于 2024-01-28 11:23:20 发布

原创最新推荐文章于 2024-01-28 11:23:20 发布 · 357 阅读

0 ·

CC 4.0 BY-SA版权

本文介绍了一种使用C++实现的最小生成树算法。通过结构体存储边的信息，并利用并查集来避免形成环路，确保最终形成的树是连通且不包含环路的。程序读取输入的顶点数和边数，然后读取每条边的两个顶点及权重，排序后按权重从小到大的顺序加入最小生成树中。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;
const int mx = 1e5+10;
int fa[mx];
long long sum;
int n,ce,m;
struct aa{
	int x,y,qu;
}p[mx*5];    
void init(){
//	memset(vis, 0, sizeof(vis));
	for(int i = 1; i < mx; i++ )
		fa[i] = i;
	sum=ce=0;
}
bool cmp(aa &a,aa &b){
	return a.qu<b.qu;
}
int find(int x){
	return x == fa[x] ? x : fa[x] = find(fa[x]);
}
 
int main(){
	while(scanf("%d%d",&n,&m)!=EOF){
		init();
		//int a,b,v;
		while(m--){
			scanf("%d%d%d",&p[ce].x,&p[ce].y,&p[ce].qu);
			ce++;
		}
		sort(p,p+ce,cmp);
		int x,y,qu,a,b;
		for(int i = 0; i < ce; i++){
			 x = p[i].x,y = p[i].y, qu = p[i].qu;
			 a=find(x),b=find(y);
			if(a == b) continue;
			else{
				fa[a] = b;
				sum += qu; 
			}
		}
		cout<<sum<<endl; 
		
	}
	return 0;
}