牛客小白月赛6 H挖沟（并查集）

题目链接：https://www.nowcoder.com/acm/contest/136/H

#include<bits/stdc++.h>
using namespace std;
const int N=5e5+5;
int n,m,tot;
int p[N];
struct node{
    int u,v,w;
}a[N];
bool cmp(node a,node b){
    return a.w<b.w;
}
int find(int x){
    return x==p[x]?x:p[x]=find(p[x]);
}
int main(){
    scanf("%d%d",&n,&m);
    for(int i=0;i<m;i++){
        scanf("%d%d%d",&a[i].u,&a[i].v,&a[i].w);
    }
    for (int i=1;i<=m;i++) p[i]=i;
    sort(a,a+m,cmp);
    int ans=0;
    for (int i=0;i<m;i++){
        int x=find(a[i].u),y=find(a[i].v);
        if (x!=y){
            ans+=a[i].w;
            p[x]=y;
        }
    }
    printf("%d",ans);
}