SL_World

红黑树原理浅谈(附Linux内核源码注释)

引言：红黑树（英语：Red–black tree）是一种自平衡二叉查找树，是在计算机科学中用到的一种数据结构，典型的用途是实现关联数组。它是在1972年由鲁道夫·贝尔发明的，他称之为"对称二叉B树"，它现代的名字是在Leo J. Guibas和Robert Sedgewick于1978年写的一篇论文中获得的。它是复杂的，但它的操作有着良好的最坏情况运行时间，并且在实践中是高效的：它可以在 O(log n) 时间内做查找，插入和删除，这里的n是树中元素的数目。——摘自维基百科

红黑树的性质：

红黑树是每个节点都带有颜色属性的二叉查找树，颜色为红色或黑色。在二叉查找树强制一般要求以外，对于任何有效的红黑树有如下的额外要求：

1.节点是红色或黑色。
2.根是黑色。
3.所有叶子都是黑色（叶子是NIL节点）。
4.每个红色节点必须有两个黑色的子节点。（从每个叶子到根的所有路径上不能有两个连续的红色节点。）
5.从任一节点到其每个叶子的所有简单路径都包含相同数目的黑色节点。

点击此处可视化网站可查看任意数据结构动态图

操作：

因为每一个红黑树也是一个特化的二叉查找树，因此红黑树上的只读操作与普通二叉查找树上的只读操作相同。然而，在红黑树上进行插入操作和删除操作会导致不再匹配红黑树的性质。恢复红黑树的性质需要少量 O(log n)的颜色变更（实际是非常快速的）和不超过三次树旋转（对于插入操作是两次）。虽然插入和删除很复杂，但操作时间仍可以保持为 O(log n)次。

红黑树的应用

它的统计性能要好于平衡二叉树（AVL树），因此，红黑树在很多地方都有应用。比如
①JDK1.8中的TreeMap 或
②C++的STL中的set和map函数或
③Linux内核中的rbtree.h和rbtree.c
都有红黑树的应用，这些集合均提供了很好的性能。

什么时候用AVL树？什么时候用红黑树？

首先AVL树和红黑树都是基于BST树(二叉搜索树)，对于只读操作均和BST树相同，但BST树连续插入顺序递增或递减结点时会导致BST树退化成链表，性能大幅度降低。BST顺序插入如图：

因此，AVL树和红黑树均在BST树的基础上进行了进一步的优化。以增加旋转操作(增加算法复杂度)来提升对于最坏情况下的性能。提升复杂度往往从两方面入手：
①提升数据结构的复杂度
②提升算法的复杂度
其中，红黑树为了降低原有算法复杂度，就以提升数据结构复杂度来间接减轻其算法复杂度。即在其结点中增加颜色属性，颜色只能有红色和黑色。在红黑树中只要任意结点左右孩子的黑色高度平衡(两边任意路径黑色结点数量相同)即可且对于有些情况可以通过变色来代替旋转从而减少一定的空间开销。

①AVL树采用平衡因子的绝对值<2来保证平衡，这种平衡是高度平衡的。要保证这种平衡需要更复杂的操作来维持。
②红黑树采用同一结点到叶结点的任意路径黑色结点数量相同来保证平衡，这种平衡只能保证从根到叶子的最长的可能路径不多于最短的可能路径的两倍长。保证这种平衡只需要较少的操作来维持。

下图是从根到叶子的最长的可能路径=最短的可能路径的两倍长的情况：

综上所述：删除结点最坏情况下，AVL树需要旋转的量级是O(log n)，而红黑树最多只需3次旋转，只需要O(1)的复杂度。

下图是知乎大神提供的性能测试：

由此可见：

①红黑树由于插入删除时较少的旋转操作，对于需要频繁进行插入删除的场景性能比AVL树好
②反之，对于不需要频繁插入删除的场景。由于AVL树的高度平衡，使得同等结点数量下，AVL树的高度比红黑树更低，使得AVL树的查找性能比红黑树好。

JDK1.8里的TreeMapEntry对红黑树数据结构的定义

 /**
     * Node in the Tree.  Doubles as a means to pass key-value pairs back to
     * user (see Map.Entry).
     */
 
    static final class TreeMapEntry<K,V> implements Map.Entry<K,V> {
        K key;
        V value;
        TreeMapEntry<K,V> left = null;//左子树
        TreeMapEntry<K,V> right = null;//右子树
        TreeMapEntry<K,V> parent;//父节点
        boolean color = BLACK;//默认为黑色
 
        /**
         * Make a new cell with given key, value, and parent, and with
         * {@code null} child links, and BLACK color.
         */
        TreeMapEntry(K key, V value, TreeMapEntry<K,V> parent) {
            this.key = key;
            this.value = value;
            this.parent = parent;
        }
 
        /**
         * Returns the key.
         *
         * @return the key
         */
        public K getKey() {
            return key;
        }
 
        /**
         * Returns the value associated with the key.
         *
         * @return the value associated with the key
         */
        public V getValue() {
            return value;
        }
        /*...各种方法*/

Linux内核文件rbtree.h中对红黑树的数据结构的定义

struct rb_node
 {/*long是四个字节，占用共4x8=32bit，其中有两位存储父结点指针+自己的颜色值*/
     unsigned long  rb_parent_color; /*注意：此处保存的是父指针+自己颜色*/
 #define    RB_RED      0
 #define    RB_BLACK    1
     struct rb_node *rb_right;
     struct rb_node *rb_left;
 } /*  __attribute__((aligned(sizeof(long))))*/;
     /* The alignment might seem pointless, but allegedly CRIS needs it */
 
 struct rb_root
 {
     struct rb_node *rb_node;
 };

以及rbtree.h中在插入新结点时，初始化红黑树结点操作

static inline void rb_init_node(struct rb_node *rb)
 {
     rb->rb_parent_color = 0;
     rb->rb_right = NULL;
     rb->rb_left = NULL;
     RB_CLEAR_NODE(rb);
 }

在此需要先了解左旋和右旋的原理

结点E左旋操作：

(动图来源于网络)
代码如下：

/*
 * 左旋操作Linux c代码(来源于Linux内核rbtree.c)
*/
static void __rb_rotate_left(struct rb_node *node, struct rb_root *root)
{
    struct rb_node *right = node->rb_right;
    struct rb_node *parent = rb_parent(node);

    if ((node->rb_right = right->rb_left))  //node的右指针指向node右孩子的左孩子
        rb_set_parent(right->rb_left, node);//node右孩子的左孩子的父亲指定为node
    right->rb_left = node;

    rb_set_parent(right, parent); //node右孩子祖父指定为node原父亲

    if (parent) //node原父亲存在，即原node不是根结点
    {
        if (node == parent->rb_left)
            parent->rb_left = right;
        else
            parent->rb_right = right;
    }
    else //原node是根结点
        root->rb_node = right;
    rb_set_parent(node, right); //node现任父亲指定为node原右孩子
}

结点S右旋操作：

(动图来源于网络，代码与左旋对调)

红黑树插入操作逻辑

百度脑图原图看更清晰

红黑树删除操作逻辑