AcWing 859. Kruskal算法求最小生成树

#include <bits/stdc++.h>
using namespace std;
const int N=2e5+10;
int n,m;
int p[N];
struct Node{
    int a,b,w;

    bool operator< (const Node &W)const{
        return w<W.w;
    }
}nodes[N];
int find(int x){
    if(p[x]!=x) p[x]=find(p[x]);
    return p[x];
}
int main(){
    scanf("%d%d",&n,&m);

    for(int i=0;i<m;i++){
        int a,b,w;
        scanf("%d%d%d",&a,&b,&w);
        nodes[i]={a,b,w};
    }
    sort(nodes,nodes+m);
    int res=0,cnt=0;
    for(int i=1;i<=n;i++) p[i]=i;

    for(int i=0;i<m;i++){
        int a=nodes[i].a,b=nodes[i].b,w=nodes[i].w;
        a=find(a),b=find(b);
        if(a!=b){
            p[a]=b;
            res+=w;
            cnt++;
        }
    }
    if(cnt<n-1) printf("impossible");
    else printf("%d",res);
    return 0;
}