第三次CCF计算机软件能力认证 最优灌溉 (Kruskal求最小生成树)

CCF官网评测

分析

Kruskal求最小生成树。

C++ 代码

#include<bits/stdc++.h>
using namespace std;
const int N= 1e5+10;
struct node{
    int a,b,w;
    
    bool operator<(const node &p) const{    //按边权从小到大排序
        return w<p.w;
    }
}edge[N];
int fa[N],n,m;
int find(int x)
{
    if(fa[x]!=x) fa[x]=find(fa[x]);
    return fa[x];
}
int kruscal()   //求最小生成树
{
    int ans=0;
    for(int i=0;i<m;i++)
    {
        int a=edge[i].a,b=edge[i].b;
        int faa=find(a),fbb=find(b);
        if(faa!=fbb)
        {
            ans+=edge[i].w;
            fa[faa]=fbb;
        }
    }
    return ans;
}
int main()
{
    cin>>n>>m;
    for(int i=1;i<=n;i++) fa[i]=i;
    for(int i=0;i<m;i++) cin>>edge[i].a>>edge[i].b>>edge[i].w;
    sort(edge,edge+m);
    cout<<kruscal();
    
    return 0;
}