模板题练手速

C++ 代码

#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
const int N=1e5+10;
int n,m;
int p[N];
struct edge{
    int a,b,w;
}e[N];
bool cmp(edge x,edge y)
{
    return x.w<y.w;
}
int find(int x)
{
    if(x!=p[x])p[x]=find(p[x]);
    return p[x];
}

int kruskal(){
    sort(e, e+m,cmp);
    int ans = 0;
    for(int i=1; i<=n; i++) p[i] = i;
    for(int i=0; i<m; i++){
        int a = e[i].a;
        int b = e[i].b;
        if(find(a)!=find(b)){
            p[find(a)] = find(b);
            ans += e[i].w;
        }
    }
    return ans;
}

int main()
{
    cin>>n>>m;
    for(int i=0;i<m;i++)
    {
        int a,b,w;
       scanf("%d%d%d",&a,&b,&w);
        e[i]={a,b,w};
    }
    sort(e,e+m,cmp);
    cout<<kruskal()<<endl;
    return 0;
}