數據結構-紅黑樹解析

小明:老師,紅黑樹是什麼樹呀?

老師:紅黑樹是數據結構的一種,在JDK的TreeMap,TreeSet中都用應用,JDK8的HashMap中,當鏈表樹超過8的時候鏈表會轉換為紅黑樹可以提高查詢效率。

小明:好深奧哦,可以詳細講講嗎?

老師:好吧,要學習紅黑樹,咱們需要先來理解二叉查找樹(Binary Search Tree)。

二叉查找樹(BST)具備什麼特性呢?

1.左子樹上所有結點的值均小於或等於它的根結點的值。

2.右子樹上所有結點的值均大於或等於它的根結點的值。

3.左、右子樹也分別為二叉排序樹。

下圖中這棵樹,就是一顆典型的二叉查找樹:

數據結構-紅黑樹解析

這樣的數據結構有什麼好處呢?我們來試著查找一下值為10的節點。

1.查看根節點9:

數據結構-紅黑樹解析

2.由於10 > 9,因此查看右孩子13:

數據結構-紅黑樹解析

3.由於10 < 13,因此查看左孩子11:

數據結構-紅黑樹解析

4.由於10 < 11,因此查看左孩子10,發現10正是要查找的節點:

數據結構-紅黑樹解析

這種方式正式二分查找的思想,查找所需的最大次數等同於二叉樹的高度。在插入節點的時候也是利用類似的方法,通過一層一層比較大小,找到新節點適合插入的位置。

小明:好方便,二叉查找樹真是個強大的數據結構!

老師:很遺憾,二叉查找樹仍然存在它的缺陷。

小明:哦,到底存在什麼缺陷呢?

老師:缺陷體現在插入新節點的時候。讓我們來看看下面這種情形:

假設初始的二叉查找樹只有三個節點,根節點值為9,左孩子值為8,右孩子值為12:

數據結構-紅黑樹解析

接下來我們依次插入如下五個節點:7,6,5,4,3。依照二叉查找樹的特性,結果會變成什麼樣呢?

數據結構-紅黑樹解析

小明:媽呀,好好的二叉查找樹變成瘸子了

老師:正式如此。這樣的形態雖然也符合二叉查找樹的特性,但是查找的性能大打折扣,幾乎變成了線性。如何解決二叉查找樹多次插入新節點而導致的不平衡呢?我們的主角【紅黑樹】應運而生了。紅黑樹(Red Black Tree)是一種自平衡的二叉查找樹。除了符合二叉查找樹的基本特性外,它還具有下列的附加特性:

1.節點是紅色或黑色。

2.根節點是黑色。

3.每個葉子節點都是黑色的空節點(NIL節點)。

4 每個紅色節點的兩個子節點都是黑色。(從每個葉子到根的所有路徑上不能有兩個連續的紅色節點)

5.從任一節點到其每個葉子的所有路徑都包含相同數目的黑色節點。

下圖中這棵樹,就是一顆典型的紅黑樹:

數據結構-紅黑樹解析

小明:這麼多條條框框呀

老師:正是因為這些規則現實,才保證了紅黑樹的自平衡。紅黑樹從根到葉子的最長路徑不會超過最短路徑的2倍。當插入或刪除節點的時候,紅黑樹的規則有可能被打破。這時候就需要做出一些調整,來繼續維持我們的規則。

什麼情況下會破壞紅黑樹的規則,什麼情況下不會破壞規則呢?我們舉兩個簡單的例子:

1.向原紅黑樹插入值為14的新節點:

數據結構-紅黑樹解析

由於父節點15是黑色節點,因此這種情況並不會破壞紅黑樹的規則,無需做任何調整。

2.向原紅黑樹插入值為21的新節點:

數據結構-紅黑樹解析

由於父節點22是紅色節點,因此這種情況打破了紅黑樹的規則4(每個紅色節點的兩個子節點都是黑色),必須進行調整,使之重新符合紅黑樹的規則。

小明:那麼,我們需要做出怎樣的調整,才能保證一顆紅黑樹始終是紅黑樹呢?

老師:調整有兩種方法:【變色】和【旋轉】。而旋轉又分成兩種形式:【左旋轉】和【右旋轉】。

變色:

為了重新符合紅黑樹的規則,嘗試把紅色節點變為黑色,或者把黑色節點變為紅色。

下圖所表示的是紅黑樹的一部分,需要注意節點25並非根節點。因為節點21和節點22連續出現了紅色,不符合規則4,所以把節點22從紅色變成黑色:

數據結構-紅黑樹解析

但這樣並不算完,因為憑空多出的黑色節點打破了規則5,所以發生連鎖反應,需要繼續把節點25從黑色變成紅色:

數據結構-紅黑樹解析

此時仍然沒有結束,因為節點25和節點27又形成了兩個連續的紅色節點,需要繼續把節點27從紅色變成黑色:

數據結構-紅黑樹解析

左旋轉:

逆時針旋轉紅黑樹的兩個節點,使得父節點被自己的右孩子取代,而自己成為自己的左孩子。說起來很怪異,大家看下圖:

數據結構-紅黑樹解析

圖中,身為右孩子的Y取代了X的位置,而X變成了自己的左孩子。此為左旋轉。

右旋轉:

順時針旋轉紅黑樹的兩個節點,使得父節點被自己的左孩子取代,而自己成為自己的右孩子。大家看下圖:

數據結構-紅黑樹解析

圖中,身為左孩子的Y取代了X的位置,而X變成了自己的右孩子。此為右旋轉。

小明:好複雜,究竟什麼時候用到變色,什麼時候用到旋轉呢?

老師:確實有些複雜。紅黑樹的插入和刪除包含很多種情況,每一種情況都有不同的處理方式。在這裡我們舉一個典型例子。我們以剛才插入節點21的情況為例:

數據結構-紅黑樹解析

首先,我們需要做的是變色,把節點25及其下方的節點變色:

數據結構-紅黑樹解析

此時節點17和節點25是連續的兩個紅色節點,那麼把節點17變成黑色節點?恐怕不合適。這樣一來不但打破了規則4,而且根據規則2(根節點是黑色),也不可能把節點13變成紅色節點。

變色已無法解決問題,我們把節點13看做X,把節點17看做Y,像剛才的示意圖那樣進行左旋轉

數據結構-紅黑樹解析

數據結構-紅黑樹解析

數據結構-紅黑樹解析

由於根節點必須是黑色節點,所以需要變色,變色結果如下:

數據結構-紅黑樹解析

這樣就結束了嗎?並沒有。因為其中兩條路徑(17 -> 8 -> 6 -> NIL)的黑色節點個數是4,其他路徑的黑色節點個數是3,不符合規則5。

這時候我們需要把節點13看做X,節點8看做Y,像剛才的示意圖那樣進行右旋轉

數據結構-紅黑樹解析

數據結構-紅黑樹解析

數據結構-紅黑樹解析

最後根據規則來進行變色:

數據結構-紅黑樹解析

如此一來,我們的紅黑樹變得重新符合規則。這一個例子的調整過程比較複雜,經歷瞭如下步驟:

變色 -> 左旋轉 -> 變色 -> 右旋轉 -> 變色

小明:還真是繞呀,我慢慢消化吧。

老師:嗯,關於紅黑樹自平衡的調整,插入和刪除節點的時候都涉及到很多種Case,由於篇幅原因沒有一一列舉,有興趣的朋友可以參考維基百科,裡面講的非常清晰。


分享到:


相關文章: