【模板】最小生成树_09

最新推荐文章于 2024-06-12 10:00:23 发布

每天都要写算法（努力版）

最新推荐文章于 2024-06-12 10:00:23 发布

阅读量699

点赞数

分类专栏：图文章标签：算法图论 c++ 最小生成树

本文链接：https://blog.csdn.net/weixin_50512050/article/details/131035115

版权

图专栏收录该内容

9 篇文章 0 订阅

订阅专栏

【模板】最小生成树

题目描述

如题，给出一个无向图，求出最小生成树，如果该图不连通，则输出 orz。

输入格式

第一行包含两个整数 $N, M$ ，表示该图共有 $N$ 个结点和 $M$ 条无向边。

接下来 $M$ 行每行包含三个整数 $X_i,Y_i,Z_i$ ，表示有一条长度为 $Z_i$ 的无向边连接结点 $X_i,Y_i$ 。

输出格式

如果该图连通，则输出一个整数表示最小生成树的各边的长度之和。如果该图不连通则输出 orz。

样例 #1

样例输入 #1

样例输出 #1

提示

数据规模：

对于 $20\%$ 的数据， $N\le 5$ ， $M\le 20$ 。

对于 $40\%$ 的数据， $N\le 50$ ， $M\le 2500$ 。

对于 $70\%$ 的数据， $N\le 500$ ， $M\le 10^4$ 。

对于 $100\%$ 的数据： $1\le N\le 5000$ ， $1\le M\le 2\times 10^5$ ， $1\le Z_i \le 10^4$ 。

所以最小生成树的总边权为 $2 + 2 + 3 = 7$ 。

#include<bits/stdc++.h>
using namespace std;
const int MAXN = 5001;
int height[MAXN];
int father[MAXN];
struct Edge{
	int from;
	int to;
	int length;
};
bool bmp(Edge a,Edge b){
	return a.length < b.length;
}
void init(int n){
	for(int i=0;i<=n;i++){
		height[i] = 0;
		father[i] = i;
	}
}
int find(int x){
	if(x!=father[x]){
		x = find(father[x]);
	}
	return x;
}
void union1(int x,int y){
	int x1 = find(x);
	int y1 = find(y);
	if(x1!=y1){
		if(height[x1]>height[y1]){
			father[y1] = x1;
		}
		else if(height[x1]<height[y1]){
			father[x1] = y1;
		}
		else if(height[x1]==height[y1]){
			father[x1] = y1;
			height[y1]++;
		}
	}
}
Edge edge[MAXN*MAXN];
int Kruscal(int n,int m){
	int result = 0;
	init(n);
	sort(edge,edge+m,bmp);
	for(int i=0;i<m;i++){
		Edge current = edge[i];
		if(find(current.from)!=find(current.to)){
			union1(current.from,current.to);
			result += current.length;
		}
	}
	return result;
}
int main(){
	int n,m;
	cin>>n>>m;
//	int from,to,len;
	int counter = 0;
	for(int i=0;i<m;i++){
		cin>>edge[i].from>>edge[i].to>>edge[i].length;
	}
	int answer = Kruscal(n,m);
	for(int i=1;i<=n;i++){
		if(find(i)==i)
		{
			counter++;
		}
	}
	if(counter > 1){
		cout<<"orz"<<endl;
//		cout<<counter<<endl;
	}
	else{
		cout<<answer<<endl;
	}
	return 0;
}