C++ STL中红黑树实现set/map的底层原理

1. 红黑树与STL容器：set/map的底层实现剖析

在C++标准模板库(STL)中，set和map是两种非常重要的关联容器，它们的高效性能很大程度上依赖于其底层实现——红黑树。红黑树是一种自平衡的二叉搜索树，它通过特定的规则确保树的高度始终保持在O(log n)级别，从而保证了各种操作的时间复杂度都是对数级别的。

1.1 红黑树的基本特性

红黑树必须满足以下五个基本性质：

1. 每个节点要么是红色，要么是黑色
2. 根节点是黑色
3. 每个叶子节点（NIL节点）是黑色
4. 如果一个节点是红色的，则它的两个子节点都是黑色的
5. 对于每个节点，从该节点到其所有后代叶子节点的简单路径上，均包含相同数目的黑色节点

这些性质保证了红黑树的关键特性：从根到最远叶子节点的路径长度不会超过从根到最近叶子节点路径长度的两倍。

1.2 STL中set/map的实现架构

在SGI STL的实现中，set和map的核心实现分布在几个关键头文件中：

• stl_tree.h：包含红黑树的核心实现
• stl_set.h和stl_multiset.h：set和multiset的实现
• stl_map.h和stl_multimap.h：map和multimap的实现

这种架构设计体现了良好的代码复用性，通过模板技术，同一套红黑树实现可以同时支持set和map两种容器。

2. 红黑树的泛型设计思想

2.1 模板参数的设计

红黑树的泛型设计是其能够同时支持set和map的关键。观察STL源码，我们可以发现红黑树的类模板声明如下：

cpp复制template <class Key, class Value, class KeyOfValue, class Compare, class Alloc = alloc>
class rb_tree;

这里有几个关键的模板参数：

1. Key：键的类型，用于查找、删除等操作
2. Value：节点实际存储的数据类型
3. KeyOfValue：从Value中提取Key的仿函数
4. Compare：键的比较函数
5. Alloc：内存分配器

这种设计使得红黑树可以灵活地适配不同的使用场景：

• 对于set：Value就是Key类型
• 对于map：Value是pair<const Key, T>类型

2.2 set和map对红黑树的实例化差异

set和map在实例化红黑树时的主要区别体现在模板参数上：

set的实例化：

cpp复制template <class Key, class Compare = less<Key>, class Alloc = alloc>
class set {
private:
    typedef rb_tree<key_type, value_type, 
                   identity<value_type>, key_compare, Alloc> rep_type;
    rep_type t;  // 红黑树实例
};

map的实例化：

cpp复制template <class Key, class T, class Compare = less<Key>, class Alloc = alloc>
class map {
private:
    typedef rb_tree<key_type, value_type,
                   select1st<value_type>, key_compare, Alloc> rep_type;
    rep_type t;  // 红黑树实例
};

关键区别在于：

1. set的Value类型就是Key类型，而map的Value类型是pair<const Key, T>
2. set使用identity从值中提取键，map使用select1st从键值对中提取键

3. 红黑树节点的核心结构

3.1 节点基类

红黑树节点的实现分为基类和派生类两个部分。基类主要包含树的连接信息和颜色标记：

cpp复制struct __rb_tree_node_base {
    typedef __rb_tree_color_type color_type;