在C++中,map 和 set 的底层数据结构是通过红黑树实现的。红黑树是一种自平衡的二叉搜索树,能够在插入、删除和查找操作中保持对数时间复杂度,从而保证高效的数据访问。
红黑树满足以下五条性质,确保树的高度始终保持在 log(n) 级别:
这些性质共同作用,使得红黑树在动态操作中能够自动调整结构,避免退化为链表。
要手动实现一个红黑树,需要定义节点结构、基本操作(插入、删除、旋转)以及修复机制来维护红黑性质。
1. 节点结构定义:
每个节点包含值、颜色、左右子节点指针和父节点指针。
enum Color { RED, BLACK };
template
struct RBNode {
T value;
Color color;
RBNode left;
RBNode right;
RBNode* parent;
RBNode(const T& val) : value(val), color(RED), left(nullptr), right(nullptr), parent(nullptr) {}};
2. 插入后的修复操作:
插入新节点后,默认设为红色。若破坏了红黑性质(如父节点也为红色),需通过变色和旋转来修复。
主要情况包括:
3. 左旋与右旋:
旋转是调整树结构的核心手段。
void rotateLeft(RBNode*& root, RBNode* x) {
RBNode* y = x->right;
x->right = y->left;
if (y->left != nullptr)
y->left->parent = x;
y->parent = x->parent;
if (x->parent == nullptr)
root = y;
else if (x == x->parent->left)
x->parent->left = y;
else
x->parent->right = y;
y->left = x;
x->parent = y;
}
右旋类似,方向相反。
C++ STL 中的 std::map 和 std::set 通常基于红黑树实现(具体由编译器决定,但主流实现如GCC的libstdc++确实如此)。
它们不支持随机访问,但提供有序遍历。由于红黑树的平衡性,性能稳定,适合频繁增删查的场景。
例如:
#include
这些操作背后都会触发红黑树的插入逻辑,并自动完成平衡调整。
相比AVL树,红黑树牺牲了一定的平衡性(最大高度略高),但换来更少的旋转操作,更适合频繁插入删除的场景。
基本上就这些。理解红黑树有助于深入掌握STL容器的行为特性,尤其是在分析性能瓶颈时更有底气。