vlambda博客
学习文章列表

红黑树是怎么实现的,看这篇真的就够了!

每天早上七点三十分,准时推送干货

大家好,我是鸭血粉丝,又是一个元气满满的周一,今天带大家一文搞懂红黑树,友情提示:本文较长,建议先收藏再观看。

红黑树由来:在1972年由Rudolf Bayer发明的,当时被称为平衡二叉B树(symmetric binary B-trees),后来,在1978年被 Leo J. Guibas 和 Robert Sedgewick 修改为如今的红黑树,就此红黑树出现在软件开发者的视野里!

一、摘要

在上篇文章中,我们详细的介绍到平衡二叉查找树的算法以及代码实践,我们知道平衡二叉查找树是一个高度平衡的二叉树,也就是说树的高度差不能大于1,在删除的时候,可能需要多次调整,也就是左旋转、右旋转操作,在树的深度很大的情况下,删除效率会非常低,如何提高这种效率?

红黑树由此诞生了,了解过红黑树的朋友们一定知道,红黑树是一个基本平衡的二叉树,英文全称:red-black-tree,简称:RBT,特性如下:

  • 1.每个节点要么是黑色要么是红色;

  • 2.根节点是黑色;

  • 3.每个叶子节点是黑色;(注意:这里叶子节点,是指为空的叶子节点)

  • 4.如果一个节点是红色的,则它的子节点必须是黑色的;(说明父子节点之间不能出现两个连续的红节点)

  • 5.从一个节点到该节点的子孙节点的所有路径上包含相同数目的黑节点;

关于它的特性,需要注意的是:

  • 特性3中的叶子节点,是只为空(NIL 或 null)的节点;

  • 特性5,确保没有一条路径会比其他路径长出俩倍,因而,红黑树是相对的接近平衡的二叉树!(比如,包含相同数目为3的黑节点路线,最短路线:`黑节点 -> 黑节点 -> 黑节点`,长度为3;最长路线:`黑节点 -> 红节点 -> 黑节点 -> 红节点 -> 黑节点`,长度为5)

红黑树示例图,如下:

红黑树,与二叉树,在查询、插入、删除方面,都是一样的,最大的不同点是,插入、删除需要重新调整树的形态,以保持红黑树的特性!

二、算法思路

在上篇平衡二叉树的文章中,我们了解到为了保证树的高度差不超过1,我们通过树高超过1这么一个判断条件,通过左旋转、右旋转来调整,从而保证树的高度平衡!

对于红黑树,其实也是一样的,对于插入、删除操作,主要也是通过左旋转、右旋转来进行调整,相比平衡二叉树,红黑树因为节点有颜色标签,多了一个颜色转变操作!

同理,我们只需要分析出哪些场景下需要进行调整,即可总结出算法,从而写出实践代码!

废话也不多说来,下面我们一起来分析一下红黑树,在插入、删除操作时,应该怎么处理!

2.1、插入场景

将一个节点插入到红黑树中,需要执行哪些步骤呢?

  • 第一步:将红黑树当作一颗二叉查找树,将节点插入树的底部;

  • 第二步:默认将插入的节点着色为红色,如果是根节点,颜色着为黑色;

  • 第三步:通过一系列的旋转或着色等操作,使之重新成为一颗红黑树;

对于第一步,比较好理解,红黑树其实就是二叉树的一种特殊形态的树形结构,先找到合适的位置,然后将节点插入到树上。

红黑树是怎么实现的,看这篇真的就够了!

对于第二步,为什么新插入的节点要设置为红色呢?

因为插入之前所有根至外部节点的路径上黑色节点数目都相同,所以如果插入的节点是黑色,肯定会导致黑色节点数目不相同!

而相对的插入红节点可能也会违反不能出现两个连续的红节点,如果违反条件,直接进行颜色转换或者旋转操作即可!

相对将插入的节点着色为黑色,红色操作可能更简单些!因为根节点为黑色,如果是新插入的节点为根节点,直接将颜色设置为黑色!

既然新插入的节点为红色,那么我们就继续来分析一下,新插入节点的场景,例如:

  • 1.插入的节点,父亲为黑色;

  • 2.插入的节点,父亲为红色;

当新插入的节点的父亲为黑色时!因为新插入的节点为红色,因此不会违反任何特性!

红黑树是怎么实现的,看这篇真的就够了!

当新插入的节点的父亲为红色时!因为新插入的节点为红色,违反不能出现两个连续的红节点,因此需要进行调整!

