Java底层 RedBlackTree 红黑树

RedBlackTree 红黑树

---

1、2-3 树

  红黑树和2-3书有着等价的关系。我们了解了红黑树之前，必须先了解2-3这种数据结构。

1.1、2-3 树的基本结构

  2-3 树也满足树的基本结构。但是一个节点可以存储一个或者两个元素。每一个节点可以有两个或者三个孩子。

  下面是一种2-3树的结构。排序的方式是根据字符的大小。

  我们可以根据上面的结构来验证2-3树是否满足树的基本结构。
A->E->F->G->H->K->M->N->R->S->Z
  可以对树结构进行中序遍历来查看该结构是否具有顺序性。我们可以得出一个结论就是：

2-3树是一种绝对平衡的二叉树

  可以看得出来任意节点的孩子节点的高度是相同的。下面就开始介绍这种数据结构是如何维持平衡性的。

1.2、2-3树添加元素

  通过介绍如何添加元素来介绍这种树结构是如何维持平衡性的。
下面以实例进行讲解，这个地方主要还是靠大家的看。
添加的顺序是：S->E->A->R->C->H->X->M->P

最关键的还是在于变换的操作。
  由于2-3树的节点最多只能存放2个元素，当有三个元素（四节点）的时候就需要进行变换。

  对于这种分裂并不会破坏树结构的绝对平衡的性质，仍是一个绝对平衡的树结构。对于新提出来的元素需要和父亲节点进行融合。

2、红黑树与2-3树的等价性

红黑树和2-3之间可以进行等价转换。

  通过上面的等价转换可以看出来，定义红色节点都是向左倾斜。用过表示节点的颜色来识别该节点是否是红色节点。
  对于我们上面的2-3树我们可以对其进行转换，如下图：

根据上面的图例，我们可以得出下面几点结论：

• 每一个节点是要么是红色的，要么是黑色的
• 每一个叶子节点（最后的空节点）均为黑色的
• 根节点为黑色节点
• 如果一个节点为红色的，那么他的孩子节点均为黑色的（否则触发变换）
• 从任意一个节点到达叶子节点，经过的黑色节点数是相同的（绝对平衡性）

解读性质：

• 对于性质一比较好理解，一个节点的颜色只有两种状态。
• 对于性质二这里的叶子节点并不是指左右子树均为空的节点，这个性质应该叫做定义，红黑树中定义空节点的颜色就是黑色，这里和性质三相吻合
• 对于性质三，和性质二是吻合的，对于一个空树，它也是一种红黑树，那么它既是叶子节点也是根节点，所以他的节点颜色也是黑色。
• 对于性质四，可以从2-3树中得出结论，红色节点的产生于3节点，红色节点连接的子节点来自于下一个2节点或者3节点，对于2节点来说，它本身的颜色就是黑色的，对于3节点而言，拆分成红黑结构的时候，是黑色的节点连接父亲节点。所以可以得出性质四
• 对于性质五，可以通过红黑树的绝对平衡性得出，因此也可以拓展出一个结论，红黑树是一种保持黑平衡的的二叉树

3、红黑树之增加元素

  由于红黑树和2-3具有等价的关系，所以添加元素主要分为向2节点和三节点进行元素的添加，两者对应不同的情况。

3.1、向 2节点中添加元素

  由于我们添加的节点都是红色的节点，并且红色节点一直保持左倾，对于下面这种情况，就需要进行一次旋转操作。如果添加元素本身就是再左边则无需变换。变换的操作就是和AVL树一样进行一次左旋转操作，并且将颜色进行互换。如果新加入的节点本身就在左变则不需要进行变换。对应的函数为 leftRotate()

3.2、向 3节点中添加元素

  我们知道3节点添加位置主要包含三种，左边、中间和右边，下面逐一进行分析。
对于添加至右边的情况：

  对于这种情况，只需要将左右孩子节点的颜色全部变成黑色即可，我们在前面说过，新生成的父亲节点要去和上面的节点继续去融合，融合就意味着该节点的颜色为红色，也是一种类似的颜色互换的形式。对应的函数称之为颜色反转，对应的函数为flipColor()

对于添加至左边的情况：

  这种情况就是对应的右旋转过程，而且注意颜色的反转变换。右旋转过程和左旋转类似，颜色也是进行互换。最后在进行一次颜色反转操作。

对于添加至中间的情况：
  添加至中间的过程是一个组合的过程，可以先看成2节点的左旋转过程和3节点的右旋转过程。

3.3、总结

  具体的添加元素操作可以直接使用下面这张图，便于理解。

