C++ 红黑树全面详解-CSDN博客

一、引言：红黑树的历史与应用价值

1.1 自平衡二叉树的工业级选择

红黑树由 Rudolf Bayer 于 1972 年提出，后经 Leo J. Guibas 与 Robert Sedgewick 优化，通过颜色约束实现 "黑平衡" 特性，成为平衡二叉树的主流实现。其核心价值在于平衡查找效率与维护成本，在 Linux 内核（进程调度 CFS、内存管理 VMA）、STL 容器（std::map/std::set）、数据库索引等场景广泛应用，GitHub 开源项目中 std::map 的使用率达 68%，印证其在工程领域的核心地位。

1.2 与 AVL 树的辩证关系

维度	红黑树	AVL 树
平衡策略	颜色约束 + 黑高平衡	严格高度平衡（平衡因子≤1）
旋转次数	插入最多 2 次，删除最多 3 次	插入 / 删除最坏 O (log n) 次
查找性能	次优（树高≤2log (n+1)）	最优（树高≤log₂(n+1)）
适用场景	频繁插入删除（如调度队列）	频繁查找（如静态索引）

二、红黑树的核心特性与数学证明

2.1 五大基本性质

节点双色性：每个节点非红即黑
根黑叶黑性：根节点与 NIL 叶子节点必为黑色
红子双黑性：红色节点的子节点必为黑色（无连续红节点）
黑高一致性：任一节点到叶节点的所有路径含相同黑节点数
路径平衡性：最长路径不超过最短路径 2 倍（由性质 3/4 推导）

2.2 树高上限证明

定理：含 n 个内部节点的红黑树树高 h≤2log₂(n+1)

黑高定义：节点到叶节点的黑节点数（不含自身），记根黑高为 bh
推导过程：
1. 根到叶路径至少含 bh 个黑节点，最多含bh 个红节点（性质 3），故树高 h≤2bh
2. 黑高为 bh 的红黑树至少含 2^bh -1 个内部节点（数学归纳法）
3. 由 n≥2^bh -1→bh≤log₂(n+1)→h≤2log₂(n+1)

三、数据结构定义与核心操作

3.1 节点结构设计

cpp

template <typename T>
struct RBNode {
    T data;          // 数据域
    bool is_red;     // 颜色标记（true为红，false为黑）
    RBNode *parent;  // 父节点指针
    RBNode *left;    // 左子节点（默认指向NIL哨兵）
    RBNode *right;   // 右子节点（默认指向NIL哨兵）
    RBNode(const T &val) : data(val), is_red(true), parent(nullptr), left(nullptr), right(nullptr) {}
};

哨兵节点：全局 NIL 节点统一表示叶节点，简化边界条件判断
颜色存储：1bit 标记，较 AVL 树的平衡因子（int）更节省内存

3.2 旋转操作机制

3.2.1 左旋（Left Rotation）

cpp

void left_rotate(RBNode *x) {
    RBNode *y = x->right;        // y为x右子节点
    x->right = y->left;         // x右子树指向y左子树
    if (y->left != NIL) y->left->parent = x;
    y->parent = x->parent;      // y继承x的父节点
    if (x->parent == NIL) root = y;
    else if (x == x->parent->left) x->parent->left = y;
    else x->parent->right = y;
    y->left = x;                // x成为y的左子节点
    x->parent = y;
}

作用：降低右子树高度，提升左子树高度
时间复杂度：O (1)，仅修改指针

3.2.2 右旋（Right Rotation）

对称操作，用于调整左子树过高问题

四、插入算法深度剖析

4.1 插入流程概览

BST 插入：按值插入新节点，初始标记为红色
性质修复：若父节点为黑色直接结束；若父节点为红色触发双红冲突，分三种场景处理

4.2 冲突修复场景

场景 1：叔父节点为红色（ recoloring ）

操作：父节点与叔父节点染黑，祖父节点染红并递归检查
示例：插入值 45 触发双红冲突，修复后祖父节点 50 染红

场景 2：叔父节点为黑色（ LL/RR 型，旋转 + 变色）

操作：父节点染黑→祖父节点染红→祖父节点右旋（LL 型）/ 左旋（RR 型）
示例：插入值 25（LL 型），经旋转后树高降低

场景 3：叔父节点为黑色（ LR/RL 型，双旋转 + 变色）

操作：父节点左旋（LR 型）/ 右旋（RL 型）→转化为场景 2
示例：插入值 35（LR 型），先左旋为 LL 型再按场景 2 处理

4.3 插入复杂度分析

时间复杂度：O (log n)，含 O (log n) 查找 + O (1) 旋转（最多 2 次）
空间复杂度：O (log n)，递归修复最坏深度

五、删除算法与重构策略

5.1 删除核心步骤

BST 删除：找到前驱 / 后继节点替换，删除目标节点
双黑修复：若删除黑色节点，触发 "黑 deficit"，分四种场景处理

5.2 双黑修复场景

场景 1：兄弟节点为红色（旋转 + 变色）

操作：兄弟节点左旋 / 右旋→兄弟染黑，父节点染红→转化为兄弟为黑场景

场景 2：兄弟节点为黑且子节点全黑（变色 + 递归）

操作：兄弟染红→父节点承担双黑→若父节点为红则染黑，否则递归修复

场景 3：兄弟节点为黑且左子为红（LL/RR 型旋转）

操作：兄弟左子染黑→兄弟染父色→父节点染黑→祖父旋转

场景 4：兄弟节点为黑且右子为红（LR/RL 型旋转）

操作：兄弟右子左旋 / 右旋→转化为场景 3

六、STL 中的红黑树实现

6.1 std::map/std::set 底层架构

共性：均基于红黑树，键唯一且有序
差异：
- std::map 存储pair<const Key, T>，支持键值对
- std::set 存储 Key，键即值

6.2 核心源码解析

cpp

// 红黑树节点定义（GCC libstdc++）
template <typename Value>
struct _Rb_tree_node {
    typedef _Rb_tree_node* _Link_type;
    _Link_type _M_parent;  // 父节点
    _Link_type _M_left;    // 左子节点
    _Link_type _M_right;   // 右子节点
    Value _M_value_field;  // 数据域
    char _M_color;         // 颜色（1红0黑）
};

// 插入接口
template <typename Key, typename Value>
pair<iterator, bool> _Rb_tree<Key, Value>::insert_unique(const Value &val) {
    // 1. BST插入逻辑
    // 2. 红黑树性质修复
    // 3. 返回插入位置迭代器
}