竹笋

首页 » 问答 » 环境 » 美团实习面试熟悉红黑树是吧能不能写一下
TUhjnbcbe - 2025/7/12 18:14:00

★★★建议星标我们★★★

年Java原创面试题库连载中

Java最全面试题库思维导图

JavaSE系列面试题汇总(共18篇)

JavaWeb系列面试题汇总(共10篇)

JavaEE系列面试题汇总(共13篇)

数据库系列面试题汇总(共6篇)

中间件系列面试题汇总(共3篇)

数据结构与算法面试题汇总(共11篇)

分布式面试题汇总(共10篇)

综合面试题系列汇总(共23篇)

-期汇总(共50篇)

如何应对高并发流量?

StringBuilder线程安全吗?为什么?

Redis的过期键删除策略有哪些?

Spring-Retry重试实现原理是什么?

数据库分库分表之后,如何解决事务问题?

为什么SQL语句不要过多的join?

说说注册中心zookeeper和eureka中的CP和AP

Java中的finally一定会被执行吗?

更多内容,点击上方名片查看

手写红黑树确实有点过分了,但我觉得写不出来也正常,只要理解就行

红黑树是数据结构中比较复杂的一种,最近与它交集颇多,于是花了一周的空闲时间跟它死磕,终于弄明白并实现了红黑树。

写文总结一下,希望能给试图理解红黑树的同学一些灵感,也让我能记得更深刻。

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

红黑树的插入和删除非常复杂,很多人并没有理解或完全实现,或实现了的没有任何注释,让人很难参考;网络上红黑树的理解方式较为单一,一般是双黑、caseN法,而插入和删除的情况很多,每种都有对应的处理方式,如果死记硬背的话,再过一段时间再回忆各种情况可能就一头雾水了。网络上讲红黑树的实现多来源于《算法导论》一书,直接讲红黑树的实现,需要处理颜色和高度两种属性约束,比较晦涩。本文通过红黑树的等同——2-3-4树,避开颜色属性约束,也弱化了高度的影响,以另一种方式去理解红黑树,虽然并不能完全降低它的复杂度,但自认为较之普遍实现,更易记一些。

红黑树

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

节点是红色或黑色。根是黑色。所有叶子都是黑色(叶子是NIL节点)。每个红色节点必须有两个黑色的子节点。(从每个叶子到根的所有路径上不能有两个连续的红色节点。)从任一节点到其每个叶子的所有简单路径都包含相同数目的黑色节点。下图就是一个典型的红黑树:

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

优势和用途

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

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

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

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

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

2-3-4树

2-3-4树是四阶的B树(BalanceTree),它的结构有以下限制:

所有叶子节点都拥有相同的深度。节点只能是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树来讲:

查找最近的叶子结点中的元素替代被删除元素,删除替代元素后,从替代元素所处叶子结点开始处理;降元:4-结点变3-结点,3-结点变2-结点;2-结点中只有一个元素,所以借兄弟结点中的元素来补充删除后的造成的空结点;当兄弟结点中也没有多个元素可以补充时,尝试将父结点降元,失败时向上递归,至到子树降元成功或到root结点树高减1;将这些规则对应到红黑树中即:

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

如上图,删除的要点是找到替代结点,如果替代结点是黑色,递归向上依次判断侄子结点、父结点是否可以补充被删除的黑色,整体思想就是将删除一个黑色结点造成的影响局限在子树内处理。

小结

当然实现过程中调试也占了很大一部分,我使用了两项方法帮助调试:

由于插入多个结点时,无法确定是处理哪个结点时出了问题,于是我给红黑树类添加了debug属性,用二分法设置此属性来找到问题结点;给红黑树类添加了printTree方法,实时打印树结构,确定代码问题再分析;由于红黑树相对其他树实在较为复杂,只通过思考就完全理解不太现实,还需要自己去试着画,试着实现,我画了5张A4纸的正反面才算理解了红黑树,即便如此,在写这篇文章时还发现了代码中的可优化点。

而且代码实现比画图还略复杂,理论中的一个旋转就包含了左旋/右旋/先左旋再右旋/先右旋再左旋几种情况,虽然有一定规律,还是自己实现一下印象最深刻。

来源

1
查看完整版本: 美团实习面试熟悉红黑树是吧能不能写一下