4、红黑树的实现

4.1、内部类的实现

  红黑树和二分搜索树的节点函数大致相同，只不过增加了对节点颜色的标识。
节点内部类的实现：

private class Node{  //内部类
    public K key;
    public V value;
    public  Node left, right;
    public boolean color;

    public Node(K key, V value){  // 节点信息的初始化
        this.key = key;
        this.value = value;
        left = null;
        right = null;
        color = RED;   // 新建的节点永远是融合节点,红色节点
    }
}

4.2、基本结构以及函数实现

红黑树的基本架构程序实现：

public class RBTree<K extends Comparable<K>, V extends Comparable<V>>{

    private static final boolean RED = true;
    private static final boolean BLACK = false;
    private class Node{  
        //内部类 省略
    }
    private Node root;
    private int size;

    public RBTree(){
        root = null;
        size = 0;
    }
}

  对于红黑树这种结构，我们增加了对颜色的判断，所有我们引入isRed函数来判断这个节点是否是红色的。
程序实现：

private boolean isRed(Node node) {
    if (node == null)
        return BLACK;
    return node.color;
}

  基本辅助函数的实现，主要是对成员变量的一个get操作以及判断红黑树是否为空。
程序实现：

public int getSize() {
    return size;
}

public boolean isEmpty() {
    return size == 0;
}

4.3、增加元素

  我们根据上面的性质可以得到，根节点的颜色必须为黑色，所以添加完元素后的根节点我们需要手动指定为黑色，因为我们新增的节点永远是红色的，所以我们需要手动将root变为黑色，调用方式仍然是调用私有递归函数。

public void add(K key, V value) {
    root = add(root, key, value);
    root.color = BLACK; //手动将根节点变为黑色
}

  在2节点中添加元素，并且在节点右边添加元素会触发左旋转，进而保证红色节点的左倾性。
左旋转程序实现：

private Node leftRotate(Node node) {
    Node x = node.right;
    // 左旋转
    node.right = x.left;
    x.left = node;
    //颜色互换
    x.color = node.color;
    node.color = RED;

    return x;
}

  在3节点中添加元素涉及左中右三种情况，上面讲过，在右边添加元素触发颜色反装。
颜色反转程序实现：

private void flipColors(Node node) {
    node.color = RED;
    node.left.color = BLACK;
    node.right.color = BLACK;
}

  在向三节点中最左边添加元素，触发右旋转过程。右旋转过程中还要进行一次颜色反转。
右旋转程序实现：

private Node rightRotate(Node node) {
    Node x = node.left;
    //右旋转
    node.left = x.right;
    x.right = node;
    //颜色互换
    x.color = node.color;
    node.color = RED;
    return x;
}

  其他的添加元素的变换都是基于颜色变换，左右旋转组合而来的。

  具体的调用方法和AVL树相同，在增加完元素的时候，对元素进行判断，是一个在回朔的过程中进行操作。
增加函数的程序实现：

private Node add(Node node, K key, V value) {
    if (node == null){
        size++;
        return new Node(key, value);
    }
    //首先进行元素的添加
    if (key.compareTo(node.key) < 0)
        node.left = add(node.left, key, value);
    else if (key.compareTo(node.key) > 0)
        node.right = add(node.right, key, value);
    else
        node.key = key;

    //添加完成后，维护平衡性
    //左节点为黑，右节点为红
    if (isRed(node.right) && !isRed(node.left)) 
        node = leftRotate(node);
    //左节点为红，左节点的左节点也为红
    if (isRed(node.left) && isRed(node.left.left))
        node = rightRotate(node);
    //左右节点均为红
    if (isRed(node.left) && isRed(node.right))
        flipColors(node);

    return node;
}

5、时间复杂度分析

  从严格意义上讲，红黑树并不是一种严格的平衡二叉树。在最差的情况下，红黑树的高度为2log(n)，这是因为在最差的情况下，每一个黑节点都有一个红色节点，但是由于时间复杂度不考虑红黑树的时间复杂度也是O(log(n))级别的。

适应场景:
  红黑树适应于添加删除元素频繁的场景。而对于查询元素来说，红黑树并不是严格意义上的平衡树，在最坏的情况树的高度可以达到2log(n)，所以说不同的场景适应于不同的算法。但是综合性能来说的话，红黑树的性能任然高于一些其他的树结构。

最后

更多精彩内容，大家可以转到我的主页：曲怪曲怪的主页

或者关注我的微信公众号：TeaUrn

源码地址：可在公众号内回复 数据结构与算法源码 即可获得。