红黑树是怎么实现的,看这篇真的就够了!

这种场景有3种调整情况,为了便于下面分析,假设将新插入节点用z代替,z的父节点用a代替,a的父节点用c代替,z的叔节点用y代替,如下:

红黑树是怎么实现的,看这篇真的就够了!

情况1:z的叔节点y是红色的

case1调整过程如下:

红黑树是怎么实现的,看这篇真的就够了!

调整说明:因为节点 z 是一个红色节点,其父节点 a 是红色的,违反了特性4,因此需要进行调整。因为其叔节点 y 是红色的,于是可以修改节点 a、节点 y 为黑色,此时节点 c 的黑高会发生变化,由原来的黑高1变成黑高2,为了节点 c 保持黑高不变,将其变成红色。

此时,由于节点 c 由黑色变成了红色,如果节点 c 的父节点是红色,也会违反特性4,继续将节点 c 看成是节点 z,向上回溯调整树!

需要注意的是:对于新插入的节点 z 是节点a 的左子树的情况,操作与上述一致;对于新插入的节点 z 是节点 c 的右子树的节点的情况,操作与上述对称!

情况2:z的叔节点y是黑色,且z是一个左孩子

case2调整过程如下:

红黑树是怎么实现的,看这篇真的就够了!

调整说明:这种情况,并不能像上面那样改变节点颜色就可以满足要求,因为如果将节点z 的父节点 a 变成了黑色, 那么树的黑高就会发生变化,必然会引起对性质5的违反。比如,此时节点y为黑色, 节点c 的右子树高度为2(忽略子树),左子树高也相同,如果简单的修改节点a 为黑色,那么节点c 的左子树的黑高会比右子树大1, 此时即使将节点c 修改为红色也于事无补!

因此,单靠颜色转变无非解决问题,需要进行旋转调整。先绕节点 a 的父节点进行右旋转,再将节点 a、节点 c 的颜色进行互换!最终结果与插入前一致!

需要注意的是:对于新插入的节点 z 是节点 c 的右子树的节点的情况,操作与上述对称!

情况3:z的叔节点y是黑色,且z是一个右孩子

case3调调整过程如下:

红黑树是怎么实现的,看这篇真的就够了!

调整说明:当节点 z 是一个右孩子时,先绕节点 a 进行左旋转,之后树的形态就变成如上面介绍的情况2,再进行右旋转、颜色转变,即可实现红黑树的特性!

需要注意的是:对于新插入的节点 z 是节点 c 的右子树的节点的情况,操作与上述对称!

以上就是红黑树新增节点时,所有可能的操作以及调整方式!

可以得出如下结论:对插入节点后的调整所做的旋转操作不会超过2次!

2.2、删除场景

我们继续来看看删除场景,对于二叉查找树操作,我们知道有如下步骤:

  • 当删除节点,只有左子树时,将右子树向上移动;

  • 当删除节点,只有右子树时,将左子树向上移动;

  • 当删除节点,有左、右子树时,通过找到删除节点的右节点的最末端左子树,也就是后继节点,进行替换并删除;

二叉查找树删除过程图:

红黑树是怎么实现的,看这篇真的就够了!

红黑树的操作也是如此,步骤如下;

  • 第一步:将红黑树当作一颗二叉查找树,将节点删除;

  • 第二步:通过旋转和变色等一系列来修正该树,使之重新成为一棵红黑树;

在第一步中,首先我们重点要弄清楚什么是删除节点?

这个删除节点,某些情况下并非我们真正传入的删除值,对于拥有左子树、右子树的节点来说,删除节点指的是被删除节点的后继节点或者前驱节点!

可能有点绕,例如上图中拥有左子树、右子树的删除场景,我们传入需要删除的节点是 80,但是实际上删除节点的是85节点(85是一个叶子节点),然后将80节点内容替换成85,做了一个偷天换日的操作!

因此无论对于哪种情况,我们可以得出结论:删除节点一定是一个单分支节点或者叶子节点,了解这个结论对于后面我们的红黑树删除过程分析会非常有用!

清楚删除流程之后,剩下的重点就是如何进行修正,以满足红黑树的特性,那么,哪些场景下的删除需要进行调整,我们一起来看一下!

2.2.1、删除的节点为红色

