红黑树是60年代中期计算机科学界找寻一种算法复杂度稳定,容易实现的数据存储算法的产物。在优先级队列、字典等实用领域都有广泛地应用,更是70年代提出的关系数据库模型--B树的鼻祖。Linux kernel中,高精度定时器也工作在红黑树之上为便于初学者掌握其基本算法,本文一步一步地演示了红黑树的创建过程。

 

    首先回顾一下红黑树的基本性质:

    1. 红黑树本质上是一个二叉查找树(BST),但是它从根到最远叶子的长度不会超过到最近叶子长度的两倍,因此是近似平衡的。

    2. 红黑树的节点不是黑的就是红的,不会有第三种颜色。

    3. 树根必须是黑色。

    4. 叶子所指的空节点必须是黑色。

    5. 如果某个节点是红色,那么它的两个儿子必须都是黑色。

    6. 从任意节点出发的所有向下的路径上包含相同个数的黑节点。这个个数我们称为黑高度Bh。

 

    有了上面的认识,我们要从无到有构造一颗红黑树的话,心里就有底了。这个构造的过程可以分两步:

    第一步:执行BST的插入算法;

    第二步:对节点着色;

 

    第一步不会有什么问题,关键是第二步怎么对节点着色才不会违反红黑树的基本性质;

 

    这是一个难点,我在写这篇文章的时候脑子也卡了一下,节点不多的时候好办,但是如果节点很多了,就必须找到一种普遍适用的算法来处理。通过这两天的观察,我发现这六条性质中最关键的其实是最后一条---从任意结点出发的任意路径黑高度相等!这才是红黑树算法保持近似平衡的精华所在!其它性质要么是这条性质的产物,要么就是为这条性质服务的。

 

    现在我演示一下怎么从无到有生成一棵红黑树。

例如我们有一组数:{9,7,15,6,11,19},现在按照从左到右的顺序存放在一颗红黑树中。

 

第一个数是9,很自然地成为了树根,如图:

                                          图-1

    每个新节点都默认地被渲染成了红色,为什么要这么做呢?后面我们将会看到它的好处!现在先忽略不谈。

 

    根节点9是红色,这违背了性质3,所以必须改成黑色,如图:

                                          图-2

 

下一个数字是7,显然要被插入到9的左边,并且这时满足红黑树的所有性质,如图:

                                   图-3

 

下一个数字是15,要插入到9的右边,并且也满足红黑树的所有性质,如图:

                                   

                                                 图-4

 

下一个数字是6,要插入到7的左边,这回似乎有麻烦了,性质5被违背了,如图:

                             

                                    图-5

 

可能有人会说,那很简单,既然违背了性质5,那我改一下6的颜色不就OK了?

                            

                                    图-6

    现在,恭喜你---也走进了我曾经走过的误区:)你的无意之举改变了新增节点路径上面的黑高度,这棵树已经不是红黑树了

 

    写代码的人都知道,对树的遍历,最简单的方法就是递归,那么树的数学模型就是一个可穷举的递归式。递归式中每一步的正确性都建立在前一步正确的基础之上。 现在我们回过头来想想为什么每次插入的新节点都被渲染成红色呢?你肯定猜到了,因为我们要保证红黑树的核心性质--黑高度不发生变化,虽然牺牲了性质5, 但这是可以补救的,并且代价很小。再来看看图-5

                            

    既然我们不能通过改变新插入节点的颜色来维持红黑树的性质,那么就只好从原来的树结构入手了。

 

    我们注意到新插入的节点的父亲是一个红节点,其叔叔也是一个红节点。那么假如我把它的父亲改成黑色,则恢复了性质5,可是从树根出发的黑高度还是不相等,干脆把它的叔叔也改成黑色吧!结果如下:

                            

                                   图-7

 

现在它满足红黑树所有的性质。好了,我们继续。

插入数字11和19的过程平淡无奇,不深入探讨,最终的树如图:

                           

                                   图-8

 

如果现在要插入数字10怎么办?它肯定会是11的左节点:

                         

                                              图-9

明显违背了性质5!难道依葫芦画瓢,把11和19都改成黑色?这样从树根出发的左边路径黑高度还是2,而右边两条路径的黑高度将变成了3,老办法失效了!

                         

                                  图-10

    其实不然,改了新增节点的父亲和叔叔的颜色以后,右边路径黑高度全加一,但我们只要把它的爷爷改成红色不就又减去了多出来的黑高度吗?

                          

                                  图-11

如果你认真看到这里,我想你的潜意识一定告诉你---这里存在某种规律等着我们去发现。

 

让我们再仔细审视一下图-5

                           

    在生成树的过程中,我们已经两次遇到类似情况了,归纳一下,这个场景就是:新节点的父亲是红色,叔叔也是红色。

 

    至于别的节点,我们大可不必关心,因为很显然,这个场景是原子的。

 

    我们的处理办法是把7和15渲染成黑色,就像图-7那样,可是因为这是一个普遍适用的场景,所以要做一个扩展:假设在这个原子树的每个节点上还有别的分 支。那么图-7就不能保证所有路径的黑高度相等了,即使把9改成红色也无济于事,因为9也许还有父节点,所以也许它的父节点又是红色,这就再次违反了性质 5。我最喜欢的作家之一--柯南.道尔,曾说过:“历史就像车轮的辐条,每一根都终将再次转回来。”眼前的场景正是如此---我们设计的原子场景再次出 现。

 

一个清晰的递归算法呈现在我们面前:

 

1. 新增节点渲染成红色;

2. 如果它的父亲是红色,则违反了性质5;

3. 如果它的叔叔也是红色,则通过同时修改其父亲和叔叔的颜色为黑色来恢复性质5;

4. 如果它的爷爷不是根节点,则有可能在另外的路径上再次违反性质5,于是我们把它的爷爷改成红色;

5. 可是如果他的太爷爷也是红色呢?很自然地,我们重新回到步骤2;

6. 不断循环,直到第二步满足就可以结束。

 

最后留给感兴趣的同学两个问题:

1. 如果新增节点的父亲是红色,但它的叔叔是黑色,该怎么办?(提示:使用旋转)

 

2. 有人说,下面这种场景,我设计的算法就失效了,真的吗?(粉色的6是新插入的节点)