转载:http://blog.csdn.net/eson_15/article/details/51144079
(友情提示,红-黑树是基于二叉搜索树的,如果对二叉搜索树不了解,可以先看看:二叉搜索树 )
从第4节的分析中可以看出,二叉搜索树是个很好的数据结构,可以快速地找到一个给定关键字的数据项,并且可以快速地插入和删除数据项。但是二叉搜索树有个很麻烦的问题,如果树中插入的是随机数据,则执行效果很好,但如果插入的是有序或者逆序的数据,那么二叉搜索树的执行速度就变得很慢。因为当插入数值有序时,二叉树就是非平衡的了,排在一条线上,其实就变成了一个链表……它的快速查找、插入和删除指定数据项的能力就丧失了。
为了能以较快的时间O(logN)来搜索一棵树,需要保证树总是平衡的(或者至少大部分是平衡的),这就是说对树中的每个节点在它左边的后代数目和在它右边的后代数目应该大致相等。红-黑树的就是这样的一棵平衡树,对一个要插入的数据项,插入例程要检查会不会破坏树的特征,如果破坏了,程序就会进行纠正,根据需要改变树的结构,从而保持树的平衡。那么红-黑树都有哪些特征呢?
1.红-黑树的特征
它主要有两个特征:1.节点都有颜色;2.在插入和删除的过程中,要遵循保持这些颜色的不同排列的规则。首先第一个特征很好解决,在节点类中店家一个数据字段,例如boolean型变量,以此来表示节点的颜色信息。第二个特征比较复杂,红-黑树有它的几个规则,如果遵循这些规则,那么树就是平衡的。红-黑树的主要规则如下:
1.每个节点不是红色就是黑色的;
2.根节点总是黑色的;
3.如果节点是红色的,则它的子节点必须是黑色的(反之不一定);
4.从根节点到叶节点或空子节点的每条路径,必须包含相同数目的黑色节点(即相同的黑色高度)。
在红-黑树中插入的节点都是红色的,这不是偶然的,因为插入一个红色节点比插入一个黑色节点违背红-黑规则的可能性更小。原因是:插入黑色节点总会改变黑色高度(违背规则4),但是插入红色节点只有一半的机会会违背规则3。另外违背规则3比违背规则4要更容易修正。当插入一个新的节点时,可能会破坏这种平衡性,那么红-黑树是如何修正的呢?
2.平衡性的修正
红-黑树主要通过三种方式对平衡进行修正,改变节点颜色、左旋和右旋。这看起来有点抽象,我们分别来介绍它们。
1.变色
改变节点颜色比较容易理解,因为它违背了规则3。假设现在有个节点E,然后插入节点A和节点S,节点A在左子节点,S在右子节点,目前是平衡的。如果此时再插一个节点,那么就出现了不平衡了,因为红色节点的子节点必须为黑色,但是新插的节点是红色的。所以这时候就必须改变节点颜色了。所以我们将根的两个子节点从红色变为黑色(至于为什么都要变,下面插入的时候会详细介绍),将父节点会从黑色变成红色。可以用如下示意图表示一下:
2.左旋
通常左旋操作用于将一个向右倾斜的红色链接旋转为向左链接。示意图如下:
左旋有个很萌萌哒的动态示意图,可以方便理解:
3.右旋
右旋可左旋刚好相反,这里不再赘述,直接看示意图:
当然咯,右旋也有个萌萌的动态图:
这里主要介绍了红-黑树对平衡的三种修正方式,大家有个感性的认识,那么什么时候该修正呢?什么时候该用哪种修正呢?这将是接下来我们要探讨的问题。
3.红-黑树的操作
红-黑树的基本操作是添加、删除和旋转。对红-黑树进行添加或删除后,可能会破坏其平衡性,会用到哪种旋转方式去修正呢?我们首先对红-黑树的节点做一介绍,然后分别对左旋和右旋的具体实现做一分析,最后我们探讨下红-黑树的具体操作。
1.红-黑树的节点
红-黑树是对二叉搜索树的改进,所以其节点与二叉搜索树是差不多的,只不过在它基础上增加了一个boolean型变量来表示节点的颜色,具体看RBNode类:
- public class RBNodeextends Comparable>{
- boolean color;
- T key;
- RBNode left;
- RBNode right;
- RBNode parent;
-
- public RBNode(T key, boolean color, RBNode parent, RBNode left, RBNode right) {
- this.key = key;
- this.color = color;
- this.parent = parent;
- this.left = left;
- this.right = right;
- }
-
- public T getKey() {
- return key;
- }
-
- public String toString() {
- return "" + key + (this.color == RED? "R" : "B");
- }
- }
2.左旋具体实现
上面对左旋的概念已经有了感性的认识了,这里就不再赘述了,我们从下面的代码中结合上面的示意图,探讨一下左旋的具体实现:
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
- private void leftRotate(RBNode x) {
-
- RBNode y = x.right;
- x.right = y.left;
-
- if(y.left != null)
- y.left.parent = x;
-
-
- y.parent = x.parent;
-
- if(x.parent == null) {
- this.root = y;
- } else {
- if(x == x.parent.left)
- x.parent.left = y;
- else
- x.parent.right = y;
- }
-
-
- y.left = x;
- x.parent = y;
- }
3.右旋具体实现
上面对右旋的概念已经有了感性的认识了,这里也不再赘述了,我们从下面的代码中结合上面的示意图,探讨一下右旋的具体实现:
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
- private void rightRotate(RBNode y) {
-
- RBNode x = y.left;
- y.left = x.right;
-
- if(x.right != null)
- x.right.parent = y;
-
-
- x.parent = y.parent;
-
- if(y.parent == null) {
- this.root = x;
- } else {
- if(y == y.parent.right)
- y.parent.right = x;
- else
- y.parent.left = x;
- }
-
-
- x.right = y;
- y.parent = x;
- }
4.插入操作
分析完了红-黑树中主要的旋转操作,接下来我们开始分析常见的插入、删除等操作了。这里先分析插入操作。 由于红-黑树是二叉搜索树的改进,所以插入操作的前半工作时相同的,即先找到待插入的位置,再将节点插入,先来看看插入的前半段代码:
-
- public void insert(T key) {
- RBNode node = new RBNode(key, RED, null, null, null);
- if(node != null)
- insert(node);
- }
-
-
- private void insert(RBNode node) {
- RBNode current = null;
- RBNode x = this.root;
-
-
- while(x != null) {
- current = x;
- int cmp = node.key.compareTo(x.key);
- if(cmp < 0)
- x = x.left;
- else
- x = x.right;
- }
- node.parent = current;
-
-
- if(current != null) {
- int cmp = node.key.compareTo(current.key);
- if(cmp < 0)
- current.left = node;
- else
- current.right = node;
- } else {
- this.root = node;
- }
-
-
- insertFixUp(node);
- }
这与二叉搜索树中实现的思路一模一样,这里不再赘述,主要看看方法里面最后一步insertFixUp操作。因为插入后可能会导致树的不平衡,insertFixUp方法里主要是分情况讨论,分析何时变色,何时左旋,何时右旋。我们先从理论上分析具体的情况,然后再看insertFixUp方法的具体实现。
如果是第一次插入,由于原树为空,所以只会违反红-黑树的规则2,所以只要把根节点涂黑即可;如果插入节点的父节点是黑色的,那不会违背红-黑树的规则,什么也不需要做;但是遇到如下三种情况时,我们就要开始变色和旋转了:
1. 插入节点的父节点和其叔叔节点(祖父节点的另一个子节点)均为红色的;
2. 插入节点的父节点是红色,叔叔节点是黑色,且插入节点是其父节点的右子节点;
3. 插入节点的父节点是红色,叔叔节点是黑色,且插入节点是其父节点的左子节点。
下面我们先挨个分析这三种情况都需要如何操作,然后再给出实现代码。
对于情况1:插入节点的父节点和其叔叔节点(祖父节点的另一个子节点)均为红色的。此时,肯定存在祖父节点,但是不知道父节点是其左子节点还是右子节点,但是由于对称性,我们只要讨论出一边的情况,另一种情况自然也与之对应。这里考虑父节点是祖父节点的左子节点的情况,如下左图所示:
对于这种情况,我们要做的操作有:将当前节点(4)的父节点(5)和叔叔节点(8)涂黑,将祖父节点(7)涂红,变成上右图所示的情况。再将当前节点指向其祖父节点,再次从新的当前节点开始算法(具体等下看下面的程序)。这样上右图就变成了情况2了。
对于情况2:插入节点的父节点是红色,叔叔节点是黑色,且插入节点是其父节点的右子节点。我们要做的操作有:将当前节点(7)的父节点(2)作为新的节点,以新的当前节点为支点做左旋操作。完成后如左下图所示,这样左下图就变成情况3了。
对于情况3:插入节点的父节点是红色,叔叔节点是黑色,且插入节点是其父节点的左子节点。我们要做的操作有:将当前节点的父节点(7)涂黑,将祖父节点(11)涂红,在祖父节点为支点做右旋操作。最后把根节点涂黑,整个红-黑树重新恢复了平衡,如右上图所示。至此,插入操作完成!
我们可以看出,如果是从情况1开始发生的,必然会走完情况2和3,也就是说这是一整个流程,当然咯,实际中可能不一定会从情况1发生,如果从情况2开始发生,那再走个情况3即可完成调整,如果直接只要调整情况3,那么前两种情况均不需要调整了。故变色和旋转之间的先后关系可以表示为:变色->左旋->右旋。
至此,我们完成了全部的插入操作。下面我们看看insertFixUp方法中的具体实现(可以结合上面的分析图,更加利与理解):
- private void insertFixUp(RBNode node) {
- RBNode parent, gparent;
-
-
- while(((parent = parentOf(node)) != null) && isRed(parent)) {
- gparent = parentOf(parent);
-
-
- if(parent == gparent.left) {
- RBNode uncle = gparent.right;
-
-
- if(uncle != null && isRed(uncle)) {
- setBlack(parent);
- setBlack(uncle);
- setRed(gparent);
- node = gparent;
- continue;
- }
-
-
- if(node == parent.right) {
- leftRotate(parent);
- RBNode tmp = parent;
- parent = node;
- node = tmp;
- }
-
-
- setBlack(parent);
- setRed(gparent);
- rightRotate(gparent);
- } else {
- RBNode uncle = gparent.left;
-
-
- if(uncle != null & isRed(uncle)) {
- setBlack(parent);
- setBlack(uncle);
- setRed(gparent);
- node = gparent;
- continue;
- }
-
-
- if(node == parent.left) {
- rightRotate(parent);
- RBNode tmp = parent;
- parent = node;
- node = tmp;
- }
-
-
- setBlack(parent);
- setRed(gparent);
- leftRotate(gparent);
- }
- }
-
-
- setBlack(this.root);
- }
5.删除操作
上面探讨完了红-黑树的插入操作,接下来讨论删除,红-黑树的删除和二叉查找树的删除是一样的,只不过删除后多了个平衡的修复而已。我们先来回忆一下二叉搜索树的删除(也可以直接阅读这篇博客:二叉搜索树):
1. 如果待删除节点没有子节点,那么直接删掉即可;
2. 如果待删除节点只有一个子节点,那么直接删掉,并用其子节点去顶替它;
3. 如果待删除节点有两个子节点,这种情况比较复杂:首选找出它的后继节点,然后处理“后继节点”和“被删除节点的父节点”之间的关系,最后处理“后继节点的子节点”和“被删除节点的子节点”之间的关系。每一步中也会有不同的情况,我们结合下面代码的分析就能弄清楚,当然了,如果已经弄懂了二叉搜索树,那自然自然都能明白,这里就不赘述了。
我们来看一下删除操作的代码及注释:
-
- public void remove(T key) {
- RBNode node;
- if((node = search(root, key)) != null)
- remove(node);
- }
-
- private void remove(RBNode node) {
- RBNode child, parent;
- boolean color;
-
-
- if((node.left != null) && (node.right != null)) {
-
- RBNode replace = node;
-
- replace = replace.right;
- while(replace.left != null)
- replace = replace.left;
-
-
- if(parentOf(node) != null) {
- if(node == parentOf(node).left)
- parentOf(node).left = replace;
- else
- parentOf(node).right = replace;
- } else {
- this.root = replace;
- }
-
-
- child = replace.right;
- parent = parentOf(replace);
- color = colorOf(replace);
- if(parent == node) {
- parent = replace;
- } else {
- if(child != null)
- setParent(child, parent);
- parent.left = child;
- replace.right = node.right;
- setParent(node.right, replace);
- }
- replace.parent = node.parent;
- replace.color = node.color;
- replace.left = node.left;
- node.left.parent = replace;
-
- if(color == BLACK) {
- removeFixUp(child, parent);
- }
- node = null;
- return;
- }
- }
下面我们主要看看方法里面最后的removeFixUp操作。因为remove后可能会导致树的不平衡,removeFixUp方法里主要是分情况讨论,分析何时变色,何时左旋,何时右旋。我们同样先从理论上分析具体的情况,然后再看removeFixUp方法的具体实现。
从上面的代码中可以看出,删除某个节点后,会用它的后继节点来填上,并且后继节点会设置为和删除节点同样的颜色,所以删除节点的那个位置是不会破坏平衡的。可能破坏平衡的是后继节点原来的位置,因为后继节点拿走了,原来的位置结构改变了,这就会导致不平衡的出现。所以removeFixUp方法中传入的参数也是后继节点的子节点和父节点。
为了方便下文的叙述,我们现在约定:后继节点的子节点称为“当前节点”。
删除操作后,如果当前节点是黑色的根节点,那么不用任何操作,因为并没有破坏树的平衡性,即没有违背红-黑树的规则,这很好理解。如果当前节点是红色的,说明刚刚移走的后继节点是黑色的,那么不管后继节点的父节点是啥颜色,我们只要将当前节点涂黑就可以了,红-黑树的平衡性就可以恢复。但是如果遇到以下四种情况,我们就需要通过变色或旋转来恢复红-黑树的平衡了。
1. 当前节点是黑色的,且兄弟节点是红色的(那么父节点和兄弟节点的子节点肯定是黑色的);
2. 当前节点是黑色的,且兄弟节点是黑色的,且兄弟节点的两个子节点均为黑色的;
3. 当前节点是黑色的,且兄弟节点是黑色的,且兄弟节点的左子节点是红色,右子节点时黑色的;
4. 当前节点是黑色的,且兄弟节点是黑色的,且兄弟节点的右子节点是红色,左子节点任意颜色。
以上四种情况中,我们可以看出2,3,4其实是“当前节点是黑色的,且兄弟节点是黑色的”的三种子集,等会在程序中可以体现出来。现在我们假设当前节点是左子节点(当然也可能是右子节点,跟左子节点相反即可,我们讨论一边就可以了),分别解决上面四种情况:
对于情况1:当前节点是黑色的,且兄弟节点是红色的(那么父节点和兄弟节点的子节点肯定是黑色的)。如左下图所示:A节点表示当前节点。针对这种情况,我们要做的操作有:将父节点(B)涂红,将兄弟节点(D)涂黑,然后将当前节点(A)的父节点(B)作为支点左旋,然后当前节点的兄弟节点就变成黑色的情况了(自然就转换成情况2,3,4的公有特征了),如右下图所示:
对于情况2:当前节点是黑色的,且兄弟节点是黑色的,且兄弟节点的两个子节点均为黑色的。如左下图所示,A表示当前节点。针对这种情况,我们要做的操作有:将兄弟节点(D)涂红,将当前节点指向其父节点(B),将其父节点指向当前节点的祖父节点,继续新的算法(具体见下面的程序),不需要旋转。这样变成了右下图所示的情况:
对于情况3:当前节点是黑色的,且兄弟节点是黑色的,且兄弟节点的左子节点是红色,右子节点时黑色的。如左下图所示,A是当前节点。针对这种情况,我们要做的操作有:把当前节点的兄弟节点(D)涂红,把兄弟节点的左子节点(C)涂黑,然后以兄弟节点作为支点做右旋操作。然后兄弟节点就变成黑色的,且兄弟节点的右子节点变成红色的情况(情况4)了。如右下图:
对于情况4:当前节点是黑色的,且兄弟节点是黑色的,且兄弟节点的右子节点是红色,左子节点任意颜色。如左下图所示:A为当前节点,针对这种情况,我们要做的操作有:把兄弟节点(D)涂成父节点的颜色,再把父节点(B)涂黑,把兄弟节点的右子节点(E)涂黑,然后以当前节点的父节点为支点做左旋操作。至此,删除修复算法就结束了,最后将根节点涂黑即可。
我们可以看出,如果是从情况1开始发生的,可能情况2,3,4中的一种:如果是情况2,就不可能再出现3和4;如果是情况3,必然会导致情况4的出现;如果2和3都不是,那必然是4。当然咯,实际中可能不一定会从情况1发生,这要看具体情况了。
至此,我们完成了全部的删除操作。下面我们看看removeFixUp方法中的具体实现(可以结合上面的分析图,更加利与理解):
-
- private void removeFixUp(RBNode node, RBNode parent) {
- RBNode other;
-
- while((node == null || isBlack(node)) && (node != this.root)) {
- if(parent.left == node) {
- other = parent.right;
- if(isRed(other)) {
- setBlack(other);
- setRed(parent);
- leftRotate(parent);
- other = parent.right;
- }
-
-
- if((other.left == null || isBlack(other.left)) &&
- (other.right == null || isBlack(other.right))) {
- setRed(other);
- node = parent;
- parent = parentOf(node);
- } else {
-
- if(other.right == null || isBlack(other.right)) {
- setBlack(other.left);
- setRed(other);
- rightRotate(other);
- other = parent.right;
- }
-
-
- setColor(other, colorOf(parent));
- setBlack(parent);
- setBlack(other.right);
- leftRotate(parent);
- node = this.root;
- break;
- }
- } else {
- other = parent.left;
-
- if (isRed(other)) {
-
- setBlack(other);
- setRed(parent);
- rightRotate(parent);
- other = parent.left;
- }
-
- if ((other.left==null || isBlack(other.left)) &&
- (other.right==null || isBlack(other.right))) {
-
- setRed(other);
- node = parent;
- parent = parentOf(node);
- } else {
-
- if (other.left==null || isBlack(other.left)) {
-
- setBlack(other.right);
- setRed(other);
- leftRotate(other);
- other = parent.left;
- }
-
-
- setColor(other, colorOf(parent));
- setBlack(parent);
- setBlack(other.left);
- rightRotate(parent);
- node = this.root;
- break;
- }
- }
- }
- if (node!=null)
- setBlack(node);
- }
4.完整源码
终于分析完了插入和删除操作的所有东西。另外,红-黑树还有一些其他操作,比如:查找特定值、遍历、返回最值、销毁树等操作我将放到源码中给大家呈现出来,详见下面红-黑树的完整代码。
- package tree;
-
-
-
-
-
-
- public class RBTreeextends Comparable> {
- private RBNode root;
- private static final boolean RED = false;
- private static final boolean BLACK = true;
-
-
- public class RBNodeextends Comparable>{
- boolean color;
- T key;
- RBNode left;
- RBNode right;
- RBNode parent;
-
- public RBNode(T key, boolean color, RBNode parent, RBNode left, RBNode right) {
- this.key = key;
- this.color = color;
- this.parent = parent;
- this.left = left;
- this.right = right;
- }
-
- public T getKey() {
- return key;
- }
-
- public String toString() {
- return "" + key + (this.color == RED? "R" : "B");
- }
- }
-
- public RBTree() {
- root = null;
- }
-
- public RBNode parentOf(RBNode node) {
- return node != null? node.parent : null;
- }
-
- public void setParent(RBNode node, RBNode parent) {
- if(node != null)
- node.parent = parent;
- }
-
- public boolean colorOf(RBNode node) {
- return node != null? node.color : BLACK;
- }
-
- public boolean isRed(RBNode node) {
- return (node != null)&&(node.color == RED)? true : false;
- }
-
- public boolean isBlack(RBNode node) {
- return !isRed(node);
- }
-
- public void setRed(RBNode node) {
- if(node != null)
- node.color = RED;
- }
-
- public void setBlack(RBNode node) {
- if(node != null) {
- node.color = BLACK;
- }
- }
-
- public void setColor(RBNode node, boolean color) {
- if(node != null)
- node.color = color;
- }
-
-
- public void preOrder() {
- preOrder(root);
- }
-
- private void preOrder(RBNode tree) {
- if(tree != null) {
- System.out.print(tree.key + " ");
- preOrder(tree.left);
- preOrder(tree.right);
- }
- }
-
-
- public void inOrder() {
- inOrder(root);
- }
-
- private void inOrder(RBNode tree) {
- if(tree != null) {
- preOrder(tree.left);
- System.out.print(tree.key + " ");
- preOrder(tree.right);
- }
- }
-
-
- public void postOrder() {
- postOrder(root);
- }
-
- private void postOrder(RBNode tree) {
- if(tree != null) {
- preOrder(tree.left);
- preOrder(tree.right);
- System.out.print(tree.key + " ");
- }
- }
-
-
- public RBNode search(T key) {
- return search(root, key);
-
- }
-
- private RBNode search(RBNode x, T key) {
- while(x != null) {
- int cmp = key.compareTo(x.key);
- if(cmp < 0)
- x = x.left;
- else if(cmp > 0)
- x = x.right;
- else
- return x;
- }
- return x;
- }
-
- private RBNode search2(RBNode x, T key) {
- if(x == null)
- return x;
- int cmp = key.compareTo(x.key);
- if(cmp < 0)
- return search2(x.left, key);
- else if(cmp > 0)
- return search2(x.right, key);
- else
- return x;
- }
-
-
- public T minValue() {
- RBNode node = minNode(root);
- if(node != null)
- return node.key;
- return null;
- }
-
- private RBNode minNode(RBNode tree) {
- if(tree == null)
- return null;
- while(tree.left != null) {
- tree = tree.left;
- }
- return tree;
- }
-
-
- public T maxValue() {
- RBNode node = maxNode(root);
- if(node != null)
- return node.key;
- return null;
- }
-
- private RBNode maxNode(RBNode tree) {
- if(tree == null)
- return null;
- while(tree.right != null)
- tree = tree.right;
- return tree;
- }
-
-
- public RBNode successor(RBNode x) {
-
- if(x.right != null)
- return minNode(x.right);
-
-
-
- RBNode p = x.parent;
- while((p != null) && (x == p.right)) {
- x = p;
- p = x.parent;
- }
- return p;
-
- }
-
-
- public RBNode predecessor(RBNode x) {
-
- if(x.left != null)
- return maxNode(x.left);
-
-
-
- RBNode p = x.parent;
- while((p != null) && (x == p.left)) {
- x = p;
- p = x.parent;
- }
- return p;
- }
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
- private void leftRotate(RBNode x) {
-
- RBNode y = x.right;
- x.right = y.left;
-
- if(y.left != null)
- y.left.parent = x;
-
-
- y.parent = x.parent;
-
- if(x.parent == null) {
- this.root = y;
- } else {
- if(x == x.parent.left)
- x.parent.left = y;
- else
- x.parent.right = y;
- }
-
-
- y.left = x;
- x.parent = y;
- }
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
- private void rightRotate(RBNode y) {
-
- RBNode x = y.left;
- y.left = x.right;
-
- if(x.right != null)
- x.right.parent = y;
-
-
- x.parent = y.parent;
-
- if(y.parent == null) {
- this.root = x;
- } else {
- if(y == y.parent.right)
- y.parent.right = x;
- else
- y.parent.left = x;
- }
-
-
- x.right = y;
- y.parent = x;
- }
-
-
- public void insert(T key) {
- RBNode node = new RBNode(key, RED, null, null, null);
- if(node != null)
- insert(node);
- }
-
-
- private void insert(RBNode node) {
- RBNode current = null;
- RBNode x = this.root;
-
-
- while(x != null) {
- current = x;
- int cmp = node.key.compareTo(x.key);
- if(cmp < 0)
- x = x.left;
- else
- x = x.right;
- }
- node.parent = current;
-
-
- if(current != null) {
- int cmp = node.key.compareTo(current.key);
- if(cmp < 0)
- current.left = node;
- else
- current.right = node;
- } else {
- this.root = node;
- }
-
-
- insertFixUp(node);
- }
-
- private void insertFixUp(RBNode node) {
- RBNode parent, gparent;
-
-
- while(((parent = parentOf(node)) != null) && isRed(parent)) {
- gparent = parentOf(parent);
-
-
- if(parent == gparent.left) {
- RBNode uncle = gparent.right;
-
-
- if(uncle != null && isRed(uncle)) {
- setBlack(parent);
- setBlack(uncle);
- setRed(gparent);
- node = gparent;
- continue;
- }
-
-
- if(node == parent.right) {
- leftRotate(parent);
- RBNode tmp = parent;
- parent = node;
- node = tmp;
- }
-
-
- setBlack(parent);
- setRed(gparent);
- rightRotate(gparent);
- } else {
- RBNode uncle = gparent.left;
-
-
- if(uncle != null & isRed(uncle)) {
- setBlack(parent);
- setBlack(uncle);
- setRed(gparent);
- node = gparent;
- continue;
- }
-
-
- if(node == parent.left) {
- rightRotate(parent);
- RBNode tmp = parent;
- parent = node;
- node = tmp;
- }
-
-
- setBlack(parent);
- setRed(gparent);
- leftRotate(gparent);
- }
- }
-
-
- setBlack(this.root);
- }
-
-
- public void remove(T key) {
- RBNode node;
- if((node = search(root, key)) != null)
- remove(node);
- }
-
- private void remove(RBNode node) {
- RBNode child, parent;
- boolean color;
-
-
- if((node.left != null) && (node.right != null)) {
-
- RBNode replace = node;
-
- replace = replace.right;
- while(replace.left != null)
- replace = replace.left;
-
-
- if(parentOf(node) != null) {
- if(node == parentOf(node).left)
- parentOf(node).left = replace;
- else
- parentOf(node).right = replace;
- } else {
- this.root = replace;
- }
-
-
- child = replace.right;
- parent = parentOf(replace);
- color = colorOf(replace);
- if(parent == node) {
- parent = replace;
- } else {
- if(child != null)
- setParent(child, parent);
- parent.left = child;
- replace.right = node.right;
- setParent(node.right, replace);
- }
- replace.parent = node.parent;
- replace.color = node.color;
- replace.left = node.left;
- node.left.parent = replace;
-
- if(color == BLACK) {
- removeFixUp(child, parent);
- }
- node = null;
- return;
- }
- }
-
- private void removeFixUp(RBNode node, RBNode parent) {
- RBNode other;
-
- while((node == null || isBlack(node)) && (node != this.root)) {
- if(parent.left == node) {
- other = parent.right;
- if(isRed(other)) {
- setBlack(other);
- setRed(parent);
- leftRotate(parent);
- other = parent.right;
- }
-
-
- if((other.left == null || isBlack(other.left)) &&
- (other.right == null || isBlack(other.right))) {
- setRed(other);
- node = parent;
- parent = parentOf(node);
- } else {
-
- if(other.right == null || isBlack(other.right)) {
- setBlack(other.left);
- setRed(other);
- rightRotate(other);
- other = parent.right;
- }
-
-
- setColor(other, colorOf(parent));
- setBlack(parent);
- setBlack(other.right);
- leftRotate(parent);
- node = this.root;
- break;
- }
- } else {
- other = parent.left;
-
- if (isRed(other)) {
-
- setBlack(other);
- setRed(parent);
- rightRotate(parent);
- other = parent.left;
- }
-
- if ((other.left==null || isBlack(other.left)) &&
- (other.right==null || isBlack(other.right))) {
-
- setRed(other);
- node = parent;
- parent = parentOf(node);
- } else {
-
- if (other.left==null || isBlack(other.left)) {
-
- setBlack(other.right);
- setRed(other);
- leftRotate(other);
- other = parent.left;
- }
-
-
- setColor(other, colorOf(parent));
- setBlack(parent);
- setBlack(other.left);
- rightRotate(parent);
- node = this.root;
- break;
- }
- }
- }
- if (node!=null)
- setBlack(node);
- }
-
-
- public void clear() {
- destroy(root);
- root = null;
- }
-
- private void destroy(RBNode tree) {
- if(tree == null)
- return;
- if(tree.left != null)
- destroy(tree.left);
- if(tree.right != null)
- destroy(tree.right);
- tree = null;
- }
-
-
- public void print() {
- if(root != null) {
- print(root, root.key, 0);
- }
- }
-
-
-
-
-
-
- private void print(RBNode tree, T key, int direction) {
- if(tree != null) {
- if(0 == direction)
- System.out.printf("%2d(B) is root\n", tree.key);
- else
- System.out.printf("%2d(%s) is %2d's %6s child\n",
- tree.key, isRed(tree)?"R":"b", key, direction == 1?"right":"left");
- print(tree.left, tree.key, -1);
- print(tree.right, tree.key, 1);
- }
- }
- }
下面附上测试程序吧:
- package test;
-
- import tree.RBTree;
-
- public class RBTreeTest {
-
- private static final int a[] = {10, 40, 30, 60, 90, 70, 20, 50, 80};
- private static final boolean mDebugInsert = true;
- private static final boolean mDebugDelete = true;
-
- public static void main(String[] args) {
- int i, ilen = a.length;
- RBTree tree = new RBTree();
-
- System.out.printf("== 原始数据: ");
- for(i=0; i
- System.out.printf("%d ", a[i]);
- System.out.printf("\n");
-
- for(i=0; i
- tree.insert(a[i]);
-
- if (mDebugInsert) {
- System.out.printf("== 添加节点: %d\n", a[i]);
- System.out.printf("== 树的详细信息: \n");
- tree.print();
- System.out.printf("\n");
- }
- }
-
- System.out.printf("== 前序遍历: ");
- tree.preOrder();
-
- System.out.printf("\n== 中序遍历: ");
- tree.inOrder();
-
- System.out.printf("\n== 后序遍历: ");
- tree.postOrder();
- System.out.printf("\n");
-
- System.out.printf("== 最小值: %s\n", tree.minValue());
- System.out.printf("== 最大值: %s\n", tree.maxValue());
- System.out.printf("== 树的详细信息: \n");
- tree.print();
- System.out.printf("\n");
-
-
- if (mDebugDelete) {
- for(i=0; i
- {
- tree.remove(a[i]);
-
- System.out.printf("== 删除节点: %d\n", a[i]);
- System.out.printf("== 树的详细信息: \n");
- tree.print();
- System.out.printf("\n");
- }
- }
- }
-
- }
5.红-黑树的复杂度
前面也说了,当数据以升序或降序插入时,二叉搜索树的性能就会下降到最低,但是红-黑树的自我修复功能保证了即使在最坏的情况下,也能保证时间复杂度在O(logN)的级别上。
至此,红-黑树的所有内容基本上讨论完了,如有错误之处,欢迎留言指正~
_____________________________________________________________________________________________________________________________________________________
-----乐于分享,共同进步!
-----本文动态图出自:http://www.cnblogs.com/yangecnu/p/Introduce-Red-Black-Tree.html
-----本文部分参考于博客专家July的这篇文章:http://blog.csdn.net/v_july_v/article/details/6105630
-----更多文章请看:http://blog.csdn.net/eson_15