最小生成树

#include <cstdio>
#include <cstdlib>
#include<algorithm>
#include <iostream>
using namespace std;
struct edge{
    int s,t,len;
} e[1005];
int vset[1005];
int n,m,ans;
int cmp(edge a,edge b){
    return a.len<b.len;
}
int find(int x){
    if(vset[x]==x){
        return x;
    }
    else{
        return vset[x]=find(vset[x]);
    }
}
void krus(){
    sort(e+1,e+m+1,cmp);
    ans+=e[1].len;
    vset[e[1].t]=vset[e[1].s];
    int j=1;
    //for(int i=1;i<=m;i++){
        //cout<<e[i].s<<' '<<e[i].t<<' '<<e[i].len<<endl;
    //}
    for(int i=2;i<=m&&j<n-1;i++){
        int x=find(e[i].s);int y=find(e[i].t);
        if(x!=y){
            vset[y]=x;
            //cout<<e[i].s<<' '<<e[i].t<<' '<<x<<' '<<vset[x]<<endl;
            j++;
            ans+=e[i].len;
        }
    }
}
int main(){
    scanf("%d%d",&m,&n);
    int x,y,z;
    for(int i=1;i<=m;i++){
        scanf("%d%d%d",&x,&y,&z);
        e[i].s=x;e[i].t=y;e[i].len=z;
    }
    for(int i=1;i<=n;i++)vset[i]=i;
    krus();
    printf("%d",ans);
}