写模板， 并查集。

合集 - 3.26~4.13 蓝桥杯省赛备赛(28)

1.P8687 [蓝桥杯 2019 省 A] 糖果2024-03-252.洛谷 P1115 最大子段和2024-03-25

3.写模板， 并查集。2024-03-25

4.洛谷 P1656 炸铁路2024-03-265.写模板，树状数组。2024-03-266.洛谷 P3374 【模板】树状数组 12024-03-267.洛谷 P3368 【模板】树状数组 22024-03-268.洛谷 P9237 [蓝桥杯 2023 省 A] 像素放置2024-03-269.洛谷 P5937 [CEOI1999] Parity Game2024-03-2710.洛谷 P8306 【模板】字典树2024-03-2711.写模板，kmp。2024-03-2712.写模板， LCA。2024-03-2813.洛谷 P1967 [NOIP2013 提高组] 货车运输2024-03-2814.写模板，线段树2024-03-2815.蓝桥杯-百亿富翁2024-03-2916.洛谷 P3372 【模板】线段树 12024-03-2917.模拟比赛-14届研究生组C++省赛2024-04-0118.写模板， 线性筛2024-04-0119.写模板-质数的简单应用2024-04-0220.洛谷 P1196 [NOI2002] 银河英雄传说2024-04-0421.洛谷 P1004 [NOIP2000 提高组] 方格取数2024-04-0422.蓝桥杯 计算方程2024-04-0423.洛谷 P1006 [NOIP2008 提高组] 传纸条2024-04-0524.蓝桥杯-算法赛第9场强者：贝贝的2.02024-04-0825.P1002 [NOIP2002 普及组] 过河卒2024-04-0826.蓝桥杯，推导部分和2024-04-0827.牛客，小红不想做完全背包2024-04-0928.蓝桥杯-岛屿个数2024-04-10

收起

1： 实现路径压缩并查集，查找，合并，判同集合。
2： 获取集合大小，集合数量。
3： 增加权重，查询单个元素到集合代表元素的权重。
4： 增加获取集合中元素的功能。

以C++的三大特性之一封装来写该模块，代码量大概是90行左右。

用到的库： vector, nnumeric(iota函数)。

class DisjointSet{
public:
    DisjointSet(int sz): sz_(sz), num_sets_(sz){
        fa_.resize(sz_);
        iota(fa_.begin(), fa_.end(), 0);

        assert(use_dist_ == use_set_size_ || (use_set_size_ == true));  //节点合并时dist_的值更新必须用到set_size_,而set_size_不依赖dist_
        if (use_set_size_){
            set_size_.resize(sz_);
            fill(set_size_.begin(), set_size_.end(), 1);
        }
        if (use_dist_){
            dist_.resize(sz_);
        }
        if (use_elements_){
            elements_.resize(sz_);
            for (int i = 0; i < sz_; ++i){
                elements_[i].emplace_back(i);
            }
        }
    }

    int findSet(int x){
        if (fa_[x] == x){
            return x;
        }
        else{
            if (use_dist_){
                updateDist(x);
                return fa_[x];
            }
            return fa_[x] = findSet(fa_[x]);
        }
    }

    int getSetSize(int x){
        assert(use_set_size_ == true);
        return set_size_[findSet(x)];
    }

    int getDist(int x){
        assert(use_dist_ == true);
        updateDist(x);
        return dist_[x];
    }

    int countSets(){
        return num_sets_;
    }

    vector<int> getSetElements(int x){
        assert(use_elements_ == true);
        return elements_[findSet(x)];
    }

    bool isSameSet(int x, int y){
        return findSet(x) == findSet(y);
    }

    bool unionSet(int x, int y){
        x = findSet(x);
        y = findSet(y);
        if (x == y){
            return false;
        }
        fa_[x] = y;
        num_sets_ --;
        if (use_elements_){
            elements_[y].insert(elements_[y].end(), elements_[x].begin(), elements_[x].end());
            elements_[x].clear();
        }
        if (use_dist_){
            dist_[x] = set_size_[y];
        }
        if (use_set_size_){
            set_size_[y] += set_size_[x];
        }
        return true;
    }

private:
    vector<int> fa_;
    int sz_, num_sets_;


    vector<int> set_size_;
    vector<int> dist_;
    vector<vector<int>> elements_;

    bool use_dist_ = false;
    bool use_set_size_ = true;
    bool use_elements_ = false;


    void updateDist(int x){
        if (fa_[x] == x){
            return;
        }
        int par = fa_[x];
        fa_[x] = findSet(par);
        dist_[x] += dist_[par];
    }

};