最小生成树【模板】

最小生成树

题目来源
洛谷P3366 【模板】最小生成树
题目描述
如题，给出一个无向图，求出最小生成树，如果该图不连通，则输出 orz。
输入格式
第一行包含两个整数N，M,表示该图共有N个结点和M条无向边。
接下来M行，每行包含三个整数Xi，Yi，Zi，表示有一条长度为Zi的无向边连接结点Xi，Yi。
输出格式
如果该图连通，则输出一个整数表示最小生成树的各边的长度之和。如果该图不连通则输出 orz。
输入样例
4 5
1 2 2
1 3 2
1 4 3
2 3 4
3 4 3
输出样例
7
说明
数据规模
1<=N<=5000, 1<=M<=2e5;
解题方法
这是一个典型的最小生成树模板题，我的做法是使用普通的kruskal算法解题，其中还需要使用到并查集的知识。

这里简单说一下kruskal算法，这个算法其实是一种贪心思想。现将所有边的权值进行一个从小到大的排列，然后按照顺序进行选取，当给出的两个点不属于同一个集合时，就将他们合并，否则就跳过。直到所有点（假设为n个）都属于同一个集合（这时只会有n-1条边）。至于查找是否属于同一个集合，以及如何将两个点合并为同一个集合，这便涉及到了并查集算法。简而言之，kruskal算法其实就是基于并查集的一种贪心思想。

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N = 2e5+5; 
struct node{
	int s;
	int t;
	ll w;
}a[N];
int fa[N]; 
int n, m, x, y, tot;
ll ans;

void init(int n){
	for(int i=1; i<=n; i++)
		fa[i] = i;
}
int find(int x){
	if(x == fa[x])
		return x;
	else return fa[x] = find(fa[x]);
}

void kruskal(){
	for(int i=1; i<=m; i++){
		x = find(a[i].s);
		y = find(a[i].t);
		if(x==y) continue;
		fa[x] = y;
		ans += a[i].w;
		tot++;
		if(tot==n-1)
			break;
	}
}

bool cmp(node x, node y){
	return x.w < y.w;
}

int main(){
	cin >> n >> m;
	init(n);
	for(int i=1; i<=m; i++)
		cin >> a[i].s >> a[i].t >> a[i].w;
	sort(a+1, a+1+m, cmp);
	kruskal();
	if(tot==n-1) cout << ans;
	else cout << "orz";
	return 0;
}