c++并查集

最新推荐文章于 2022-12-17 22:14:45 发布

lc520xyp

最新推荐文章于 2022-12-17 22:14:45 发布

阅读量263

点赞数 2

分类专栏： C++ 文章标签： c++

本文链接：https://blog.csdn.net/qq_40967961/article/details/112766171

版权

C++ 专栏收录该内容

2 篇文章 0 订阅

订阅专栏

转载保存c++并查集的思路和代码

// 注意：使用该代码，并不能使得所有的元素都直接指向根节点，仍然存在间接的指向
class UnionFind {
public:
    vector<int> parent;
    vector<int> size;
    int n;
    // 当前连通分量数目
    int setCount;
    
public:
    UnionFind(int _n): n(_n), setCount(_n), parent(_n), size(_n, 1) {
        iota(parent.begin(), parent.end(), 0);
    }
    
    int findset(int x) {
        return parent[x] == x ? x : parent[x] = findset(parent[x]);
    }
    
    bool unite(int x, int y) {
        x = findset(x);
        y = findset(y);
        if (x == y) {
            return false;
        }
        if (size[x] < size[y]) {
            swap(x, y);
        }
        parent[y] = x;
        size[x] += size[y];
        --setCount;
        return true;
    }
    
    bool connected(int x, int y) {
        x = findset(x);
        y = findset(y);
        return x == y;
    }
};