当删除的节点为红色时,这种情况直接将删除节点移除,理由如下:

  • 树中各个节点的黑高没有变化;

  • 删除后满足性质4,因为不会出现红红相连的情况;

  • 删除的不可能是根节点,因为根节点是黑色的;

2.2.2、删除的节点为黑色

当删除的节点为黑色时,因为删除节点的父节点失去了一个黑色子节点,这种情况会导致左右子树不平衡,因此需要进行调整,假设x为被删节点的替换节点,也就是说:

  • 当被删节点的左子树为空时,x为被删节点的右孩子;

  • 当被删节点的右子树为空时,x为被删节点的左孩子;

  • 当被删节点的左、右子树都为空时,x是空节点(即删除的是终端节点);

  • 当被删节点的左、右子树都不为空时,x为被删节点的右子树的最末端的左子树,也就是中序遍历直接后继的右孩子;

wx的兄弟节点,有以下几种情况!

1)x的兄弟节点w是红色的

case1调整过程如下:

红黑树是怎么实现的,看这篇真的就够了!

调整说明:因为节点 c 的左子树被删去了一个黑色节点,导致节点 c 的左子树黑高少了1,所以节点c 是不平衡的。可以对节点c 进行一次左旋,然后修改节点80和节点120 的颜色。

此时,x的父亲节点c依然不平衡,节点x 的兄弟节点w 变成黑色的!

继续看下面的分析,这种不平衡由下面的几种情况进行处理!

2)x的兄弟节点w是黑色的,并且w的两个子节点都是黑色的

当x的兄弟节点 w 是黑色的,并且w的两个子节点都是黑色的时,此时需要分两种情况!

情况2.1:x的父节点为红色

case2.1调整过程如下:

红黑树是怎么实现的,看这篇真的就够了!

调整说明:在删除节点后,左图节点 c 不平衡,节点c 左子树的黑高为hl+1,节点c 左子树的黑高为hr+2,左子树黑高小于右子树黑高。因此直接将节点c 修改为黑色,节点 w 修改为红色,此时的黑高又恢复如初!但是节点c 作为子节点,因为黑高减少,需要继续向上回溯调整树的黑高,此时节点c 作为新的节点x。

情况2.2:x的父节点为黑色

case2.2调整过程如下:

红黑树是怎么实现的,看这篇真的就够了!

调整说明:只需要将节点 w 修改为红色,继续向上回溯调整树的黑高,此时节点 c 作为新的节点x。

3)x的兄弟节点w是黑色的,并且w的右孩子是红色的

当x的兄弟节点w是黑色的,并且w的右孩子是红色的,此时也需要分四种情况!

情况3.1:x的父节点为黑色,w的左孩子是黑色的

case3.1调整过程如下:

红黑树是怎么实现的,看这篇真的就够了!

调整说明:因为删除节点导致节点c 不平衡,对节点c 进行一次左旋转,将节点w 的右孩子颜色修改为黑色。此时节点c 已经达到平衡,同时节点w 也达到平衡,整棵树已经平衡了!

情况3.2:x的父节点为黑色,w的左孩子是红色的

case3.2调整过程如下:

红黑树是怎么实现的,看这篇真的就够了!

调整说明:同样的,对节点c 进行一次左旋转,将节点w 的右孩子颜色修改为黑色。此时节点c 已经达到平衡,同时节点w 也达到平衡,整棵树已经平衡了!

情况3.3:x的父节点为红色,w的左孩子是黑色的

case3.3调整过程如下:

红黑树是怎么实现的,看这篇真的就够了!

调整说明:对节点c 进行一次左旋转,将节点 w 的右孩子颜色修改为黑色,同时将节点80 和节点100 颜色进行互换。此时节点c 已经达到平衡,同时节点w 也达到平衡,整棵树已经平衡了!

情况3.4:x的父节点为红色,w的左孩子是红色的

case3.4调整过程如下:

红黑树是怎么实现的,看这篇真的就够了!

调整说明:同样的,对节点c 进行一次左旋转,将节点 w 的右孩子颜色修改为黑色,同时将节点80 和节点100 颜色进行互换。此时节点c 已经达到平衡,同时节点w 也达到平衡,整棵树已经平衡了!

4)x的兄弟节点w是黑色的,而且w的左孩子是红色的,w的右孩子是黑色的

当x的兄弟节点w是黑色的,而且w的左孩子是红色的,w的右孩子是黑色的,此时也需要分2种情况!

情况4.1:x的父节点为红色

case4.1调整过程如下:

