前言
前段時間在研究 JDK1.8 的 hashmap 源碼，看到 put 方法的插入環(huán)節(jié)，遇到了紅黑樹，不得不停止閱讀源碼的過程，因?yàn)檫€沒掌握紅黑樹是無法完全讀透 hashmap 源碼的。紅黑樹作為一種數(shù)據(jù)結(jié)構(gòu)，它被應(yīng)用得非常多，可能很多人不認(rèn)識它，但其實(shí)它已經(jīng)在默默為我們的代碼在發(fā)光發(fā)熱。例如，你只要在 Java 中用到 map，基本上就是在用紅黑樹（當(dāng)元素個數(shù)到達(dá)八個時鏈表轉(zhuǎn)紅黑樹）。
PS：在看這篇文章前，必須先了解普通的二叉查找樹和平衡查找樹（AVL）樹、2-3-4樹。不然看起來會非常吃力。
歡迎加入交流群一起討論：611481448
紅黑樹的性質(zhì)
紅黑樹是一種自平衡樹，它也是一顆二叉樹。既然能保持平衡，說明它和 AVL 樹類似，在插入或者刪除時肯定有調(diào)整的過程，只不過這個調(diào)整過程并不像 AVL 樹那樣繁瑣。為何紅黑樹使用得比 AVL 樹更多，就是因?yàn)榧t黑樹它的調(diào)整過程迅速且簡介。紅黑樹有以下五個特性：
性質(zhì)1：節(jié)點(diǎn)是紅色或黑色
性質(zhì)2：根是黑色
性質(zhì)3：所有葉子都是黑色。葉子是 NIL 節(jié)點(diǎn)，也就是 Null 節(jié)點(diǎn)
性質(zhì)4：如果一個節(jié)點(diǎn)是紅的，則它的兩個兒子都是黑的
性質(zhì)5：從任一節(jié)點(diǎn)到其葉子的所有簡單路徑都包含相同數(shù)目的黑色節(jié)點(diǎn)。
下圖展示了一棵紅黑樹。
分析：
注意理解上，葉子并不等價于黑色節(jié)點(diǎn)，但是他們的顏色都是黑色。
為何要給節(jié)點(diǎn)指定紅或者黑的顏色？
作者這種設(shè)計，只是為了從編程上達(dá)到一種便利的效果。另外可以讓它們在插入時達(dá)到近似的平衡，并不像 AVL 樹那樣絕對平衡。實(shí)際上，紅黑樹是2-3樹的一種變體，某種情況下，它又相當(dāng)于2-3-4樹。因?yàn)?-3樹在編程上需要比較多的代碼量，所以誕生了紅黑樹這種巧妙的設(shè)計。通過加了顏色來區(qū)分結(jié)點(diǎn)，這樣編程上就可以當(dāng)成二叉樹來寫程序，不用分別用三個指針表示左、中、右孩子了。
看一下紅黑樹和2-3樹的等價性聯(lián)系
從上面可以看到，把紅色節(jié)點(diǎn)放到與父親齊平，就是2-3樹中的一個2-3節(jié)點(diǎn)。
紅黑樹的操作
數(shù)據(jù)結(jié)構(gòu)的性質(zhì)看完之后，就要掌握它到底會存在哪種操作？例如 hashmap，它最常見的操作就是 get 和 put、擴(kuò)容。同理，紅黑樹也有它的基本操作。因?yàn)樗旧砩弦彩且豢枚娌檎覙洌灾攸c(diǎn)關(guān)注的操作無非就是查找、插入、刪除。
1. 查找操作
紅黑樹的查找方式很簡單，只要是樹，查找的過程無非就是一個遞歸過程。如果查找的元素小于當(dāng)前節(jié)點(diǎn)，那么查找其左子樹；如果查找的元素大于當(dāng)前元素，則查找其右子樹。
2. 插入操作
插入操作首先需要通過查找操作找到合適的插入點(diǎn)，然后插入新節(jié)點(diǎn)。如果在插入節(jié)點(diǎn)后，發(fā)生了違背紅黑樹特性的情況時，需要對紅黑樹進(jìn)行旋轉(zhuǎn)染色等操作，使其重新滿足特性。
2.1 插入新節(jié)點(diǎn)
為了在插入新節(jié)點(diǎn)時盡可能少的違反紅黑樹特性且更容易調(diào)整紅黑樹，就先將新節(jié)點(diǎn)染成紅色。這樣就只可能會違反特性4。如果這里沒有違反特性4，那么就不需要對紅黑樹進(jìn)行調(diào)整，插入操作完成。
2.2 調(diào)整子樹
那么，在違反了特性4的時候，新節(jié)點(diǎn)的父節(jié)點(diǎn)為紅色節(jié)點(diǎn)。根據(jù)特性2可知，父節(jié)點(diǎn)不是根節(jié)點(diǎn)，則新節(jié)點(diǎn)必有祖父節(jié)點(diǎn)。又根據(jù)特性3可推論出紅色節(jié)點(diǎn)必有兩個黑色子節(jié)點(diǎn)（空節(jié)點(diǎn)為黑色）。此時會出現(xiàn)兩種情況：叔節(jié)點(diǎn)為紅色、叔節(jié)點(diǎn)為黑色。
圖例：C 表示當(dāng)前節(jié)點(diǎn)，P 表示父節(jié)點(diǎn)，U 表示叔節(jié)點(diǎn)，G 表示祖父節(jié)點(diǎn)
（1）父節(jié)點(diǎn)與叔節(jié)點(diǎn)都為紅色的情況
在這種情況下，需要將父節(jié)點(diǎn)和叔節(jié)點(diǎn)變?yōu)楹谏賹⒆娓腹?jié)點(diǎn)變?yōu)榧t色。這樣，圖上所展示的子樹就滿足了紅黑樹的特性。如下圖所示。
但是這里又可能會產(chǎn)生新的違反特性情況，因?yàn)樽娓腹?jié)點(diǎn)變成了紅色，那么它可能會造成違反特性4的情況。所以，這里就將祖父節(jié)點(diǎn)作為當(dāng)前節(jié)點(diǎn)，進(jìn)行新一輪的調(diào)整操作。
（2）父節(jié)點(diǎn)為紅色，叔節(jié)點(diǎn)為黑色的情況
在這種情況下，對其調(diào)整的核心就是保持父節(jié)點(diǎn)分支符合特性4，而叔節(jié)點(diǎn)分支保持符合特性5。
第一步，旋轉(zhuǎn)。對祖父節(jié)點(diǎn)進(jìn)行左旋或者右旋。如果父節(jié)點(diǎn)是祖父節(jié)點(diǎn)的右子節(jié)點(diǎn)，那么對祖父節(jié)點(diǎn)進(jìn)行左旋；否則，對祖父節(jié)點(diǎn)進(jìn)行右旋。
第二步，染色。將祖父節(jié)點(diǎn)染為紅色，而父節(jié)點(diǎn)染為黑色。
進(jìn)過這兩步，上圖的情況會轉(zhuǎn)換為下圖所示。
可以看出，父節(jié)點(diǎn)這一分支進(jìn)過調(diào)整后，當(dāng)前節(jié)點(diǎn)與父節(jié)點(diǎn)的顏色不再是連續(xù)紅色，滿足特性4。而叔節(jié)點(diǎn)這一分支的黑色節(jié)點(diǎn)數(shù)目沒有發(fā)生變化，滿足特性5。對原祖父節(jié)點(diǎn)的父節(jié)點(diǎn)來說，該子樹沒有發(fā)生違反特性的變化。該子樹調(diào)整完成。
2.3 檢查根節(jié)點(diǎn)
當(dāng)上述調(diào)整執(zhí)行完后，還有最后一步，就是檢查是否滿足特性2。這一步只需要將根節(jié)點(diǎn)染成黑色就可以，無需再多加判斷。
3. 刪除操作
需要重點(diǎn)理解的就是刪除操作，這個也是我覺得是紅黑樹中最難的部分。
刪除操作要比插入操作略微復(fù)雜一些。因?yàn)閯h除的節(jié)點(diǎn)可能是出現(xiàn)在樹的中間層的節(jié)點(diǎn)，此時刪除該節(jié)點(diǎn)會遇到很復(fù)雜的情況。所以，在刪除節(jié)點(diǎn)的時候，需要先對紅黑樹進(jìn)行一些調(diào)整，使得刪除節(jié)點(diǎn)對整個樹的影響降到最低。
3.1 替換刪除節(jié)點(diǎn)
首先根據(jù) BST 刪除節(jié)點(diǎn)的規(guī)則，使用當(dāng)前節(jié)點(diǎn)左子樹的最大值節(jié)點(diǎn)或者右子樹的最小值節(jié)點(diǎn)代替其刪除。這兩個節(jié)點(diǎn)是其子樹中數(shù)值上最貼近當(dāng)前節(jié)點(diǎn)數(shù)值的節(jié)點(diǎn)）。 這一點(diǎn)，只要懂了二叉查找樹的刪除操作就明白了，在這里不多說了。如下圖所示：
圖例：D 表示當(dāng)前節(jié)點(diǎn)，P 表示父節(jié)點(diǎn)，B 表示兄弟節(jié)點(diǎn)，BR 表示兄弟節(jié)點(diǎn)的右子節(jié)點(diǎn)，BL 表示兄弟節(jié)點(diǎn)的左子節(jié)點(diǎn)
既然待刪除節(jié)點(diǎn)是要被移走的，那肯定有一個節(jié)點(diǎn)要替換到它的位置上去。如何找到這個替換節(jié)點(diǎn)，這個過程和二叉查找樹一模一樣，要么在它的左子樹下一直往右找到最大節(jié)點(diǎn)，要么在右子樹下找到最小節(jié)點(diǎn)。
下面的描述過程采用的是右子樹的最小值節(jié)點(diǎn)代替
當(dāng)找到替換節(jié)點(diǎn)之后，現(xiàn)在需要考慮的情況就減少了，只可能會出現(xiàn)以下幾種情況（因?yàn)樾枰獫M足紅黑樹特性）：
無子節(jié)點(diǎn)，節(jié)點(diǎn)為紅色
無子節(jié)點(diǎn)，節(jié)點(diǎn)為黑色
只有右子節(jié)點(diǎn)，右子節(jié)點(diǎn)為紅色，節(jié)點(diǎn)本身為黑色
上面這三種情況，說的是新待刪除節(jié)點(diǎn)。新待刪除節(jié)點(diǎn)，就是即將被替換到待刪除位置的節(jié)點(diǎn)。
因?yàn)?D 節(jié)點(diǎn)就是即將要替換到待刪節(jié)點(diǎn)位置的節(jié)點(diǎn)，它同時又是右子樹的最小值，既然是最小值了，它就不再可能擁有左子樹了，所以只有可能有右子節(jié)點(diǎn)。另外，假如它有右節(jié)點(diǎn)且右節(jié)點(diǎn)的顏色是黑色，它自身顏色是紅色，根本不成立。因?yàn)榧偃缢陨頌榧t色且又有黑孩子，那它必須要有兩個黑孩子才滿足紅黑樹性質(zhì)，所以不滿足。 那有沒有可能，它自身是黑色且右孩子也為黑色呢？也不可能！因?yàn)樗蠛⒆右呀?jīng)為空了，說明它從自身出發(fā)到左子樹的葉子的距離就是1，假如它右孩子也為黑色，那它從自身出發(fā)到右子樹葉子的距離肯定大于等于2了，明顯不可能。
所以總的來說只可能有下面三種情況：
情況1：只需要直接刪除節(jié)點(diǎn)就可以。刪了一個紅色新待刪節(jié)點(diǎn)，不會影響紅黑樹性質(zhì)。
情況2：刪除該 D 節(jié)點(diǎn)后，違反了紅黑樹特性5，需要調(diào)整（不考慮待刪除節(jié)點(diǎn)為根節(jié)點(diǎn)的情況）
情況3：用右子節(jié)點(diǎn) R 占據(jù)待刪除節(jié)點(diǎn) D，再將其染成黑色即可，不違反紅黑樹特性。因?yàn)樽筮叡緛砭褪强樟耍鋵?shí)右子樹下即使有多少個黑色節(jié)點(diǎn)，也不會影響整體特性。
在這三種情況中，情況1和情況3比較簡單，不需要多余的調(diào)整。情況2則需要后續(xù)的調(diào)整步驟使其滿足紅黑樹特性。
3.2 調(diào)整紅黑樹
上述情況2的調(diào)整比較復(fù)雜。下面對各種情況進(jìn)行講解。
根據(jù)紅黑樹的特性5，待刪除節(jié)點(diǎn)必然有兄弟節(jié)點(diǎn)。
為什么這么說呢？因?yàn)槲覀円呀?jīng)假設(shè)上面的 D 節(jié)點(diǎn)不為根了，那說明它肯定有父親。首先它是沒有孩子的，它下面直接就是葉子了，既然有父親，不論它是父親的左孩子或者右孩子，從父親出發(fā)到它自身，黑色節(jié)點(diǎn)的個數(shù)為1。反證法：假如父親只有它一個孩子，那說明父親到另一邊子樹的葉子距離就為0，因?yàn)?個節(jié)點(diǎn)。這明顯不符合，所以說明父親肯定有兩個孩子，那從而得知待刪節(jié)點(diǎn)D必有兄弟。
下面根據(jù)其兄弟節(jié)點(diǎn)所在分支的不同，來分情況討論。
以下是以關(guān)注待刪節(jié)點(diǎn)為父節(jié)點(diǎn)的左子節(jié)點(diǎn)進(jìn)行描述，如果遇到關(guān)注節(jié)點(diǎn)為父節(jié)點(diǎn)的右子節(jié)點(diǎn)的情況，則鏡像處理。
思路：下面的任何調(diào)整只有一個目的，就是不斷調(diào)整，直到調(diào)整到可以直接將 D 移除又不會影響紅黑樹特性的情況。但關(guān)鍵是調(diào)整過程中紅黑樹特性也不會發(fā)生改變。
圖例：D 表示當(dāng)前節(jié)點(diǎn)，P 表示父節(jié)點(diǎn)，B 表示兄弟節(jié)點(diǎn)，BR 表示兄弟節(jié)點(diǎn)的右子節(jié)點(diǎn)，BL 表示兄弟節(jié)點(diǎn)的左子節(jié)點(diǎn)
（1）兄弟節(jié)點(diǎn)為紅色
將父節(jié)點(diǎn)染成紅色，兄弟節(jié)點(diǎn)染成黑色，然后對父節(jié)點(diǎn)進(jìn)行左旋操作。此時就轉(zhuǎn)換為了下面的（4），之后按照（4）繼續(xù)進(jìn)行調(diào)整。
分析：這種情況，樹的整體高度為2，變色左旋之后，整體高度還是保持在2。
（2）兄弟節(jié)點(diǎn)為黑色，遠(yuǎn)侄節(jié)點(diǎn)為紅色
這種情況下，不需要考慮父節(jié)點(diǎn)的顏色。
將父節(jié)點(diǎn) P 與兄弟節(jié)點(diǎn) B 的顏色互換 ，這個過程父親染黑
將兄弟節(jié)點(diǎn)的右子節(jié)點(diǎn) BR 染成黑色
對父節(jié)點(diǎn) P 進(jìn)行左旋操作
可以看到，原本高度就是符合紅黑樹特性的，左右子樹的高度都為1，因?yàn)楹谏?jié)點(diǎn)只有一個。經(jīng)過這三步的調(diào)整后，直接刪除節(jié)點(diǎn) D 后仍然滿足紅黑樹的特性，調(diào)整完成，跳出算法循環(huán)。
（3）兄弟節(jié)點(diǎn)為黑色，遠(yuǎn)侄節(jié)點(diǎn)為黑色，近侄節(jié)點(diǎn)為紅色
這種情況下，兄弟節(jié)點(diǎn)的左節(jié)點(diǎn)染成黑色。兄弟節(jié)點(diǎn)染紅。然后對兄弟節(jié)點(diǎn)做右旋。此時的狀況就和（2）一樣了。之后就通過（2）的調(diào)整方式進(jìn)行調(diào)整。
（4）父節(jié)點(diǎn)為紅色，兄弟節(jié)點(diǎn)為黑色，兄弟節(jié)點(diǎn)無子節(jié)點(diǎn)
這種情況下，將父節(jié)點(diǎn)P染成黑色，再將兄弟節(jié)點(diǎn)染成紅色。經(jīng)過這樣的操作后，除去節(jié)點(diǎn)D后，以P為根節(jié)點(diǎn)的子樹的黑節(jié)點(diǎn)深度并沒有發(fā)生變化。調(diào)整完成。
怎么理解這個操作？
可以看左邊，沒調(diào)整前，P 的左右子樹的黑色結(jié)點(diǎn)的數(shù)目都是1，是相同的，符合紅黑樹的性質(zhì)：從任一節(jié)點(diǎn)到其葉子的所有簡單路徑都包含相同數(shù)目的黑色節(jié)點(diǎn)。然后再看右邊，調(diào)整后，刪掉 D 之后，P 結(jié)點(diǎn)的左右子樹的黑色結(jié)點(diǎn)都是0個，仍然滿足性質(zhì)，所以調(diào)整完成。
（5）父節(jié)點(diǎn)為黑色，兄弟節(jié)點(diǎn)為黑色，兄弟節(jié)點(diǎn)無子節(jié)點(diǎn)
這種情況下，為了在刪除節(jié)點(diǎn) D 后使以 P 為根節(jié)點(diǎn)的子樹能滿足紅黑樹特性5，將兄弟節(jié)點(diǎn) B 染成紅色。但是這樣操作后，以 P 為根節(jié)點(diǎn)的子樹的黑色節(jié)點(diǎn)深度變小了。所以需要繼續(xù)調(diào)整。
因?yàn)镻節(jié)點(diǎn)子樹的黑色深度發(fā)生了減少，可以把其當(dāng)作待刪除節(jié)點(diǎn)，那么此時就以 P 節(jié)點(diǎn)為關(guān)注節(jié)點(diǎn)進(jìn)行進(jìn)一步調(diào)整（繼續(xù)向上調(diào)整）。 這句話的意思我們再以 P 為起始點(diǎn)，繼續(xù)根據(jù)情況進(jìn)行平衡操作。就是把 P 當(dāng)成 D，只是不要再刪除 P 了。再看是這五種中的哪種情況，再進(jìn)行對應(yīng)的調(diào)整，這樣一直向上，直到新的起始點(diǎn)為根節(jié)點(diǎn)或者關(guān)注節(jié)點(diǎn)不為黑色。
第五種情況，不會一直連續(xù)回溯的。假如能一直回溯，指針向上走之后，兄弟節(jié)點(diǎn)會一直都沒有右孩子嗎？不存在的。假如有這種情況，說明樹的路徑長度已經(jīng)嚴(yán)重往左傾斜，肯定不可能。所以回溯這個情況只會回溯一次，不會連續(xù)回溯。第五個這種情況出現(xiàn)之后，下一次進(jìn)入算法循環(huán)，肯定就是進(jìn)入其他情況，直到遇到 break，跳出循環(huán)，終止整個算法過程。
3.3 檢查根節(jié)點(diǎn)及刪除節(jié)點(diǎn)
經(jīng)過上述的調(diào)整后，此時基本滿足了紅黑樹的特性。但是存在根節(jié)點(diǎn)變成紅色的情況。所以需要將根節(jié)點(diǎn)染成黑色的操作。 最后，執(zhí)行刪除操作，將待刪除節(jié)點(diǎn)刪掉。
當(dāng)然從編程的角度，你也可以調(diào)整指針先把待刪除節(jié)點(diǎn)移掉，然后再開始平衡調(diào)整過程。注意這里說的平衡調(diào)整，并不是 AVL 樹的絕對平衡調(diào)整，而是滿足紅黑樹特性的平衡調(diào)整。紅黑樹的平衡和 AVL 的平衡是有區(qū)別的。
總結(jié)
紅黑樹的刪除操作是整個紅黑樹中最復(fù)雜的一部分，理解了這部分，紅黑樹就算基本拿下了。理解完一種數(shù)據(jù)結(jié)構(gòu)，要能 get 到作者當(dāng)初設(shè)計時的點(diǎn)，才算是一次積累。紅黑樹的刪除操作，它非常地巧妙，整一個算法循環(huán)過程，它不會超過三次，調(diào)整過程基本都在子樹內(nèi)完成，指針不需要一直向上回溯，相比 AVL 樹，AVL 樹在刪除節(jié)點(diǎn)時，指針有可能會一直回溯到根為止。
”我自己是一名從事了十余年的后端的老程序員，辭職后目前在做講師，近期我花了一個月整理了一份最適合2018年學(xué)習(xí)的JAVA干貨（里面有高可用、高并發(fā)、高性能及分布式、Jvm性能調(diào)優(yōu)、Spring源碼，MyBatis，Netty,Redis,Kafka,Mysql,Zookeeper,Tomcat,Docker,Dubbo,Nginx等多個知識點(diǎn)的架構(gòu)資料）從事后端的小伙伴們都可以來了解一下的，這里是程序員秘密聚集地，各位還在架構(gòu)師的道路上掙扎的小伙伴們速來。“
加QQ群：611481448（名額有限哦！）
