温馨提示×

温馨提示×

您好,登录后才能下订单哦!

密码登录×
登录注册×
其他方式登录
点击 登录注册 即表示同意《亿速云用户服务条款》

C++ set在并查集优化中的应用

发布时间:2024-08-05 15:52:09 来源:亿速云 阅读:79 作者:小樊 栏目:编程语言

在并查集(Disjoint Set Union)的优化中,可以使用C++的set容器来实现路径压缩和按秩合并。路径压缩是指在查找根节点的过程中,将沿途的节点的父节点直接设为根节点,以减少查找路径的长度。按秩合并是指将两个集合合并时,将rank较小的树作为rank较大的树的子树,以减少树的深度。

下面是一个使用C++的set容器实现并查集优化的示例代码:

#include <iostream>
#include <set>
#include <vector>

using namespace std;

class UnionFind {
public:
    vector<int> parent;
    vector<int> rank;

    UnionFind(int n) {
        parent.resize(n);
        rank.resize(n, 0);
        for (int i = 0; i < n; i++) {
            parent[i] = i;
        }
    }

    int find(int x) {
        if (parent[x] != x) {
            parent[x] = find(parent[x]);
        }
        return parent[x];
    }

    void unite(int x, int y) {
        int rootX = find(x);
        int rootY = find(y);

        if (rootX != rootY) {
            if (rank[rootX] < rank[rootY]) {
                parent[rootX] = rootY;
            } else if (rank[rootX] > rank[rootY]) {
                parent[rootY] = rootX;
            } else {
                parent[rootY] = rootX;
                rank[rootX]++;
            }
        }
    }
};

int main() {
    UnionFind uf(5);

    uf.unite(0, 1);
    uf.unite(1, 2);
    uf.unite(3, 4);

    cout << uf.find(0) << endl; // output: 1
    cout << uf.find(2) << endl; // output: 1
    cout << uf.find(3) << endl; // output: 4

    return 0;
}

在上面的示例代码中,我们使用了C++的set容器来存储父节点,并实现了路径压缩和按秩合并。通过这种方式,可以优化并查集的性能,使其在实际应用中更加高效。

向AI问一下细节

免责声明:本站发布的内容(图片、视频和文字)以原创、转载和分享为主,文章观点不代表本网站立场,如果涉及侵权请联系站长邮箱:is@yisu.com进行举报,并提供相关证据,一经查实,将立刻删除涉嫌侵权内容。

c++
AI