红黑树是怎么实现的,看这篇真的就够了!

调整说明:对节点w 进行一次右旋转,将节点90和节点100 进行颜色互换,此时节点x 和节点w 的关系变成:x的兄弟节点w是黑色的,并且w的右孩子是红色的。此时按照case3.3情况进行处理即可!

情况4.2:x的父节点为黑色

case4.2调整过程如下:

调整说明:同样的,对节点w 进行一次右旋转,将节点90和节点100进行颜色互换,此时节点x 和节点w 的关系变成:x的兄弟节点w是黑色的,并且w的右孩子是红色的。此时按照case3.1情况进行处理即可!

删除的场景相比插入要多很多,情况也比较复杂,但是基本有自己的规律,我们只需要把规律总结出来,然后就可以用逻辑代码来实现!

需要注意的是:此次节点x 位于节点c 的左子树,如果位于右子树,操作与之对称!

可以得出如下结论:对删除节点后的调整所做的旋转操作不会超过3次!

三、代码实践

接下来,我们从代码层面来定义一下树的实体结构,如下:

public class RBTNode<E extends Comparable<E>> {

    /**节点颜色*/
    boolean color;

    /**节点关键字*/
    E key;

    /**父节点*/
    RBTNode<E> parent;

    /**左子节点*/
    RBTNode<E> left;

    /**右子节点*/
    RBTNode<E> right;

    public RBTNode(E key, boolean color, RBTNode<E> parent, RBTNode<E> left, RBTNode<E> right) {
        this.key = key;
        this.color = color;
        this.parent = parent;
        this.left = left;
        this.right = right;
    }

    @Override
    public String toString() {
        return "RBTNode{" +
                "color=" + (color ? "red":"black") +
                ", key=" + key +
                '}';
    }
}

我们创建一个算法类RBTSolution,完整实现如下:

public class RBTSolution<E extends Comparable<E>> {

    /**根节点*/
    public RBTNode<E> root;

    /**
     * 颜色常量 false表示红色,true表示黑色
     */

    private static final boolean RED = false;
    private static final boolean BLACK = true;


    /*
     * 新建结点(key),并将其插入到红黑树中
     * 参数说明:key 插入结点的键值
     */

    public void insert(E key) {
        System.out.println("插入[" + key + "]:");
        RBTNode<E> node=new RBTNode<E>(key, BLACK,null,null,null);
        // 如果新建结点失败,则返回。
        if (node != null)
            insert(node);
    }

    /*
     * 将结点插入到红黑树中
     *
     * 参数说明:
     *     node 插入的结点        // 对应《算法导论》中的node
     */

    private void insert(RBTNode<E> node) {
        int cmp;
        RBTNode<E> y = null;
        RBTNode<E> x = this.root;

        // 1. 将红黑树当作一颗二叉查找树,将节点添加到二叉查找树中。
        while (x != null) {
            y = x;
            cmp = node.key.compareTo(x.key);
            if (cmp < 0)
                x = x.left;
            else
                x = x.right;
        }

        node.parent = y;
        if (y!=null) {
            cmp = node.key.compareTo(y.key);
            if (cmp < 0)
                y.left = node;
            else
                y.right = node;
        } else {
            this.root = node;
        }

        // 2. 设置节点的颜色为红色
        node.color = RED;

        // 3. 将它重新修正为一颗二叉查找树
        insertFixUp(node);
    }

    /*
     * 红黑树插入修正函数
     *
     * 在向红黑树中插入节点之后(失去平衡),再调用该函数;
     * 目的是将它重新塑造成一颗红黑树。
     *
     * 参数说明:
     *     node 插入的结点        // 对应《算法导论》中的z
     */

