codevs 1231 最优布线问题

Kruskal算法构建最小生成树

最新推荐文章于 2019-04-26 17:46:07 发布

原创最新推荐文章于 2019-04-26 17:46:07 发布 · 231 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#kruskal

最小生成树专栏收录该内容

3 篇文章

订阅专栏

本文介绍了一个使用Kruskal算法解决最小生成树问题的C++实现案例。通过输入顶点数和边数，读取每条边的两个顶点及权重，并将边按权重从小到大排序，避免形成环路的同时选择权值最小的边加入生成树中，直至形成最小生成树。

题目链接
最小生成树模板题。下面是用kruskal：

#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
const int N=1e5+7;
struct node{
    int u,v,w;
    friend bool operator<(node a,node b){return a.w<b.w;}
};
int fa[N],n,m;long long ans;vector<node>e;
int find(int x){return fa[x]==x?x:fa[x]=find(fa[x]);}
void kruskal(){
    sort(e.begin(),e.end());
    for(int i=1;i<=n;i++)fa[i]=i;
    for(int i=0;i<e.size();i++){
        int p=e[i].u,q=e[i].v;
        if(find(p)==find(q))continue;
        fa[find(p)]=find(q);
        ans+=e[i].w;
    }
    cout<<ans<<endl;
}
int main(){
    cin>>n>>m;
    for(int i=1;i<=m;i++){
        node a;cin>>a.u>>a.v>>a.w;
        e.push_back(a);
    }
    kruskal();
}