码迷,mamicode.com
首页 > 编程语言 > 详细

最小生成树算法Kruskal详解

时间:2018-04-06 17:31:36      阅读:230      评论:0      收藏:0      [点我收藏+]

标签:应该   lin   声明   技术分享   ret   view   bool   保留   iostream   

要讲Kruskal,我们先来看下面一组样例。

4 5
1 2 3
1 4 5
2 4 7
2 3 6
3 4 8

 

14

  

技术分享图片

画出来更直观一些,就是上面的这张图。

智商只要不是0的(了解最小生成树是什么的童鞋)应该都知道要选择1<->4, 1<->2, 2<->3三条边那么大家就会问为什么选择这三条呢。

一棵树边的数量等于这棵输的点的数量减1。(不信自己画画试试)这里不再解释。

假设有N个点,那么我们肯定要选择n-1条边来生成一棵树。这棵树就是这张图的生成树。

很显然生成树有一堆。

而最小生成树是指边权之和最小的生成树。那么我们自然而然的想到了贪心!!!

没错,就是贪心,我们按照边的权值进行贪心。

同时为了保证到最后所有的点都连在了一起,并且没有多余的边。我们就要用到并查集。只要两个顶点被一条边所连接。那么他们就有一个共同的祖先。我们在以后就只要判断一下,只要两个顶点没有共同祖先,就可以将他们连入生成树中。当边的条数加到n-1时,我们就可以结束了。

下面是代码

#include <iostream>
#include <cstdio>
#include <algorithm>

using namespace std;

int n, m;
int f[100008];
int sum, tot;
struct edge{
    int u, v, w;
}ed[1000008];

bool cmp(edge a, edge b) {
    return a.w < b.w;
}

int find(int x) {
    if(x == f[x]) return x;
    else return f[x] = find(f[x]);
}

int main() {
    scanf("%d%d", &n, &m);
    for(int i=1; i<=n; i++) {
        f[i] = i;
    }
    for(int i=1; i<=m; i++) {
        scanf("%d%d%d", &ed[i].u, &ed[i].v, &ed[i].w);
    }
    sort(ed+1, ed+1+m, cmp);
    for(int i=1; i<=m; i++) {
        int xx = find(ed[i].u), yy = find(ed[i].v);
        if(xx != yy) {
            f[xx] = find(yy);
            tot++;
            sum += ed[i].w;
        }
        if(tot == n-1) {
            break;
        }
    }
    printf("%d", sum);
}
作者:wlz
本文版权归作者和博客园共有,欢迎转载,但未经作者同意必须保留此段声明,且在文章页面明显位置给出原文连接,否则保留追究法律责任的权利。

最小生成树算法Kruskal详解

标签:应该   lin   声明   技术分享   ret   view   bool   保留   iostream   

原文地址:https://www.cnblogs.com/bljfy/p/8728133.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!