本篇是我学习红黑树的笔记,方便以后遗忘时回忆。
红黑树可以看作4阶B树的具体实现。4阶B树逻辑上比较简单,在纸上画出来很简单,但是写到程序中会非常麻烦。红黑树作为4阶B树的具体实现,可以比较简单的写到程序中。jdk1.8中hashmap中就用到了红黑树。
将红黑树中红色节点上移到父节点同高处的位置,就可以看成一个4阶B树。如下图
1图是一个正常的红黑树,2图是将红节点上移之后的,3图是对应的4阶B树
综上,在学习红黑树时可以辅以4阶B树加深记忆。红黑树有5个特点,从其他人的博客中可以找到,我不多说什么。
在学习红黑树之前一定要熟悉4阶B树和平衡二叉树。平衡二叉树中的旋转在红黑树中也需要用到。
红黑树的插入:
在插入前需要寻找插入位置。寻找方法与平衡二叉树寻找插入节点方法一样,这里不多说。
所有插入的节点都是红节点。以下列举出插入时会遇到的情况。
1.插入的是一颗空树,直接将插入节点作为根节点,并将颜色变为黑色。
2.插入节点的父节点是黑色,没有改变红黑树的性质,什么都不用做。
3.插入节点的父节点是红色且父节点为祖先的左孩子
3.1插入节点的叔父节点为红节点
插入节点的叔父节点是红节点。此时父节点与叔父节点为红,祖先节点为黑。将祖先节点变红,将父节点与叔父节点变黑,继续递归检查祖先节点。
3.2插入节点的叔父节点为空或黑节点。(我在做题时没有遇见过叔父为黑的情况,但JDK1.8的源码中将这种情况包含在内了,我觉得如果叔父为黑不符合红黑树定义)
3.2.1插入节点为父节点的右孩子,以父节点左旋,变为3.2.2的情况
3.2.2插入节点为父节点左孩子,将父节点与祖父节点交换颜色并以祖父节点右旋。
4.插入节点的父节点是红色且父节点为祖先的右孩子(与情况3类似)
4.1插入节点的叔父节点为红节点
插入节点的叔父节点是红节点。此时父节点与叔父节点为红,祖先节点为黑。将祖先节点变红,将父节点与叔父节点变黑,继续递归检查祖先节点。
4.2插入节点的叔父节点为空或黑节点。
4.2.1插入节点为父节点的左孩子,以父节点右旋,变为4.2.2的情况
4.2.2插入节点为父节点左孩子,将父节点与祖父节点交换颜色并以祖父节点左旋。
最后根节点变黑。步骤3.1和4.1可能会将根节点变红。
以下为插入样例,我尽量覆盖全部情况。
一次插入70,40,20(40节点为红色)。此时插入20节点的叔父节点为空,且父节点为红,情况3.2.2,调整如下。
插入10之后父节点与叔父节点为红,且10节点是左孩子,情况3.1。将20,70变黑,40变红。最后再将根节点变黑。由于中间情况简单,所以我省略了步骤。
插入15,父节点为左孩子,叔父节点为空,插入节点为右孩子,情况3.2.1。先以10节点左旋,变为3.2.2,再将15与20交换颜色并右旋,调整为右图。
继续插入5节点,情况3.1.将15变红之后需要继续向上判断,此时情况符合红黑树定义,无需继续调整。
继续插入60,80。情况简单不多赘述。
(第一幅图中5为红节点,笔误)插入72之后符合情况4.1.将70变红之后需要继续向上判断。此时情况符合红黑树定义,无需继续调整。
插入75,情况3.2.1。以72左旋调整为3.2.2。调整为右图。
插入71,情况3.1。将75变红之后继续向上判断,出现两个连续的红节点,此时将75当作插入节点继续判断属于情况4.1。将15,70变黑,40变红。最后再将根节点变黑。能力有限未能覆盖全部情况,读者可以自行尝试。
下面再附上与之对应的4阶B树的插入过程。
在上图最后一步分裂时将10或15分裂到父节点都可以,此处是为了与上面的红黑树做对应所以选择了15。
在上图第一步分裂时选择70或72向上都可以,此处是为了与上面的红黑树做对应所以选择了70。之后的分裂情况也都一样。
左边是最后的4阶B树,右侧是上面的红黑书。
红黑树的删除:
首先寻找删除节点,删除节点有3种情况,1.叶子节点,2.有一个子树,3.有两个子树。具体处理办法与平衡二叉树的删除相似,这里不多做赘述。
删除相比插入麻烦一些,删除之后的调整。下面写删除时遇到的情况。
只有删除黑色节点需要调整,删除红色节点直接删除,不用调整。
1.删除节点是左孩子
1.1删除节点的兄弟节点为红:
兄弟节点为红,被删除节点为黑,所以兄弟节点必有2个孩子节点且是黑节点,否则不符合红黑树定义。此时将父节点与兄弟节点交换颜色,并以父节点左旋,可以将删除节点的兄弟节点换成黑节点。转为以下情况。
1.2删除节点的兄弟节点为黑色:
1.2.1删除节点的兄弟节点孩子为2个黑节点或2个孩子为空。(为空也可以看作2个黑节点)
直接删除该节点,并将兄弟节点变红,父节点变黑。若父节点原本就是黑色,则必须向上继续判断;如果父节点原本为红色,变为黑色之后红黑树性质未改变。
1.2.2删除节点的兄弟节点左孩子为红,右孩子为黑
将兄弟节点与兄弟节点左孩子交换颜色,并以兄弟节点右旋,此时变为以下情况
1.2.3删除节点节点的兄弟节点的左孩子为红或黑或空,右孩子为红
将兄弟节点变为父节点的颜色,父节点与兄弟节点的右孩子变黑,以父节点左旋。
2.删除节点是右孩子(与左孩子类似)
2.1删除节点的兄弟节点为红
兄弟节点为红,被删除节点为黑,所以兄弟节点必有2个孩子节点且是黑节点,否则不符合红黑树定义。此时将父节点与兄弟节点交换颜色,并以父节点右旋,可以将删除节点的兄弟节点换成黑节点。转为以下情况。
2.2删除节点的兄弟节点为黑色
2.2.1删除节点的兄弟节点孩子为2个黑色节点或2个孩子为空。
待删除节点直接删除,兄弟节点变红,父节点变黑。若父节点原本就是黑色,则必须向上继续判断;如果父节点原本为红色,变为黑色之后红黑树性质未改变。
2.2.2删除节点的兄弟节点左孩子为黑,右孩子为红
将兄弟节点与兄弟节点的右孩子交换颜色,并以兄弟节点左旋,变为以下情况
2.2.3删除节点的兄弟节点的左孩子为红,右孩子为黑或红或空
将兄弟节点变为父节点的颜色,父节点与兄弟节点的左孩子变黑,以父节点右旋。
以下是示例,我尽量覆盖全部情况。
原图还是上面插入的图。删除20节点,20节点的兄弟节点是黑节点,兄弟节点的左孩子为红,右孩子为空(也可以认为是黑),情况2.2.3,将10变为15的颜色,5和15变黑,以15右旋。
删除60节点,60节点的兄弟为红色,情况1.1,兄弟节点与父节点交换颜色,以父节点左旋。变为情况1.2.2,71与72交换颜色并以72右旋,变为情况1.2.3,71变为70的颜色,70,72变黑并以70左旋。
删除80,兄弟节点为黑,且孩子为空(或者可以看作双黑),情况2.2.1,72变红,75变黑。由于75原本是红色,改变后符合红黑树性质,不用继续向上判断。
删除5节点,情况1.2.1,15变红,10变黑。由于10原本就是黑色,需要继续向上判断。10节点的兄弟节点为黑,且孩子为双黑,情况1.2.1,71变红,40变黑。由于40原本就是黑色,需要继续向上判断,但40已经到根节点,所以停止。
下面给出对应4阶B树的删除
4阶B书删除比较简单,不多赘述。最后的4阶B树与前边删除部分的最后一个红黑树等价,读者可以自行调整查看。
下面给出gdk1.8中的红黑树代码,代码的判断基本与我写的情况一致。
插入后条调整
private void fixAfterInsertion(Entry<K,V> x) {
//新节点都为红色
x.color = RED;
//x存在且c不是根节点且x的父节点为红色
while (x != null && x != root && x.parent.color == RED) {
//如果x的父节点是祖父节点的左子树的话
if (parentOf(x) == leftOf(parentOf(parentOf(x)))) {
//取出祖父节点的右子树
Entry<K,V> y = rightOf(parentOf(parentOf(x)));
//判断祖父节点右子树是否为红色
if (colorOf(y) == RED) {
//红色
//将父节点变成黑色
setColor(parentOf(x), BLACK);
//祖父节点的右子树变成黑色
setColor(y, BLACK);
//祖父节点变成红色
setColor(parentOf(parentOf(x)), RED);
//将x的引用指向祖父节点
x = parentOf(parentOf(x));
} else {
//祖父节点右子树为黑色
//x节点是父节点的右子树
if (x == rightOf(parentOf(x))) {
//x引用指向父节点
x = parentOf(x);
//左旋
rotateLeft(x);
}
//将x的父节点变成黑色
setColor(parentOf(x), BLACK);
//x的祖父节点变成红色
setColor(parentOf(parentOf(x)), RED);
//右旋
rotateRight(parentOf(parentOf(x)));
}
} else {
//如果x的父节点是祖父节点的右子树的话
//取出祖父节点的左子树
Entry<K,V> y = leftOf(parentOf(parentOf(x)));
//祖父节点左子树为红色
if (colorOf(y) == RED) {
//相关变色操作
setColor(parentOf(x), BLACK);
setColor(y, BLACK);
setColor(parentOf(parentOf(x)), RED);
x = parentOf(parentOf(x));
} else {
//祖父节点左子树为红色
//入股x是父节点的左子树
if (x == leftOf(parentOf(x))) {
x = parentOf(x);
//右旋
rotateRight(x);
}
//相关变色操作
setColor(parentOf(x), BLACK);
setColor(parentOf(parentOf(x)), RED);
//左旋
rotateLeft(parentOf(parentOf(x)));
}
}
}
root.color = BLACK;
}
删除后调整
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);
setColor(parentOf(x), RED);
rotateLeft(parentOf(x));
sib = rightOf(parentOf(x));
}
if (colorOf(leftOf(sib)) == BLACK &&
colorOf(rightOf(sib)) == BLACK) {
setColor(sib, RED);
x = parentOf(x);
} else {
if (colorOf(rightOf(sib)) == BLACK) {
setColor(leftOf(sib), BLACK);
setColor(sib, RED);
rotateRight(sib);
sib = rightOf(parentOf(x));
}
setColor(sib, colorOf(parentOf(x)));
setColor(parentOf(x), BLACK);
setColor(rightOf(sib), BLACK);
rotateLeft(parentOf(x));
x = root;
}
} else { // symmetric
Entry<K,V> sib = leftOf(parentOf(x));
if (colorOf(sib) == RED) {
setColor(sib, BLACK);
setColor(parentOf(x), RED);
rotateRight(parentOf(x));
sib = leftOf(parentOf(x));
}
if (colorOf(rightOf(sib)) == BLACK &&
colorOf(leftOf(sib)) == BLACK) {
setColor(sib, RED);
x = parentOf(x);
} else {
if (colorOf(leftOf(sib)) == BLACK) {
setColor(rightOf(sib), BLACK);
setColor(sib, RED);
rotateLeft(sib);
sib = leftOf(parentOf(x));
}
setColor(sib, colorOf(parentOf(x)));
setColor(parentOf(x), BLACK);
setColor(leftOf(sib), BLACK);
rotateRight(parentOf(x));
x = root;
}
}
}
setColor(x, BLACK);
}