练习37，最小生成树MST【Kruskal算法】

P3366 【模板】最小生成树 - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)

模板题

#include<bits/stdc++.h>
using namespace std;
#define INF 0x3f3f3f3f
#define NOTLE ios::sync_with_stdio(0),cin.tie(0),cout.tie(0)
#define endl '\n'
#define T int TT;cin >> TT;while (TT--)
#define lint long long
#define pb push_back
#define eb emplace_back
int n,m,ans=0,cnt=0;
int fa[5010];
struct node{
	int x,y,w;
} edge[200010];
int cmp(node e1,node e2){
	return e1.w<e2.w;
}

//并查集
void init(){ //初始化
    for(int i=1;i<=n;i++) fa[i]=i;
}
int find(int x){ //寻找根节点
    if(fa[x]!=x) fa[x]=find(fa[x]);
    return fa[x];
}
void unity(int x,int y){ //合并集合，将x集根节点并入y集根节点
    fa[find(x)]=find(y);
}

void kruskal(){
	for(int i=1;i<=m;i++){
		if(find(edge[i].x)==find(edge[i].y)) continue;
		unity(edge[i].x,edge[i].y);
		ans+=edge[i].w;
		cnt++;
		if(cnt==n-1) break; //n个点的图MST有n-1条边
	}
}
int main(){
    NOTLE;
    cin >> n >> m;
    for(int i=1;i<=m;i++)
    	cin >> edge[i].x >> edge[i].y >> edge[i].w;
    sort(edge+1,edge+1+m,cmp);
    init();
    kruskal();
    if(cnt==n-1) cout << ans << endl;
    else cout << "orz" << endl;
    return 0;
}