最小生成树--kruskal

ling_fly

于 2017-08-20 13:23:35 发布

阅读量266

点赞数 1

分类专栏：图论

本文链接：https://blog.csdn.net/qq_36392612/article/details/77428048

版权

图论专栏收录该内容

3 篇文章 0 订阅

订阅专栏

kruskal算法用于求稀疏图的最小生成树

图中边权最小的边一定属于某一颗最小生成树，所以先对所有的边按边权由小到大排序，在将边加入到生成树中。

加入过程中用并查集检查边是否形成环

传送门：hihoCode：1098

代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define maxn  1000010
#define maxm  10010
typedef long long LL;
using namespace std;
struct node{
    int u,v,w;
}p[maxn];
int fa[maxn];
bool cmp(node a,node b){
    return a.w<b.w;
}
int find(int x);
bool uni(int x,int y);
int main() {
    //freopen("in.txt","r",stdin);
    //cin.tie(0);
    //cin.sync_with_stdio(false);
    int n,m,i;
    scanf("%d %d",&n,&m);
    for(i=0;i<m;i++)
        scanf("%d %d %d",&p[i].u,&p[i].v,&p[i].w);
    sort(p,p+m,cmp);
    
    for(i=0;i<n;i++)
        fa[i]=i;
    int cnt=0,sum=0;
    for(i=0;i<m;i++){
        if(uni(p[i].u,p[i].v)){
            //printf("u=%d,v=%d,w=%d\n",p[i].u,p[i].v,p[i].w);
            cnt++;
            sum+=p[i].w;
        }
        if(cnt==n-1)break;
    }
    printf("%d\n",sum);
    return 0;
}
int find(int x){
    if(fa[x]==x)return x;
    return fa[x]=find(fa[x]);
}
bool uni(int x,int y){
    int fx=find(x);
    int fy=find(y);
    if(fx!=fy){
        fa[fx]=fy;
        return true;
    }
    else return false;
}