kruskal算法

上一节课我们学习了并查集

今天我们学习的kruskal算法就是依赖于并查集的。

看一看这个图

首先将边权从小到大排序

然后选择边权最小的边

如果出现回路，就不选(用并查集的判断在同一个团伙)

代码

#include <iostream>
#include <cstdio>
using namespace std;

int fa[1000010];

int find(int x)
{
    return fa[x] == x ? x : fa[x] = find(fa[x]);
}

struct Edge
{
    int a, b, c;
    bool operator < (const Edge & cmp2) const
    {
        return c < cmp2.c;
    }
}
a[1000010];

int main()
{
    int n, m;
    scanf("%d%d", &n, &m);
    for(int i = 0; i < m; i ++)
    {
        scanf("%d%d%d", &edge[i].a, &edge[i].b, &edge[i].c);
    }
    sort(edge, edge + m);
    for(int i = 1; i <= n; i ++)
    {
        fa[i] = i;
    }
    int ret = 0;
    int cnt = 0;
    for(int i = 0; i < m; i ++)
    {
        int a = edge[i].a, b = edge[i].b, c = edge[i].c;
        int ta = find(a);
        int tb = find(b);
        if(ta != tb)
        {
            cnt ++;
            ret += c;
        }
    }
    if(cnt != n - 1)
    {
        puts("-1");
    }
    else
    {
        printf("%d\n", ret);
    }
    return 0;
}