    private void insertFixUp(RBTNode<E> node) {
        RBTNode<E> parent, gparent;

        // 若“父节点存在,并且父节点的颜色是红色”
        while (((parent = parentOf(node))!=null) && isRed(parent)) {
            gparent = parentOf(parent);

            //若“父节点”是“祖父节点的左孩子”
            if (parent == gparent.left) {
                // Case 1条件:叔叔节点是红色
                RBTNode<E> uncle = gparent.right;
                if ((uncle!=null) && isRed(uncle)) {
                    setBlack(uncle);
                    setBlack(parent);
                    setRed(gparent);
                    node = gparent;
                    continue;
                }

                // Case 3条件:叔叔是黑色,且当前节点是右孩子
                if (parent.right == node) {
                    RBTNode<E> tmp;
                    leftRotate(parent);
                    tmp = parent;
                    parent = node;
                    node = tmp;
                }

                // Case 2条件:叔叔是黑色,且当前节点是左孩子。
                setBlack(parent);
                setRed(gparent);
                rightRotate(gparent);
            } else {    //若“z的父节点”是“z的祖父节点的右孩子”
                // Case 1条件:叔叔节点是红色
                RBTNode<E> uncle = gparent.left;
                if ((uncle!=null) && isRed(uncle)) {
                    setBlack(uncle);
                    setBlack(parent);
                    setRed(gparent);
                    node = gparent;
                    continue;
                }

                // Case 2条件:叔叔是黑色,且当前节点是左孩子
                if (parent.left == node) {
                    RBTNode<E> tmp;
                    rightRotate(parent);
                    tmp = parent;
                    parent = node;
                    node = tmp;
                }

                // Case 3条件:叔叔是黑色,且当前节点是右孩子。
                setBlack(parent);
                setRed(gparent);
                leftRotate(gparent);
            }
        }

        // 将根节点设为黑色
        setBlack(this.root);
    }

    /*
     * 删除结点(z),并返回被删除的结点
     *
     * 参数说明:
     *     tree 红黑树的根结点
     *     z 删除的结点
     */

    public void remove(E key) {
        RBTNode<E> node;

        if ((node = search(root, key)) != null)
            remove(node);
    }


    /*
     * 删除结点(node),并返回被删除的结点
     *
     * 参数说明:
     *     node 删除的结点
     */

