Red-Black Tree,又称为“红黑树”。
红黑树是一种特殊的二叉查找树。红黑树的每个节点上都有存储位表示节点的颜色,可以是红(Red)或黑(Black)。
红黑树特性:
1、每个节点上都有颜色,红色或者黑色
2、根节点是黑色
3、每个叶子节点(NULL或NIL的叶子结点)是黑色
4、如果一个节点是红色,那么它的子节点必须是黑色
5、从一个节点到叶子结点的所有路径上,黑色结点数相同
由于特性5,确保没有一条路径会比其他路径长两倍,因而,红黑树是相对平衡的二叉树。
红黑树应用:
红黑树的应用比较广泛,主要用来存储有序的数据。它是时间复杂度为O(lgn),效率非常之高。
定理:一棵含有n个节点的红黑树的高度至多为2log(n+1)。
红黑树基本操作:左旋和右旋
1、左旋
伪代码:
LEFT-ROTATE(T, x)
01 y ← right[x] // 前提:这里假设x的右孩子为y。下面开始正式操作
02 right[x] ← left[y] // 将 “y的左孩子” 设为 “x的右孩子”,即 将β设为x的右孩子
03 p[left[y]] ← x // 将 “x” 设为 “y的左孩子的父亲”,即 将β的父亲设为x
04 p[y] ← p[x] // 将 “x的父亲” 设为 “y的父亲”
05 if p[x] = nil[T]
06 then root[T] ← y // 情况1:如果 “x的父亲” 是空节点,则将y设为根节点
07 else if x = left[p[x]]
08 then left[p[x]] ← y // 情况2:如果 x是它父节点的左孩子,则将y设为“x的父节点的左孩子”
09 else right[p[x]] ← y // 情况3:(x是它父节点的右孩子) 将y设为“x的父节点的右孩子”
10 left[y] ← x // 将 “x” 设为 “y的左孩子”
11 p[x] ← y // 将 “x的父节点” 设为 “y”
简单实例:
2. 右旋
伪代码:
RIGHT-ROTATE(T, y)
01 x ← left[y] // 前提:这里假设y的左孩子为x。下面开始正式操作
02 left[y] ← right[x] // 将 “x的右孩子” 设为 “y的左孩子”,即 将β设为y的左孩子
03 p[right[x]] ← y // 将 “y” 设为 “x的右孩子的父亲”,即 将β的父亲设为y
04 p[x] ← p[y] // 将 “y的父亲” 设为 “x的父亲”
05 if p[y] = nil[T]
06 then root[T] ← x // 情况1:如果 “y的父亲” 是空节点,则将x设为根节点
07 else if y = right[p[y]]
08 then right[p[y]] ← x // 情况2:如果 y是它父节点的右孩子,则将x设为“y的父节点的左孩子”
09 else left[p[y]] ← x // 情况3:(y是它父节点的左孩子) 将x设为“y的父节点的左孩子”
10 right[x] ← y // 将 “y” 设为 “x的右孩子”
11 p[y] ← x // 将 “y的父节点” 设为 “x”
简单实例
红黑树的操作:插入
首先,将红黑树当作一颗二叉查找树,将节点插入;然后,将节点着色为红色;最后,通过旋转和重新着色等方法来修正该树,使之重新成为一颗红黑树。详细步骤如下:
第一步:将红黑树视为普通的二叉查找树,将节点插入二叉查找树
红黑树本质就是一颗二叉查找树,节点插入后,树的键值仍然要保持有序性。无论如何操作,它仍是一颗二叉查找树。
第二步:插入的节点标记为红色
因为特性5的原因,插入的切点要标记为红色,接下来只要再努力满足其它特性即可。
第三步:通过一系列的旋转和着色,使之重新满足红黑树的特性
对于第二步操作完成以后,前三个特性基本都是满足的,只有第四个特性可能会不满足(插入红色节点,父节点也是红色的情况),那么只要满足第四个特性即可。
伪代码如下:
1)向红黑树插入节点z并标记为红色:
RB-INSERT(T, z)
01 y ← nil[T] // 新建节点“y”,将y设为空节点。用来保存z的父节点
02 x ← root[T] //因为节点的插入要从根节点开始比较,所以将根节点赋值x,比较使用
03 while x ≠ nil[T] // 找出要插入的节点“z”在二叉树T中的位置x, 当x为nil时,则x所在的节点位置就是节点z要插入的位置
04 do y ← x
05 if key[z] < key[x]
06 then x ← left[x]
07 else x ← right[x]
//循环执行完成之后,x == left[y] 或者 x == right[y],即x应该是y的左孩子节点或者右孩子节点,或者说z结点是y的左孩子节点或者右孩子节点
08 p[z] ← y // 设置 “z的父亲” 为 “y”
09 if y == nil[T]
10 then root[T] ← z // 情况1:若y是空节点,那么原始二叉树为一颗空二叉树,将z设为根
11 else if key[z] < key[y]
12 then left[y] ← z // 情况2:若“z所包含的值” < “y所包含的值”,则将z设为“y的左孩子”
13 else right[y] ← z // 情况3:(“z所包含的值” >= “y所包含的值”)将z设为“y的右孩子”
14 left[z] ← nil[T] // z的左孩子设为空
15 right[z] ← nil[T] // z的右孩子设为空。至此,已经完成将“节点z插入到二叉树”中了。
16 color[z] ← RED // 将z着色为“红色”
17 RB-INSERT-FIXUP(T, z) // 通过RB-INSERT-FIXUP对红黑树的节点进行颜色修改以及旋转,让树T仍然是一颗红黑树
最后一行的RB-INSERT-FIXUP(T, z) 即是修正红黑树的方法,伪代码如下:
RB-INSERT-FIXUP(T, z)
while (color[p[z]] == RED){ // 若“当前节点(z)的父节点是红色”,则进行以下处理
if (p[z] == left[p[p[z]]]){ // 若“z的父节点”是“z的祖父节点的左孩子”,则进行以下处理:
y ← right[p[p[z]]] // 将y设置为 “z的叔叔节点(z的祖父节点的右孩子)”
if (color[y] == RED){ // 如果:叔叔是红色
color[p[z]] ← BLACK // (01) 将“父节点”设为黑色。
color[y] ← BLACK // (02) 将“叔叔节点”设为黑色。
color[p[p[z]]] ← RED // (03) 将“祖父节点”设为“红色”。
z ← p[p[z]] // (04) 将“祖父节点”设为“当前节点”(红色节点)
continue // (05) 继续下一次while循环
}
else{ // 如果:叔叔是黑色
if (z == right[p[z]]){ // 如果当前节点是右孩子
z ← p[z] // (01) 将“父节点”作为“新的当前节点”。
LEFT-ROTATE(T, z) // (02) 以“新的当前节点”为支点进行左旋
}
color[p[z]] ← BLACK // (01) 将“父节点”设为“黑色”。
color[p[p[z]]] ← RED // (02) 将“祖父节点”设为“红色”。
RIGHT-ROTATE(T, p[p[z]]) // (03) 以“祖父节点”为支点进行右旋。
}
}
else{ // 若“z的父节点”是“z的祖父节点的右孩子”,则进行以下处理:
y ← left[p[p[z]]] // 将y设置为“z的叔叔节点(z的祖父节点的左孩子)”
if (color[y] = RED){ // 如果:叔叔是红色
color[p[z]] ← BLACK // (01) 将“父节点”设为黑色
color[y] ← BLACK // (02) 将“叔叔节点”设为黑色
color[p[p[z]]] ← RED // (03) 将“祖父节点”设为“红色”
z ← p[p[z]] // (04) 将“祖父节点”设为“当前节点”(红色节点)
continue // (05) 继续下一次while循环
}
else{ // 如果:叔叔是黑色
if (z = left[p[z]]){ // 如果当前节点是左孩子
z ← p[z] // (01) 将“父节点”作为“新的当前节点”。
RIGHT-ROTATE(T, p[p[z]]) // (02) 以“祖父节点”为支点进行右旋。
}
color[p[z]] ← BLACK // (01) 将“父节点”设为“黑色”
color[p[p[z]]] ← RED // (02) 将“祖父节点”设为“红色”
LEFT-ROTATE(T, z) // (03) 以“新的当前节点”为支点进行左旋
}
}
}
color[root[T]] ← BLACK //设置根节点为黑色