【模板】最小生成树

题目链接

题目描述

如题，给出一个无向图，求出最小生成树，如果该图不连通，则输出 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

输出

7

#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;
}