    private void remove(RBTNode<E> node) {
        RBTNode<E> child, parent;
        boolean color;

        // 被删除节点的"左右孩子都不为空"的情况。
        if ( (node.left!=null) && (node.right!=null) ) {
            // 被删节点的后继节点。(称为"取代节点")
            // 用它来取代"被删节点"的位置,然后再将"被删节点"去掉。
            RBTNode<E> replace = node;

            // 获取后继节点
            replace = replace.right;
            while (replace.left != null)
                replace = replace.left;

            // "node节点"不是根节点(只有根节点不存在父节点)
            if (parentOf(node)!=null) {
                if (parentOf(node).left == node)
                    parentOf(node).left = replace;
                else
                    parentOf(node).right = replace;
            } else {
                // "node节点"是根节点,更新根节点。
                this.root = replace;
            }

            // child是"取代节点"的右孩子,也是需要"调整的节点"。
            // "取代节点"肯定不存在左孩子!因为它是一个后继节点。
            child = replace.right;
            parent = parentOf(replace);
            // 保存"取代节点"的颜色
            color = colorOf(replace);

            // "被删除节点"是"它的后继节点的父节点"
            if (parent == node) {
                parent = replace;
            } else {
                // child不为空
                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 ;
        }

        if (node.left !=null) {
            child = node.left;
        } else {
            child = node.right;
        }

        parent = node.parent;
        // 保存"取代节点"的颜色
        color = node.color;

        if (child!=null)
            child.parent = parent;

        // "node节点"不是根节点
        if (parent!=null) {
            if (parent.left == node)
                parent.left = child;
            else
                parent.right = child;
        } else {
            this.root = child;
        }

        if (color == BLACK)
            removeFixUp(child, parent);
        node = null;
    }

    /*
     * 红黑树删除修正函数
     *
     * 在从红黑树中删除插入节点之后(红黑树失去平衡),再调用该函数;
     * 目的是将它重新塑造成一颗红黑树。
     *
     * 参数说明:
     *     node 待修正的节点
     */

    private void removeFixUp(RBTNode<E> node, RBTNode<E> parent) {
        RBTNode<E> other;

        while ((node==null || isBlack(node)) && (node != this.root)) {
            if (parent.left == node) {
                other = parent.right;
                if (isRed(other)) {
                    // Case 1: x的兄弟w是红色的
                    setBlack(other);
                    setRed(parent);
                    leftRotate(parent);
                    other = parent.right;
                }

                if ((other.left==null || isBlack(other.left)) &&
                        (other.right==null || isBlack(other.right))) {
                    // Case 2: x的兄弟w是黑色,且w的俩个孩子也都是黑色的
                    setRed(other);
                    node = parent;
                    parent = parentOf(node);
                } else {

                    if (other.right==null || isBlack(other.right)) {
                        // Case 4: x的兄弟w是黑色的,并且w的左孩子是红色,右孩子为黑色。
                        setBlack(other.left);
                        setRed(other);
                        rightRotate(other);
                        other = parent.right;
                    }
                    // Case 3: x的兄弟w是黑色的;并且w的右孩子是红色的,左孩子任意颜色。
                    setColor(other, colorOf(parent));
                    setBlack(parent);
                    setBlack(other.right);
                    leftRotate(parent);
                    node = this.root;
                    break;
                }
            } else {

                other = parent.left;
                if (isRed(other)) {
                    // Case 1: x的兄弟w是红色的
                    setBlack(other);
                    setRed(parent);
                    rightRotate(parent);
                    other = parent.left;
                }

                if ((other.left==null || isBlack(other.left)) &&
                        (other.right==null || isBlack(other.right))) {
                    // Case 2: x的兄弟w是黑色,且w的俩个孩子也都是黑色的
                    setRed(other);
                    node = parent;
                    parent = parentOf(node);
                } else {

                    if (other.left==null || isBlack(other.left)) {
                        // Case 4: x的兄弟w是黑色的,并且w的左孩子是红色,右孩子为黑色。
                        setBlack(other.right);
                        setRed(other);
                        leftRotate(other);
                        other = parent.left;
                    }

                    // Case 3: x的兄弟w是黑色的;并且w的右孩子是红色的,左孩子任意颜色。
                    setColor(other, colorOf(parent));
                    setBlack(parent);
                    setBlack(other.left);
                    rightRotate(parent);
                    node = this.root;
                    break;
                }
            }
        }

        if (node!=null)
            setBlack(node);
    }


    /**
     * 查询节点
     * @param key
     * @return
     */

    public RBTNode<E> search(E key) {
        return search(root, key);
    }


    /*
     * (递归实现)查找"红黑树x"中键值为key的节点
     */

    private RBTNode<E> search(RBTNode<E> x, E key) {
        if (x==null)
            return x;

        int cmp = key.compareTo(x.key);
        if (cmp < 0)
            return search(x.left, key);
        else if (cmp > 0)
            return search(x.right, key);
        else
            return x;
    }

    /**
     * 中序遍历
     * @param node
     */

    public void middleTreeIterator(RBTNode<E> node){
        if(node != null){
            middleTreeIterator(node.left);//遍历当前节点左子树
            System.out.println("key:" + node.key);
            middleTreeIterator(node.right);//遍历当前节点右子树
        }
    }



    private RBTNode<E> parentOf(RBTNode<E> node) {
        return node!=null ? node.parent : null;
    }
    private boolean colorOf(RBTNode<E> node) {
        return node!=null ? node.color : BLACK;
    }
    private boolean isRed(RBTNode<E> node) {
        return ((node!=null)&&(node.color==RED)) ? true : false;
    }
    private boolean isBlack(RBTNode<E> node) {
        return !isRed(node);
    }
    private void setBlack(RBTNode<E> node) {
        if (node!=null)
            node.color = BLACK;
    }
    private void setRed(RBTNode<E> node) {
        if (node!=null)
            node.color = RED;
    }
    private void setParent(RBTNode<E> node, RBTNode<E> parent) {
        if (node!=null)
            node.parent = parent;
    }
    private void setColor(RBTNode<E> node, boolean color) {
        if (node!=null)
            node.color = color;
    }


    /*
     * 对红黑树的节点(x)进行左旋转
     *
     * 左旋示意图(对节点x进行左旋):
     *      px                              px
     *     /                               /
     *    x                               y
     *   /  \      --(左旋)-.           / \                #
     *  lx   y                          x  ry
     *     /   \                       /  \
     *    ly   ry                     lx  ly
     *
     *
     */

    private void leftRotate(RBTNode<E> x) {
        // 设置x的右孩子为y
        RBTNode<E> y = x.right;

        // 将 “y的左孩子” 设为 “x的右孩子”;
        // 如果y的左孩子非空,将 “x” 设为 “y的左孩子的父亲”
        x.right = y.left;
        if (y.left != null)
            y.left.parent = x;

        // 将 “x的父亲” 设为 “y的父亲”
        y.parent = x.parent;

        if (x.parent == null) {
            this.root = y;            // 如果 “x的父亲” 是空节点,则将y设为根节点
        } else {
            if (x.parent.left == x)
                x.parent.left = y;    // 如果 x是它父节点的左孩子,则将y设为“x的父节点的左孩子”
            else
                x.parent.right = y;    // 如果 x是它父节点的左孩子,则将y设为“x的父节点的左孩子”
        }

        // 将 “x” 设为 “y的左孩子”
        y.left = x;
        // 将 “x的父节点” 设为 “y”
        x.parent = y;
    }

    /*
     * 对红黑树的节点(y)进行右旋转
     *
     * 右旋示意图(对节点y进行左旋):
     *            py                               py
     *           /                                /
     *          y                                x
     *         /  \      --(右旋)-.            /  \                     #
     *        x   ry                           lx   y
     *       / \                                   / \                   #
     *      lx  rx                                rx  ry
     *
     */

    private void rightRotate(RBTNode<E> y) {
        // 设置x是当前节点的左孩子。
        RBTNode<E> x = y.left;

        // 将 “x的右孩子” 设为 “y的左孩子”;
        // 如果"x的右孩子"不为空的话,将 “y” 设为 “x的右孩子的父亲”
        y.left = x.right;
        if (x.right != null)
            x.right.parent = y;

        // 将 “y的父亲” 设为 “x的父亲”
        x.parent = y.parent;

        if (y.parent == null) {
            this.root = x;            // 如果 “y的父亲” 是空节点,则将x设为根节点
        } else {
            if (y == y.parent.right)
                y.parent.right = x;    // 如果 y是它父节点的右孩子,则将x设为“y的父节点的右孩子”
            else
                y.parent.left = x;    // (y是它父节点的左孩子) 将x设为“x的父节点的左孩子”
        }

        // 将 “y” 设为 “x的右孩子”
        x.right = y;

        // 将 “y的父节点” 设为 “x”
        y.parent = x;
    }
}

测试代码,如下:

public class RBTClient {

