一篇文章教你透彻了解红黑树
Posted knowalker
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了一篇文章教你透彻了解红黑树相关的知识,希望对你有一定的参考价值。
cnblogs.com/skywang12345/p/3245399.html
参考了《算法导论》中红黑树相关章节的内容(文章较长,慎点)
本文以图文并茂的形式对红黑树进行了详细剖析,涉及的主要内容包括:红黑树的特性,红黑树的时间复杂度以及它的证明,红黑树的左旋、右旋、插入、删除等操作。
R-B Tree简介
R-B Tree,全称是Red-Black Tree,又称为“红黑树”,它是一种特殊的二叉查找树。红黑树的每个节点上都有存储位表示节点的颜色,可以是红(Red)或黑(Black)。红黑树的特性:
每个节点或者是黑色,或者是红色。
根节点必须是黑色。
每个叶子节点(NIL)是黑色。 [注意:这里叶子节点,是指为空(NIL或NULL)的叶子节点!]
如果一个节点是红色的,则它的子节点必须是黑色的。
从一个节点到该节点的子孙节点的所有路径上包含相同数目的黑节点。
注意:
特性(3)中的叶子节点,是只为空(NIL或null)的节点。
特性(5),确保没有一条路径会比其他路径长出俩倍。因而,红黑树是相对接近平衡的二叉树。
红黑树示意图如下:
红黑树的应用
红黑树的应用比较广泛,主要是用它来存储有序的数据,它的时间复杂度是O(lgn),效率非常之高。例如,Java集合中的TreeSet和TreeMap,C++ STL中的set、map,以及Linux虚拟内存的管理,都是通过红黑树去实现的。
红黑树的时间复杂度和相关证明
红黑树的时间复杂度为: O(lgn),下面通过“数学归纳法”对红黑树的时间复杂度进行证明。
定理:一棵含有n个节点的红黑树的高度至多为2log(n+1)。
证明:"一棵含有n个节点的红黑树的高度至多为2log(n+1)" 的逆否命题是"高度为h的红黑树,它所包含的内节点个数至少为2^(h/2)-1个"。
我们只需要证明逆否命题,即可证明原命题为真;即只需证明"高度为h的红黑树,它所包含的内节点个数至少为2^(h/2)-1个"。
从某个节点x出发(不包括该节点)到达一个叶节点的任意一条路径上,黑色节点的个数称为该节点的黑高度(x's black height),记为bh(x)。关于bh(x)有两点需要说明:
第1点:根据红黑树的"特性(5),即从一个节点到该节点的子孙节点的所有路径上包含相同数目的黑节点 "可知,从节点x出发到达所有的叶节点具有相同数目的黑节点。这也就意味着,bh(x)的值是唯一的。
第2点:根据红黑色的"特性(4),即 如果一个节点是红色的,则它的子节点必须是黑色的"可知,从节点x出发达到叶节点"所经历的黑节点数目">= "所经历的红节点的数目"。假设x是根节点,则可以得出结论" bh(x) >= h/2 "。进而,我们只需证明"高度为h的红黑树,它所包含的内节点个数至少为 2^bh(x) -1个"即可。
到这里,我们将需要证明的定理已经由"一棵含有n个节点的红黑树的高度至多为2log(n+1)"转变成只需要证明"高度为h的红黑树,它所包含的内节点个数至少为 2^bh(x) -1个"。
下面通过"数学归纳法"开始论证高度为h的红黑树,它所包含的内节点个数至少为 2^bh(x) -1个"。
当树的高度 h=0 时,内节点个数是0,bh(x) 为 0,2^bh(x) -1 也为 0。显然,原命题成立。
当 h>0,且树的高度为 h-1 时,它包含的节点个数至少为 2^bh(x)-1 -1。这是根据(1)推断出来的。
下面,由树的高度为 h-1 的已知条件推出“树的高度为 h 时,它所包含的节点树为 2^bh(x) -1”。
当树的高度为h时,对于节点x(x为根节点),其黑高度为 bh(x);对于节点x的左右子树,它们黑高度为 bh(x) 或者 bh(x)-1。根据(2)的已知条件,我们已知"x的左右子树,即高度为 h-1 的节点,它包含的节点至少为 2^bh(x)-1 -1 个";
所以,节点x所包含的节点至少为 (2^bh(x)-1 -1) + (2^bh(x)-1 -1) + 1 = 2^bh(x) -1。即节点x所包含的节点至少为 2^bh(x) -1。
因此,原命题成立。
由(1)、(2)得出,"高度为h的红黑树,它所包含的内节点个数至少为 2^bh(x) -1个"。因此,“一棵含有n个节点的红黑树的高度至多为 2log(n+1)”。
红黑树的基本操作(一) 左旋和右旋
红黑树的基本操作是添加、删除 。在对红黑树进行添加或删除之后,都会用到旋转方法。为什么呢?道理很简单,添加或删除红黑树中的节点之后,红黑树就发生了变化,可能不满足红黑树的5条性质,也就不再是一颗红黑树了,而是一颗普通的树。而通过旋转,可以使这颗树重新成为红黑树。简单点说,旋转的目的是让树保持红黑树的特性。
旋转包括两种:左旋和右旋 。下面分别对它们进行介绍。
1. 左旋
对x进行左旋,意味着"将x变成一个左节点"。左旋的伪代码来自《算法导论》,参考上面的示意图和下面的伪代码,理解“红黑树T的节点x进行左旋”是如何进行的。
LEFT-ROTATE(T, x)
01 y = x.right
02 x.right = y.left
03 y.left.p = x
04 y.p = x.p
05 if x.p == T.nil
06 T.root = y
07 else if x == x.p.left
08 x.p.left = y
09 else x.p.right = y
10 y.left = x
11 x.p = y
理解左旋之后,看看下面一个更鲜明的例子。你可以先不看右边的结果,自己尝试一下。
2. 右旋理解左旋之后,看看下面一个更鲜明的例子。你可以先不看右边的结果,自己尝试一下。
对y进行右旋,意味着"将y变成一个右节点"。右旋的伪代码来自《算法导论》,参考上面的示意图和下面的伪代码,理解“红黑树T的节点y进行右旋”是如何进行的。
RIGHT-ROTATE(T, y)
01 x = y.left
02 y.left = x.right
03 x.right.p = y
04 x.p = y.p
05 if y.p == T.nil
06 T.root = x
07 else if y == y.p.right
08 y.p.right = x
09 else y.p.left = x
10 x.right = y
11 y.p = x
理解右旋之后,看看下面一个更鲜明的例子。你可以先不看右边的结果,自己尝试一下。
旋转总结 :
左旋和右旋是相对的两个概念,原理类似。理解一个也就理解了另一个。
下面谈谈如何区分左旋和右旋。 在实际应用中,若没有彻底理解左旋和右旋,可能会将它们混淆。下面谈谈我对如何区分左旋和右旋的理解。
3. 区分左旋和右旋
仔细观察上面"左旋"和"右旋"的示意图。我们能清晰的发现,它们是对称的。无论是左旋还是右旋,被旋转的树,在旋转前是二叉查找树,并且旋转之后仍然是一颗二叉查找树。
左旋示例图 (以x为节点进行左旋):
z
x /
/ --(左旋)--> x
y z /
y
对x进行左旋,意味着,将“x的右孩子”设为“x的父亲节点”;即,将 x变成了一个左节点(x成了为z的左孩子)!。 因此, 左旋中的“左”,意味着“被旋转的节点将变成一个左节点” 。
右旋示例图 (以x为节点进行右旋):
y
x
/ --(右旋)--> x
y z
z
对x进行右旋,意味着,将“x的左孩子”设为“x的父亲节点”;即,将 x变成了一个右节点(x成了为y的右孩子)! 因此, 右旋中的“右”,意味着“被旋转的节点将变成一个右节点” 。
红黑树的基本操作(二) 添加
将一个节点插入到红黑树中,需要执行哪些步骤呢?首先,将红黑树当作一颗二叉查找树,将节点插入;然后,将节点着色为红色;最后,通过旋转和重新着色等方法来修正该树,使之重新成为一颗红黑树。详细描述如下:
第一步: 将红黑树当作一颗二叉查找树,将节点插入。
红黑树本身就是一颗二叉查找树,将节点插入后,该树仍然是一颗二叉查找树。也就意味着,树的键值仍然是有序的。此外,无论是左旋还是右旋,若旋转之前这棵树是二叉查找树,旋转之后它一定还是二叉查找树。这也就意味着,任何的旋转和重新着色操作,都不会改变它仍然是一颗二叉查找树的事实。
好吧?那接下来,我们就来想方设法的旋转以及重新着色,使这颗树重新成为红黑树!
第二步:将插入的节点着色为"红色"。
为什么着色成红色,而不是黑色呢?为什么呢?在回答之前,我们需要重新温习一下红黑树的特性:
每个节点或者是黑色,或者是红色。
根节点是黑色。
每个叶子节点是黑色。 [注意:这里叶子节点,是指为空的叶子节点!]
如果一个节点是红色的,则它的子节点必须是黑色的。
从一个节点到该节点的子孙节点的所有路径上包含相同数目的黑节点。
将插入的节点着色为红色,不会违背"特性(5)"!少违背一条特性,就意味着我们需要处理的情况越少。接下来,就要努力的让这棵树满足其它性质即可;满足了的话,它就又是一颗红黑树了。
第三步: 通过一系列的旋转或着色等操作,使之重新成为一颗红黑树。
第二步中,将插入节点着色为"红色"之后,不会违背"特性(5)"。那它到底会违背哪些特性呢?
对于"特性(1)",显然不会违背了。因为我们已经将它涂成红色了。
对于"特性(2)",显然也不会违背。在第一步中,我们是将红黑树当作二叉查找树,然后执行的插入操作。而根据二叉查找数的特点,插入操作不会改变根节点。所以,根节点仍然是黑色。
对于"特性(3)",显然不会违背了。这里的叶子节点是指的空叶子节点,插入非空节点并不会对它们造成影响。
对于"特性(4)",是有可能违背的!
那接下来,想办法使之"满足特性(4)",就可以将树重新构造成红黑树了。
下面看看代码到底是怎样实现这三步的。
插入操作的伪代码如下(摘自《算法导论》):
RB-INSERT(T, z)
01 y = T.nil
02 x = T.root
03 while x ≠ T.nil
04 y = x
05 if z.key < x.key
06 x = x.left
07 else x = x.right
08 z.p = y
09 if y == T.nil
10 T.root = z
11 else if z.key < y.key
12 y.left = z
13 else y.right = z
14 z.left = T.nil
15 z.right = T.nil
16 z.color = RED
17 RB-INSERT-FIXUP(T, z)
最后,通过RB-INSERT-FIXUP对红黑树中的节点进行修正,保证树T仍然是一颗红黑树。
结合伪代码先理解RB-INSERT。理解RB-INSERT之后,我们接着对 RB-INSERT-FIXUP的伪代码进行说明, RB-INSERT-FIXUP主要对节点进行颜色修改以及旋转操作。
执行修正操作的伪代码如下(摘自《算法导论》):
RB-INSERT-FIXUP(T, z)
while z.p.color = RED
if z.p == z.p.p.left//前提条件
y = z.p.p.right
if y.color == RED
z.p.color = BLACK //Case1
y.color = BLACK //Case1
z.p.p.color = RED //Case1
z = z.p.p //Case1
else if z == z.p.right
z = z.p //Case2
LEFT-ROTATE(T, z)//Case2,左旋之后,进入Case3
z.p.color = BLACK //Case3
z.p.p.color = RED //Case3
RIGHT-ROTATE(T, z.p.p)//Case3
else(same as then clause
with "right" and "left" exchanged)
T.root.color = BLACK
根据被插入节点的父节点的情况,可以将"当节点z被着色为红色节点,并插入二叉树"划分为三种情况来处理。
① 情况说明:被插入的节点是根节点。
处理方法:直接把此节点涂为黑色。
② 情况说明:被插入的节点的父节点是黑色。
处理方法:什么也不需要做。节点被插入后,仍然是红黑树。
③ 情况说明:被插入的节点的父节点是红色。
处理方法:那么,该情况与红黑树的“特性(5)”相冲突。这种情况下,被插入节点是一定存在非空祖父节点的;进一步的讲,被插入节点也一定存在叔叔节点(即使叔叔节点为空,我们也视之为存在,空节点本身就是黑色节点)。理解这点之后,我们依据"叔叔节点的情况",将这种情况进一步划分为3种子情况(Case),如下所述。
三种子情况(Case)处理问题的核心思路都是:将红色的节点移到根节点;然后,将根节点设为黑色。下面就三种情形进行介绍。
1. Case 1:父节点是红色,叔叔是红色(红叔)
处理策略:
1. 将“父节点”设为黑色。
2. 将“叔叔节点”设为黑色。
3. 将“祖父节点”设为“红色”。
4. 将“祖父节点”设为“当前节点”(红色节点);之后继续对“当前节点”进行操作。
下面谈谈为什么要这样处理。 (建议理解的时候,通过下面的图进行对比)
“当前节点”和“父节点”都是红色,违背“特性4”。所以,将“父节点”设置“黑色”以解决这个问题。但是,将“父节点”由“红色”变成“黑色”之后,违背了“特性5”:因为,包含“父节点”的分支的黑色节点的总数增加了1。 解决这个问题的办法是:将“祖父节点”由“黑色”变成红色,同时,将“叔叔节点”由“红色”变成“黑色”。
为了便于说明,S代表新插入节点,P表示S的父节点,G表示S的祖父节点,U表示S的叔叔节点。
Case1 情况处理过程示意图如下:
上述Case1情况处理之后,发现它变成了Case2的情况,下面再接着就Case2的情形进行分析。
2. Case 2:父节点是红色,叔叔是黑色(黑叔),且当前节点是其父节点的右孩子
处理策略:
1. 将“父节点”作为“新的当前节点”。
2. 以“新的当前节点”为支点进行左旋。
下面谈谈为什么要这样处理。 (建议理解的时候,通过下面的图进行对比)
为了便于说明,设置“父节点”的代号为P,“当前节点”的代号为S。为什么要“以P为支点进行左旋”呢?上面我们说到,处理红黑树的核心思想是将红色的节点移到根节点;然后,将根节点设为黑色。既然是“将红色的节点移到根节点”,也就是说要不断的将破坏红黑树特性的红色节点上移(即向根方向移动)。 而S又是一个右孩子,因此,我们可以通过“左旋”来将S上移!
Case2 情况处理过程示意图如下:
上述Case2情况处理之后,发现它变成了Case3的情况,下面再接着就Case3的情形进行分析。
3. Case 3:父节点是红色,叔叔是黑色(黑叔),且当前节点是父节点的左孩子
处理策略:
1. 将“父节点”设为“黑色”。
2. 将“祖父节点”设为“红色”。
3. 以“祖父节点”为支点进行右旋。
下面谈谈为什么要这样处理。 (建议理解的时候,通过下面的图进行对比)
S和P都是红色,违背了红黑树的“特性4,我们可以将P由“红色”变为“黑色”,就解决了“违背特性4”的问题;但却引起了其它问题:违背特性5,因为将P由红色改为黑色之后,所有经过P的分支的黑色节点的个数增加了1。那我们如何解决“所有经过P的分支的黑色节点个数增加1”的问题呢? 我们可以通过“将G由黑色变成红色”,同时“以G为支点进行右旋”来解决。
Case3 情况处理过程示意图如下:
经上,发现已调整完毕,它又重新变回了一颗红黑树^_^!
调整函数RB-INSERT-FIXUP的JAVA实现代码如下所示。其中,用到了上文中提到的LEFT-ROTATE函数和RIGHT-ROTATE函数。通过代码我们能够看到,Case2其实是落在Case3内的。Case4~Case6跟前三种情况是对称的,因此图解中并没有画出后三种情况,读者可以参考代码自行理解。
//红黑树调整函数
private void fixAfterInsertion(Entry<K,V> x) {
x.color = RED;
while (x != null && x != root && x.parent.color == RED) {
if (parentOf(x) == leftOf(parentOf(parentOf(x)))) {
Entry<K,V> y = rightOf(parentOf(parentOf(x)));
if (colorOf(y) == RED) {//如果y为null,则视为BLACK
setColor(parentOf(x), BLACK); //Case1
setColor(y, BLACK); //Case1
setColor(parentOf(parentOf(x)), RED);//Case1
x = parentOf(parentOf(x)); //Case1
} else {
if (x == rightOf(parentOf(x))) {
x = parentOf(x); //Case2
rotateLeft(x); //Case2
}
setColor(parentOf(x), BLACK); //Case3
setColor(parentOf(parentOf(x)), RED); //Case3
rotateRight(parentOf(parentOf(x))); //Case3
}
} else {
Entry<K,V> y = leftOf(parentOf(parentOf(x)));
if (colorOf(y) == RED) {
setColor(parentOf(x), BLACK); //Case4
setColor(y, BLACK); //Case4
setColor(parentOf(parentOf(x)), RED); //Case4
x = parentOf(parentOf(x)); //Case4
} else {
if (x == leftOf(parentOf(x))) {
x = parentOf(x); //Case5
rotateRight(x);//Case5
}
setColor(parentOf(x), BLACK); //Case6
setColor(parentOf(parentOf(x)), RED); //Case6
rotateLeft(parentOf(parentOf(x))); //Case6
}
}
}
root.color = BLACK;
}
红黑树的基本操作(三) 删除
将红黑树内的某一个节点删除。需要执行的操作依次是:首先,将红黑树当作一颗二叉查找树,将该节点从二叉查找树中删除;然后,通过"旋转和重新着色"等一系列来修正该树,使之重新成为一棵红黑树。详细描述如下:
第一步:将红黑树当作一颗二叉查找树,将节点删除。
这和"删除常规二叉查找树中删除节点的方法是一样的"。分3种情况:
① 情况说明: 被删除节点没有儿子(即为叶节点)
处理方法:直接将该节点删除就OK了。
② 情况说明:被删除节点只有一个儿子
处理方法:直接删除该节点,并用该节点的唯一子节点顶替它的位置。
③ 情况说明:被删除节点有两个儿子
处理方法: 先找出它的后继节点;然后把“它的后继节点的内容”复制给“该节点的内容”;之后,删除“它的后继节点”。
在这里说明一下,后继节点相当于被删节点的替身,在将后继节点的内容复制给"被删除节点"之后,再将后继节点删除,这样做,就巧妙的将问题转换为"删除后继节点"的情况了,下面就考虑后继节点。 在"被删除节点"有两个非空子节点的情况下,它的后继节点不可能是双子非空。既然"它的后继节点"不可能双子都非空,就意味着"该节点的后继节点"要么没有儿子,要么只有一个儿子。若没有儿子,则按"情况① "进行处理;若只有一个儿子,则按"情况② "进行处理。
ps:对二叉树进行中序遍历,紧随某节点之后的一个节点称为它的后续节点,相应的,前一个节点则称为它的前续节点。
第二步:通过"旋转和重新着色"等一系列操作来修正该树,使之重新成为一棵红黑树。
因为"第一步"中删除节点之后,可能会违背红黑树的特性。所以需要通过"旋转和重新着色"来修正该树,使之重新成为一棵红黑树。
删除操作的伪代码如下(摘自《算法导论》)
RB-DELETE(T, z)
01 if z.left == T.nil or z.right == T.nil
02 y = z
03 else y = TREE-SUCCESSOR(z)
04 if y.left == T.nil
05 x = y.left
06 else x = y.right
07 x.p = y.p
08 if y.p = T.nil
09 T.root = x
10 else if y == y.p.left
11 y.p.left = x
12 else y.p.right = x
13 if y ≠ z
14 z.key = y.key
15 copy y's satellite data into z
16 if y.color = BLACK
17 then RB-DELETE-FIXUP(T, x)
18 return y
结合伪代码以及为代码上面的说明,先理解RB-DELETE。理解了RB-DELETE之后,接着对 RB-DELETE-FIXUP的伪代码进行说明
RB-DELETE-FIXUP(T, x)
01 while x ≠ T.root and x.color = BLACK
02 if x == x.p.left
03 w =x.p.right //“w”为“x的兄弟”
04 if w.color == RED // Case 1
05 w.color = BLACK//Case 1
06 x.p.color = RED //Case 1
07 LEFT-ROTATE(T, x.p) //Case 1
08 w = x.p.right // Case 1
09 if w.left.color == BLACK and w.right.color == BLACK // Case 2
10 w.color = RED // Case 2
11 x = x.p // Case 2
12 else if w.right.color == BLACK // Case 3
13 w.left.color = BLACK// Case 3
14 w.color = RED // Case 3
15 RIGHT-ROTATE(T, w)// Case 3
16 w = x.p.right // Case 3
17 w.color = x.p.color //Case 4
18 x.p.color = BLACK // Case 4
19 w.right.color = BLACK // Case 4
20 LEFT-ROTATE(T, p[x]) // Case 4
21 x = T.root // Case 4
22 else (same as then clause
with "right" and "left" exchanged)
23 x.color = BLACK
下面对删除函数进行分析。在分析之前,我们再次温习一下红黑树的几个特性:
每个节点或者是黑色,或者是红色。
根节点是黑色。
每个叶子节点是黑色。 [注意:这里叶子节点,是指为空的叶子节点!]
如果一个节点是红色的,则它的子节点必须是黑色的。
从一个节点到该节点的子孙节点的所有路径上包含相同数目的黑节点。
前面我们将"删除红黑树中的节点"大致分为两步,在第一步中"将红黑树当作一颗二叉查找树,将节点删除"后,可能违反"2、4、5"三个特性。第二步需要解决上面的三个问题,进而保持红黑树的全部特性。
为了便于分析,我们假设"x包含一个额外的黑色"(x原本的颜色还存在),这样就不会违反"特性5"。为什么呢?
通过RB-DELETE算法,我们知道:删除节点y之后,x占据了原来节点y的位置。 既然删除y(y是黑色),意味着减少一个黑色节点;那么,再在该位置上增加一个黑色即可。这样,当我们假设"x包含一个额外的黑色",就正好弥补了"删除y所丢失的黑色节点",也就不会违反"特性5"。 因此,假设"x包含一个额外的黑色"(x原本的颜色还存在),这样就不会违反"特性5"。
现在,x不仅包含它原本的颜色属性,x还包含一个额外的黑色。即x的颜色属性是"红+黑"或"黑+黑",它违反了"特性1"。
现在,我们面临的问题,由解决"违反2、4、5这三个特性"转换成了"解决违反1、2、4三个特性"。RB-DELETE-FIXUP需要做的就是通过算法恢复红黑树的特性。RB-DELETE-FIXUP的思想是:将x所包含的额外的黑色不断沿树上移(向根方向移动),直到出现下面的情况:
① 情况说明:x是“红+黑”节点。
处理方法:直接把x设为黑色,结束。此时红黑树性质全部恢复。
② 情况说明:x是“黑+黑”节点,且x是根。
处理方法:什么都不做,结束。此时红黑树性质全部恢复。
③ 情况说明:x是“黑+黑”节点,且x不是根。
处理方法:这种情况又可以划分为4种子情况,如下所述。
1. Case 1:x是"黑+黑"节点,x的兄弟节点是红色
处理策略:
1. 将x的兄弟节点设为“黑色”。
2. 将x的父节点设为“红色”。
3. 对x的父节点进行左旋。
4. 左旋后,重新设置x的兄弟节点。
下面谈谈为什么要这样处理。 (建议理解的时候,通过下面的图进行对比)这样做的目的是将“Case 1”转换为“Case 2”、“Case 3”或“Case 4”,从而进行进一步的处理。对x的父节点进行左旋;左旋后,为了保持红黑树特性,就需要在左旋前“将x的兄弟节点设为黑色”,同时“将x的父节点设为红色”;左旋后,由于x的兄弟节点发生了变化,需要更新x的兄弟节点,从而进行后续处理。
Case1 情况处理过程示意图如下:
还有一种情形如下:
2. Case 2:x是"黑+黑"节点,x的兄弟节点是黑色,x的兄弟节点的两个孩子都是黑色
处理策略:
1. 将x的兄弟节点设为“红色”。
2. 设置“x的父节点”为“新的x节点”。
下面谈谈为什么要这样处理。 (建议理解的时候,通过下面的图进行对比)这个情况的处理思想:是将“x中多余的一个黑色属性上移(往根方向移动)”。 x是“黑+黑”节点,我们将x由“黑+黑”节点 变成 “黑”节点,多余的一个“黑”属性移到x的父节点中,即x的父节点多出了一个黑属性(若x的父节点原先是“黑”,则此时变成了“黑+黑”;若x的父节点原先时“红”,则此时变成了“红+黑”)。 此时,需要注意的是:所有经过x的分支中黑节点个数没变化;但是,所有经过x的兄弟节点的分支中黑色节点的个数增加了1(因为x的父节点多了一个黑色属性)。为了解决这个问题,我们需要将“所有经过x的兄弟节点的分支中黑色节点的个数减1”即可,那么就可以通过“将x的兄弟节点由黑色变成红色”来实现。
经过上面的步骤(将x的兄弟节点设为红色),多余的一个颜色属性(黑色)已经跑到x的父节点中。我们需要将x的父节点设为“新的x节点”进行处理。若“新的x节点”是“黑+红”,直接将“新的x节点”设为黑色,即可完全解决该问题;若“新的x节点”是“黑+黑”,则需要对“新的x节点”进行进一步处理。
Case2 情况处理过程示意图如下:
3. Case 3:x是“黑+黑”节点,x的兄弟节点是黑色;x的兄弟节点的左孩子是红色,右孩子是黑色
处理策略:
1. 将x兄弟节点的左孩子设为“黑色”。
2. 将x兄弟节点设为“红色”。
3. 对x的兄弟节点进行右旋。
4. 右旋后,重新设置x的兄弟节点。
下面谈谈为什么要这样处理。 (建议理解的时候,通过下面的图进行对比)我们处理“Case 3”的目的是为了将“Case 3”进行转换成“Case 4”,从而进行进一步的处理。转换的方式是对x的兄弟节点进行右旋;为了保证右旋后,它仍然是红黑树,就需要在右旋前“将x的兄弟节点的左孩子设为黑色”,同时“将x的兄弟节点设为红色”;右旋后,由于x的兄弟节点发生了变化,需要更新x的兄弟节点,从而进行后续处理。
Case3 情况处理过程示意图如下:
4. Case 4:x是“黑+黑”节点,x的兄弟节点是黑色;x的兄弟节点的右孩子是红色的,x的兄弟节点的左孩子任意颜色
处理策略:
1. 将x父节点颜色 赋值给 x的兄弟节点。
2. 将x父节点设为“黑色”。
3. 将x兄弟节点的右子节设为“黑色”。
4. 对x的父节点进行左旋。
5. 设置“x”为“根节点”。
下面谈谈为什么要这样处理。 (建议理解的时候,通过下面的图进行对比)
为了便于说明,我们设置“当前节点”为S,“兄弟节点”为W,“兄弟节点的左孩子”为ch_l,“兄弟节点的右孩子”为ch_r,“父节点”为P。
我们要对P进行左旋。但在左旋前,我们需要调换P和W的颜色,并设置ch_r为黑色。为什么需要这里处理呢?因为左旋后,P和ch_l是父子关系,倘若ch_l是红色,P是红色,则违背了“特性4”;为了避免这一问题,我们将“P设置为黑色”。 但是,P设置为黑色之后,为了满足“特性5”,必须保证左旋之后:(1)经过根节点和S的分支的黑色节点个数不变;(2)经过根节点和ch_l的分支的黑色节点数不变;(3)经过根节点和ch_r的分支的黑色节点数不变。若要满足(1),只需S丢弃它多余的颜色即可。因为S的颜色是“黑+黑”,而左旋后“同时经过根节点和S的分支的黑色节点个数”增加了1;现在只需将S由“黑+黑”变成单独的“黑”节点即可。若要满足(2),只需将“P的原始颜色”与W的颜色调换一下即可;若要满足(2),只需要将ch_r设置为“黑色”即可。经过,上面的处理之后。红黑树的特性全部得到的满足!接着,我们将x设为根节点,就可以跳出while循环(可以参考伪代码)。
Case4 情况处理过程示意图如下:
删除后调整函数fixAfterDeletion()的JAVA实现代码如下,其中用到了上文中提到的rotateLeft()和rotateRight()函数。通过代码我们能够看到,Case3其实是落在Case4内的。Case5~Case8跟前四种情况是对称的,因此图解中并没有画出后四种情况,读者可以参考代码自行理解。
private void fixAfterDeletion(Entry<K,V> x) {
while (x != root && colorOf(x) == BLACK) {
if (x == leftOf(parentOf(x))) {
Entry<K,V> sib = rightOf(parentOf(x));
if (colorOf(sib) == RED) {
setColor(sib, BLACK); // Case1
setColor(parentOf(x), RED);// Case1
rotateLeft(parentOf(x)); // Case1
sib = rightOf(parentOf(x)); // Case1
}
if (colorOf(leftOf(sib)) == BLACK &&
colorOf(rightOf(sib)) == BLACK) {
setColor(sib, RED); // Case2
x = parentOf(x); // Case2
} else {
if (colorOf(rightOf(sib)) == BLACK) {
setColor(leftOf(sib), BLACK); // Case3
setColor(sib, RED); // Case3
rotateRight(sib); // Case3
sib = rightOf(parentOf(x)); // Case3
}
setColor(sib, colorOf(parentOf(x))); // Case4
setColor(parentOf(x), BLACK); // Case4
setColor(rightOf(sib), BLACK); // Case4
rotateLeft(parentOf(x)); // Case4
x = root; // Case4
}
} else { //与前四种情况对称
Entry<K,V> sib = leftOf(parentOf(x));
if (colorOf(sib) == RED) {
setColor(sib, BLACK); // Case5
setColor(parentOf(x), RED); // Case5
rotateRight(parentOf(x)); // Case5
sib = leftOf(parentOf(x)); // Case5
}
if (colorOf(rightOf(sib)) == BLACK &&
colorOf(leftOf(sib)) == BLACK) {
setColor(sib, RED); // Case6
x = parentOf(x); // Case6
} else {
if (colorOf(leftOf(sib)) == BLACK) {
setColor(rightOf(sib), BLACK); // Case7
setColor(sib, RED); // Case7
rotateLeft(sib); // Case7
sib = leftOf(parentOf(x)); // Case7
}
setColor(sib, colorOf(parentOf(x))); // Case8
setColor(parentOf(x), BLACK); // Case8
setColor(leftOf(sib), BLACK); // Case8
rotateRight(parentOf(x)); // Case8
x = root; // Case8
}
}
}
setColor(x, BLACK);
}
至此,红黑树的理论知识就讲完了,最后祝大家周末学习愉快O(∩_∩)O~Happy Weekend !
以上是关于一篇文章教你透彻了解红黑树的主要内容,如果未能解决你的问题,请参考以下文章