并查集模板类

cpp 复制代码
// 并查集模板
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;
    }
};

在这个模板中,有以下几个主要函数:

  1. UnionFind(int _n): 构造函数,用于初始化并查集,参数 _n 表示元素的个数。
  2. findset(int x): 查找元素 x 所属的集合的代表元素,这里使用了路径压缩的优化,可以减小查询时间。
  3. unite(int x, int y): 合并元素 xy 所属的集合,即将它们连接起来。如果 xy 已经在同一个集合中,则返回 false;否则返回 true,表示合并成功。
  4. connected(int x, int y): 判断元素 xy 是否属于同一个集合,即它们是否连接在一起。
cpp 复制代码
//用法
UnionFind uf(n);//初始化一个n个元素的并查集
uf.unite(x,y);//合并x和y这两个元素
uf.connected(x,y);//判断x和y这两个元素是否在同一个连通块
相关推荐
十八岁讨厌编程2 小时前
【算法训练营 · 补充】LeetCode Hot100(上)
算法·leetcode
爱学习的小鱼gogo2 小时前
python 单词搜索(回溯-矩阵-字符串-中等)含源码(二十)
开发语言·数据结构·python·矩阵·字符串·回溯·递归栈
CandyU22 小时前
C++ 学习 —— 02 - 排序算法
c++·学习·排序算法
一条星星鱼2 小时前
从0到1:如何用统计学“看透”不同睡眠PSG数据集的差异(域偏差分析实战)
人工智能·深度学习·算法·概率论·归一化·睡眠psg
浮灯Foden2 小时前
算法-每日一题(DAY18)多数元素
开发语言·数据结构·c++·算法·leetcode·面试
小欣加油2 小时前
leetcode 844 比较含退格的字符串
算法·leetcode·职场和发展
小龙报2 小时前
《算法每日一题(1)--- 第31场蓝桥算法挑战赛》
c语言·开发语言·c++·git·算法·学习方法
llz_1122 小时前
五子棋小游戏
开发语言·c++·算法
liulilittle2 小时前
在 Android Shell 终端上直接运行 OPENPPP2 网关路由配置指南
android·linux·开发语言·网络·c++·编程语言·通信
violet-lz2 小时前
数据结构八大排序:归并排序-原理+C语言实现+优化+面试题
c语言·数据结构·排序算法