Kruskal算法

最新推荐文章于 2024-07-06 18:27:39 发布

thessuye

最新推荐文章于 2024-07-06 18:27:39 发布

阅读量994

点赞数

分类专栏：图论文章标签：算法 kruskal

本文链接：https://blog.csdn.net/ACMLCER/article/details/52090980

版权

图论专栏收录该内容

3 篇文章 0 订阅

订阅专栏

算法实现思想及功能
Kruskal算法是求解最小生成树的一种算法。实现思想是首先将边按照权值大小排列，然后从权值最小的边开始加入最小生成树中，但是如果出现新加入的边与原来的图连成环，则不取该边。（这里使用查找根节点的方法实现，如果相同，则成环。如果不同，则加入，并把新加入的边的权值根节点更新。）
代码展示

#include <iostream>
#include <cstring>
#include <string>
#include <cstdio>
#include <algorithm>
using namespace std;
#define MAX 1000
int father[MAX], son[MAX];
int v, l;

typedef struct Kruskal //存储边的信息
{
    int a;
    int b;
    int value;
};

bool cmp(const Kruskal & a, const Kruskal & b)
{
    return a.value < b.value;
}

int unionsearch(int x) //查找根结点+路径压缩
{
    return x == father[x] ? x : unionsearch(father[x]);
}

bool join(int x, int y) //合并
{
    int root1, root2;
    root1 = unionsearch(x);
    root2 = unionsearch(y);
    if(root1 == root2) //为环
        return false;
    else if(son[root1] >= son[root2])   father[root2] = root1;
    else    father[root1] = root2;
    return true;
}

int main()
{
    int ncase, ltotal, sum, flag;
    Kruskal edge[MAX];
    scanf("%d", &ncase);
    while(ncase--)
    {
        scanf("%d%d", &v, &l);
        ltotal = 0, sum = 0, flag = 0;

        for(int i = 1; i <= v; ++i) //初始化
        father[i] = i;

        for(int i = 1; i <= l ; ++i)
        scanf("%d%d%d", &edge[i].a, &edge[i].b, &edge[i].value);

        sort(edge + 1, edge + 1 + l, cmp); //按权值由小到大排序
        for(int i = 1; i <= l; ++i)
        {
            if(join(edge[i].a, edge[i].b))
            {
                ltotal++;               //边数加1
                sum += edge[i].value;   //记录权值之和
                cout<<edge[i].a<<"->"<<edge[i].b<<endl;
            }
            if(ltotal == v - 1)         //最小生成树条件：边数=顶点数-1
            {
                flag = 1;
                break;
            }
        }
        if(flag) printf("%d\n", sum);
        else printf("data error.\n");
    }
    return 0;
}