数据结构之最小生成树

最新推荐文章于 2024-09-22 17:32:31 发布

旧巷小新

最新推荐文章于 2024-09-22 17:32:31 发布

阅读量178

点赞数

分类专栏：数据结构文章标签：数据结构

本文链接：https://blog.csdn.net/weixin_43518637/article/details/107750000

版权

数据结构专栏收录该内容

18 篇文章 0 订阅

订阅专栏

文章目录

1.Kruskal算法实现
2.Prim算法实现

1.Kruskal算法实现

使用了并查集进行优化查找，使得时间复杂度降低。
核心代码：

class Kruskal {
public:
    vector<int> father;
    int findFather(int x) {
        while (x != father[x]) {
            x = father[x];
        }
        return x;
    }
    int miniSpanningTree(int n, int m, vector<vector<int> >& cost) {
        if (cost.empty()) return 0;
        int res = 0, count = 0;
        sort(cost.begin(), cost.end(), [](vector<int>& u, vector<int>& v) {
            return u[2] < v[2];
        });
        father.resize(n + 1, 0);
        for (int i = 0; i <= n; i++) 
            father[i] = i;
        for (vector<int>& vec : cost) {
            int rx = findFather(vec[0]), ry = findFather(vec[1]);
            if (rx == ry) continue;
            father[rx] = ry;
            res += vec[2];
            count++;
        }
        return count + 1 == n ? res : -1;
    }
};