第三次CCF软件能力认证D题-最优灌溉 题型：最小生成树模板题

3210. 最优灌溉 - AcWing题库

#include<bits/stdc++.h>
using namespace std;
const int N=1e5+10;
int p[N];
int n,m;

struct Edge
{
    int a,b,c;
    bool operator <(const Edge& W)const
    {
        return c<W.c;
    }
}edge[N];

int find(int x)
{
    if(p[x]!=x)p[x]=find(p[x]);
return p[x];
}

int main()
{
    cin>>n>>m;
    
    for(int i=1;i<=n;i++)p[i]=i;   //初始化一下并查集数组
    for(int i=0;i<m;i++)
    {
        int a,b,c;cin>>a>>b>>c;
        edge[i]={a,b,c};
    }
    
    sort(edge,edge+m);   //给边排个序
    
    int res=0;
    for(int i=0;i<m;i++)
    {
        int a=edge[i].a,b=edge[i].b,c=edge[i].c;
        a=find(a),b=find(b);
        if(a!=b)
        {
            p[a]=b;   //合并a,b所在的集合
            res+=c;   //res存储最小生成树的所有树边的权值和
        }
    }
    
    cout<<res<<endl;
    return 0;
}