【模板】最小生成树

最新推荐文章于 2024-03-24 14:27:30 发布

linjiayina

最新推荐文章于 2024-03-24 14:27:30 发布

阅读量210

点赞数

CC 4.0 BY-SA版权

分类专栏：算法 # 图论

本文链接：https://blog.csdn.net/linjiayina/article/details/113091648

算法同时被 2 个专栏收录

226 篇文章

订阅专栏

图论

31 篇文章

订阅专栏

这篇博客介绍了如何运用kruskal算法解决图的最小生成树问题。通过输入无向图的节点数和边数，以及边的权重，程序会计算并输出最小生成树的总权重。如果图不连通，程序将输出'orz'。示例展示了对于给定的一组边，如何通过排序和并查集实现kruskal算法。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目链接

题目描述

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

输入格式

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

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

输出格式

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

输入输出样例

输入

4 5
1 2 2
1 3 2
1 4 3
2 3 4
3 4 3

输出

#include<bits/stdc++.h>
using namespace std;
const int maxn = 2e5 + 10;
int vis[maxn], cnt, ans;

struct node{
	int x, y, z; 
}f[maxn];

bool cmp(node a, node b)
{
	return a.z < b.z;
}

int find(int x)
{
	if(vis[x] == x) return x;
	vis[x] = find(vis[x]);
	return vis[x];
}

bool merge(int a, int b)
{
	int p = find(a);
	int q = find(b);
	if(p == q) return false;
	else{
		vis[p] = q;
		return true;
	}
}

int main()
{
	int n, m;
	cin >> n >> m;
	for(int i = 0; i < n; i++){
		vis[i] = i;
	}
	for(int i = 0; i < m; i++){
		cin >> f[i].x >> f[i].y >> f[i].z;
	}
	sort(f, f + m, cmp);
	cnt = 0, ans = 0;
	for(int i = 0; i < m; i++){
		if(merge(f[i].x, f[i].y)){
			ans += f[i].z;
			cnt++;
		}
		if(cnt == n - 1) break;
	}
	if(cnt != n - 1) cout << "orz" << endl;
	else cout << ans << endl;
	return 0;
}