通过2-3-4树理解红黑树

转载

灰色程序 2021-07-17 10:08:46

前言

红黑树是数据结构中比较复杂的一种，最近与它交集颇多，于是花了一周的空闲时间跟它死磕，终于弄明白并实现了红黑树。写文总结一下，希望能给试图理解红黑树的同学一些灵感，也让我能记得更深刻。

在研究红黑树时吃了不少苦头，原因有二：

红黑树的插入和删除非常复杂，很多人并没有理解或完全实现，或实现了的没有任何注释，让人很难参考；
网络上红黑树的理解方式较为单一，一般是 双黑、caseN 法，而插入和删除的情况很多，每种都有对应的处理方式，如果死记硬背的话，再过一段时间再回忆各种情况可能就一头雾水了。

网络上讲红黑树的实现多来源于《算法导论》一书，直接讲红黑树的实现，需要处理颜色和高度两种属性约束，比较晦涩。本文通过红黑树的等同—— 2-3-4树，避开颜色属性约束，也弱化了高度的影响，以另一种方式去理解红黑树，虽然并不能完全降低它的复杂度，但自认为较之普遍实现，更易记一些。

文章最前面先放上红黑树的实现源码，代码在 Github 上，一开始实现时使用我最熟练的 PHP，后续添加了 Java 版，代码都可以直接运行。源码链接：GitHub-枕边书-RBTree，欢迎star。

红黑树

红黑树是一种结点带有颜色属性的二叉查找树，但它在二叉查找树之外，还有以下要求：

下图就是一个典型的红黑树：

但实现上我省略了其中的 Nil 结点，一般如下图，大家理解时也可以忽略它们。

我们知道二叉查找树在不停地添加或删除结点后，可能会导致结点情况如下：

这种情况下，二叉查找树的查找效率最坏会降低为 O(n)。

而红黑树由于在插入和删除结点时都会进行变色旋转等操作，在符合红黑树条件的情况下，即使一边子树全是黑色结点，另一边子树全是红黑相间，两子树的高度差也不会超过一半。一棵有 n 个结点的红黑树高度至多为 2log(n+1)，查找效率最坏为 O(log(n))。

所以红黑树常被用于需求查找效率稳定的场景，如 Linux 中内核使用它管理内存区域对象、Java8 中 HashMap 的实现等，所以了解红黑树也很有意义。

下面介绍一下红黑树的等同 2-3-4树。

2-3-4树

2-3-4树是四阶的 B树(Balance Tree)，它的结构有以下限制：

所有叶子节点都拥有相同的深度。
节点只能是 2-节点、3-节点、4-节点之一。
- 2-节点：包含 1 个元素的节点，有 2 个子节点；
- 3-节点：包含 2 个元素的节点，有 3 个子节点；
- 4-节点：包含 3 个元素的节点，有 4 个子节点；
元素始终保持排序顺序，整体上保持二叉查找树的性质，即父结点大于左子结点，小于右子结点；而且结点有多个元素时，每个元素必须大于它左边的和它的左子树中元素。

下图是一个典型的 2-3-4树（来自维基百科）：

2-3-4树的查询操作像普通的二叉搜索树一样，非常简单，但由于其结点元素数不确定，在一些编程语言中实现起来并不方便，实现一般使用它的等同——红黑树。

至于为什么说红黑树是 2-3-4树的一种等同呢，这是因为 2-3-4树的每一个结点都对应红黑树的一种结构，所以每一棵 2-3-4树也都对应一棵红黑树，下图是 2-3-4树不同结点与红黑树子树的对应。

而上文中的 2-3-4树也可以转换成一棵红黑树：

由红黑树的性质5，和 2-3-4树的性质1，为了便于理解红黑树和 2-3-4树的对应关系，我们可以把红黑树从根结点到叶子结点的黑色结点个数定义为高度。

红黑树和 2-3-4树的结点添加和删除都有一个基本规则：避免子树高度变化，因为无论是 2-3-4树还是红黑树，一旦子树高度有变动，势必会影响其他子树进行调整，所以我们在插入和删除结点时尽量通过子树内部调整来达到平衡，2-3-4树实现平衡是通过结点的旋转和结点元素数变化，红黑树是通过结点旋转和变色。

下面来对照着 2-3-4树说一下红黑树结点的添加和删除：

结点插入

2-3-4树中结点添加需要遵守以下规则：

插入都是向最下面一层插入；
升元：将插入结点由 2-结点升级成 3-结点，或由 3-结点升级成 4-结点；
向 4-结点插入元素后，需要将中间元素提到父结点升元，原结点变成两个 2-结点，再把元素插入 2-结点中，如果父结点也是 4-结点，则递归向上层升元，至到根结点后将树高加1；

而将这些规则对应到红黑树里，就是：

新插入的结点颜色为红色，这样才可能不会对红黑树的高度产生影响。
2-结点对应红黑树中的单个黑色结点，插入时直接成功（对应 2-结点升元）。
3-结点对应红黑树中的黑+红子树，插入后将其修复成 红+黑+红 子树（对应 3-结点升元）；
4-结点对应红黑树中的红+黑+红子树，插入后将其修复成红色祖父+黑色父叔+红色孩子子树，然后再把祖父结点当成新插入的红色结点递归向上层修复，直至修复成功或遇到 root 结点；

如上图所示，虽然向红黑树中插入了一个新结点，但由于旋转和变色，子树的高度保持不变。

删除结点

红黑树的删除要比插入要复杂一些，我们还是类比 2-3-4树来讲：

将这些规则对应到红黑树中即：

查找离当前结点最近的叶子结点作为替代结点（左子树的最右结点或右子树的最左结点都能保证替换后保证二叉查找树的结点的排序性质，叶子结点的替代结点是自身）替换掉被删除结点，从替代的叶子结点向上递归修复；
替代结点颜色为红色（对应 2-3-4树中 4-结点或 3-结点）时删除子结点直接成功；
替代结点为黑色（对应 2-3-4树中 2-结点）时，意味着替代结点所在的子树会降一层，需要依次检验以下三项，以恢复子树高度：
- 兄弟结点的子结点中有红色结点（兄弟结点对应 3-结点或 4-结点）能够“借用”，旋转过来后修正颜色；
- 父结点是红色结点（父结点对应 3-结点或 4-结点，可以降元）时，将父结点变黑色，自身和兄弟结点变红色后删除；
- 父结点和兄弟结点都是黑色时，将子树降一层后把父结点当作替代结点递归向上处理。