本篇是我学习红黑树的笔记，方便以后遗忘时回忆。
红黑树可以看作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);
}
699
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