    public static void main(String[] args) {
        RBTSolution<Integer> tree = new RBTSolution<Integer>();
        //插入节点
        System.out.println("========插入元素========");
        tree.insert(new Integer(100));
        tree.insert(new Integer(85));
        tree.insert(new Integer(120));
        tree.insert(new Integer(60));
        tree.insert(new Integer(90));
        tree.insert(new Integer(80));
        tree.insert(new Integer(130));
        System.out.println("========中序遍历元素========");

        //中序遍历
        tree.middleTreeIterator(tree.root);
        System.out.println("========查找key为100的元素========");

        //查询节点
        RBTNode<Integer> searchResult = tree.search(100);
        System.out.println("查找结果:" + searchResult);
        System.out.println("========删除key为90的元素========");

        //删除节点
        tree.remove(90);
        System.out.println("========再次中序遍历元素========");

        //中序遍历
        tree.middleTreeIterator(tree.root);
    }
}

输出结果如下:

========插入元素========
插入[100]:
插入[85]:
插入[120]:
插入[60]:
插入[90]:
插入[80]:
插入[130]:
========中序遍历元素========
key:60
key:80
key:85
key:90
key:100
key:120
key:130
========查找key为100的元素========
查找结果:RBTNode{color=red, key=100}
========删除key为90的元素========
========再次中序遍历元素========
key:60
key:80
key:85
key:100
key:120
key:130

四、总结

本篇文章,前前后后写了大概一个星期,尤其是删除逻辑,比较复杂,如果有理解不对的地方,欢迎网友们指出!

以下是红黑树总结:

  • 1、红黑树是一个基本平衡的二叉树,在查询方面,与二叉查找树思路相同;在插入方面,单次回溯不会超过2次旋转;在删除方面,单次回溯不会超过3次旋转!

  • 2、相比于平衡二叉树,红黑树在查询、插入方面,效率差不多;在删除方面,平衡二叉树最多需要log(n)次变化,以达到严格平衡,而红黑树最多不会超过3次变化,因此效率要高于平衡二叉树!

  • 3、在实际应用中,很多语言都实现了红黑树的数据结构,比如 Java 中的 TreeMap、 TreeSet,以及 jdk1.8 中的 HashMap!

五、参考

1、简书 - 遇见技术 - JAVA学习-红黑树详解

2、博客园 - skywang12345 - 红黑树(五)之 Java的实现

3、ivanzz1001 - 红黑树的原理及实现

< END >

如果大家喜欢我们的文章,欢迎大家转发,点击在看让更多的人看到。也欢迎大家热爱技术和学习的朋友加入的我们的知识星球当中,我们共同成长,进步。


往期 精彩回顾