最小生成树-Kruskal-模板-洛谷P3366

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

输入格式
第一行包含两个整数 N,MN,M，表示该图共有 NN 个结点和 MM 条无向边。

接下来 MM 行每行包含三个整数 X_i,Y_i,Z_iX ，表示有一条长度为 Z_iZ
​ 的无向边连接结点 X_i,Y_iX

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

输入输出样例
输入
4 5
1 2 2
1 3 2
1 4 3
2 3 4
3 4 3
输出
7

题解：运用kruskal做法，先sort排序从最小边开始，然后贪心去加到n - 1条边，维护并查集记录连接关系。

#include <iostream>
#include <cstdio>
#include <algorithm>
using namespace std;
#define ll long long
const int N = 200005;

int f[N];
int n,m;
int u,v;
int cnt;
ll ans;
struct node{
	int u,v;
	ll val;
}q[N];

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

int find(int x){
	return f[x] == x ? x : f[x] = find(f[x]);
}

void kru(){
	for(int i = 1; i <= m; i++){
		u = find(q[i].u);
		v = find(q[i].v);
		if(u == v) continue;
		f[u] = v;
		ans += q[i].val;
		cnt++;
		if(cnt == n - 1) break;
	}
}

int main(){
	scanf("%d%d", &n, &m);
	for(int i = 1; i <= n; i++) f[i] = i;
	for(int i = 1; i <= m; i++){
		scanf("%d%d%d", &q[i].u, &q[i].v, &q[i].val);
	}
	sort(q + 1, q + m + 1, cmp);
	kru();
	if(cnt == n - 1){
		printf("%d", ans);
	}else printf("orz");
	return 0;
}
/*
4 5
1 2 2
1 3 2
1 4 3
2 3 4
3 4 3
*/