逐行解读HashMap源码之红黑树篇

原创
开发 后端
在上一篇文章《逐行解读HashMap源码》中,笔者主要分析了HashMap类的成员变量与成员方法,未涉及对HashMap中红黑树相关内容的分析。于是,笔者经过一段的时间的编写与打磨,写出了本文。本文主要内容包括对上一篇文章的些许补充以及对红黑树源码的详细解读。

【51CTO.com原创稿件】

写在前面

本文是之前发表的文章《逐行解读HashMap源码》的下一篇,也是最后一篇。在《逐行解读HashMap源码》文章中,笔者主要分析了 HashMap 类的成员变量与成员方法,未涉及对 HashMap 中红黑树相关内容的分析。于是,笔者经过一段的时间的编写与打磨,写出了本文。本文主要内容包括对上一篇文章的些许补充以及对红黑树源码的详细解读。

如果读者没有阅读过上一篇内容,可以微信搜索公众号“代码艺术”搜索并阅读文章。

1. 红黑树概述

从本章节开始,笔者将对 HashMap 内部的 TreeNode 内部类源码开始分析,这部分的内容也就是我们所说的红黑树。

红黑树是建立在二叉查找树的基础之上的,二叉查找树又是建立在二叉树的基础之上。所以,我们需要从二叉树开始学习红黑树的发展脉络。

2. 二叉树

二叉树(英语:Binary tree)是每个节点最多只有两个分支(即不存在分支度大于2的节点)的树结构。通常分支被称作“左子树”或“右子树”。二叉树的分支具有左右次序,不能随意颠倒。

3. 二叉查找树

二叉查找树(英语:Binary Search Tree),也称为二叉搜索树、有序二叉树(ordered binary tree)或排序二叉树(sorted binary tree),是指一棵空树或者具有下列性质的二叉树:

  1. 若任意节点的左子树不空,则左子树上所有节点的值均小于它的根节点的值;
  2. 若任意节点的右子树不空,则右子树上所有节点的值均大于或等于它的根节点的值;
  3. 任意节点的左、右子树也分别为二叉查找树;

一颗二叉查找树如下图所示:

但是,当我们顺序插入一系列节点后,二叉查找树就退化为线性表,如下图所示:

 

虽然二叉查找树退化为线性表之后,最坏效率降为 O(n)。但依旧有很多改进版的二叉查找树可以使树高为O(log n),从而将最坏效率降至O(log n),如AVL树、红黑树等。

4. AVL树

AVL 树是最早被发明的自平衡二叉查找树。在AVL树中,任一节点对应的两棵子树的最大高度差为 1 ,因此它也被称为高度平衡树。查找、插入和删除在平均和最坏情况下的时间复杂度都是 O(logn)。增加和删除元素的操作则可能需要借由一次或多次树旋转,以实现树的重新平衡。

在 AVL 树中,节点的**平衡因子**是它的左子树的高度减去它的右子树的高度(有时相反)。带有平衡因子 1、0 或 -1 的节点被认为是平衡的。带有平衡因子 -2 或 2 的节点被认为是不平衡的,并需要重新平衡这个树。平衡因子可以直接存储在每个节点中,或从可能存储在节点中的子树高度计算出来。

AVL 树自平衡的方式是做一次或多次所谓的"AVL旋转"。

5. 红黑树

红黑树(英语:Red–black tree)也是一种自平衡二叉查找树。红黑树在二叉查找树的基础之上,对每个节点都增加了颜色属性,分为红色或黑色。在二叉查找树强制的一般要求以外,对于任何有效的红黑树增加了如下 5 条额外要求:

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

下面是一个具体的红黑树的图例:

 

红黑树的这些特性,保证了红黑树**从根到叶子的最长的可能路径不多于最短的可能路径的两倍长**,造成的结果是红黑树大致上是平衡的。如上图所示,"nil叶子"或"空(null)叶子"不包含数据而只充当树在此结束的指示。

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

本章节开始结合 HashMap 源码讲解红黑树的插入、红黑树的删除、红黑树的自平衡、左旋、右旋等内容。

6. 链表转为红黑树的过程

HashMap 的桶类型除了 Node (链表)外,还有 TreeNode (树)。

TreeNode 类包含成员方法 `treeify()` ,该方法的作用是形成以当前 TreeNode 对象为根节点的红黑树。

树化过程不外乎是循环遍历链表,构造一颗二叉查找树,最后使用红黑树的平衡方法进行自平衡。

该方法源码如下: 

  1. final void treeify(Node[] tab) {  
  2. TreeNode root = null 
  3. // 步骤①:循环遍历当前TreeNode链表  
  4. for (TreeNode x = this, next; x != null; x = next) {  
  5. next = (TreeNode)x.next 
  6. x.left = x.right = null 
  7. // 步骤②:如果还未设置根节点..  
  8. if (root == null) {  
  9. x.parent = null 
  10. x.red = false 
  11. root = x;  
  12.  
  13. // 步骤③:如果已设置根节点..  
  14. else {  
  15. K k = x.key 
  16. int h = x.hash;  
  17. Class kc = null 
  18. // 步骤④:从根节点开始遍历,插入新节点  
  19. for (TreeNode p = root;;) {  
  20. int dir, ph;  
  21. K pk = p.key 
  22. // 步骤⑤:比较当前节点的hash值与新节点的hash值  
  23. // 若是新节点hash值较小  
  24. if ((ph = p.hash) > h)  
  25. dir = -1;  
  26. // 若是新节点的hash值较大  
  27. else if (ph < h)  
  28. dir = 1;  
  29. // 若是新节点与当前节点的hash值相等  
  30. else if (  
  31. // 如果新节点的key没有实现Comparable接口..  
  32. (kc == null && (kc = comparableClassFor(k)) == null 
  33. // 或者实现了Comparable接口但是k.compareTo(pk)结果为0  
  34. ||(dir = compareComparables(kc, k, pk)) == 0)  
  35. // 则调用tieBreakOrder继续比较大小  
  36. dir = tieBreakOrder(k, pk);  
  37. TreeNode xp = p;  
  38. // 步骤⑥:如果新节点经比较后小于等于当前节点且当前节点的左子节点为null,则插入新节点,反之亦然  
  39. if ((p = (dir <= 0) ? p.left : p.right) == null) {  
  40. x.parent = xp;  
  41. if (dir <= 0)  
  42. xp.left = x;  
  43. else  
  44. xp.right = x;  
  45. // 步骤⑦:平衡红黑树  
  46. root = balanceInsertion(root, x);  
  47. break;  
  48.  
  49.  
  50.  
  51.  
  52. // 步骤⑧:确保给定的根节点是所在桶的第一个节点  
  53. moveRootToFront(tab, root);  
  54. }  

7. 红黑树的左旋和右旋

红黑树的左旋和右旋其实很简单,所谓的左旋是把要平衡的节点向左下旋转成一个叶子节点。

如下图所示:

 

所谓的右旋是把要平衡的节点向右下旋转成一个叶子节点。

如下图所示:

 

在 HashMap 的源码中,左旋的实现代码如下: 

  1. static TreeNode rotateLeft(TreeNode root,  
  2. TreeNode p) {  
  3. // p: 当前节点  
  4. // r: 当前节点的右儿子  
  5. // rl: 当前节点的右儿子的左儿子  
  6. TreeNode r, pp, rl;  
  7. // 如果当前节点和当前节点的右儿子不为空,就左旋  
  8. if (p != null && (r = p.right) != null) {  
  9. // 步骤①:当前节点的右儿子的左儿子成为当前节点的右儿子  
  10. if ((rl = p.right = r.left) != null 
  11. rl.parent = p;  
  12. // 步骤②:当前节点的右儿子成为当前节点的父节点  
  13. if ((pp = r.parent = p.parent) == null 
  14. // 如果当前节点是根节点,那么r的颜色必须是黑色  
  15. (root = r).red = false 
  16. else if (pp.left == p)  
  17. pp.left = r;  
  18. else  
  19. pp.right = r;  
  20. r.left = p;  
  21. p.parent = r;  
  22.  
  23. return root;  
  24. }

右旋的实现代码如下: 

  1. static TreeNode rotateRight(TreeNode root,  
  2. TreeNode p) {  
  3. // p: 当前节点  
  4. // l: 当前节点的左儿子  
  5. // lr: 当前节点的左儿子的右儿子  
  6. TreeNode l, pp, lr;  
  7. // 如果当前节点和当前节点的左儿子不为空,就右旋  
  8. if (p != null && (l = p.left) != null) {  
  9. // 步骤①:当前节点的左儿子的右儿子成为当前节点的左儿子  
  10. if ((lr = p.left = l.right) != null 
  11. lr.parent = p;  
  12. // 步骤②:当前节点的左儿子成为当前节点的父节点  
  13. if ((pp = l.parent = p.parent) == null 
  14. // 如果当前节点是根节点,那么r的颜色必须是黑色  
  15. (root = l).red = false 
  16. else if (pp.right == p)  
  17. pp.right = l;  
  18. else  
  19. pp.left = l;  
  20. l.right = p;  
  21. p.parent = l;  
  22.  
  23. return root;  
  24. }

 8. 红黑树的插入

红黑树是一棵特殊的二叉查找树,如同二叉查找树的插入一样,红黑树的插入,也需要判断插入节点与当前节点的大小。如果插入节点大于或等于当前节点,则插入当前节点的右子树;否则,插入到左子树。循环此过程,直到找到为空的叶子节点放入即可。 

  1. /**  
  2. * Tree version of putVal.  
  3. */  
  4. final TreeNode putTreeVal(HashMap map, Node[] tab,  
  5. int h, K k, V v) {  
  6. Class kc = null 
  7. boolean searched = false 
  8. // root: 树根节点  
  9. TreeNode root = (parent != null) ? root() : this;  
  10. for (TreeNode p = root;;) {  
  11. // p: 当前节点  
  12. // dir: 标识新节点应该插入到当前节点的左子树还是右子树  
  13. // ph: 当前节点的hash值  
  14. // pk: 当前节点的key  
  15. int dir, ph; K pk;  
  16. // 步骤①:判断新节点应该插入到当前节点的左子树还是右子树  
  17. if ((ph = p.hash) > h)  
  18. dir = -1;  
  19. else if (ph < h)  
  20. dir = 1;  
  21. else if ((pk = p.key) == k || (k != null && k.equals(pk)))  
  22. return p;  
  23. else if ((kc == null &&  
  24. (kc = comparableClassFor(k)) == null) ||  
  25. (dir = compareComparables(kc, k, pk)) == 0) {  
  26. if (!searched) {  
  27. TreeNode q, ch;  
  28. searched = true 
  29. if (((ch = p.left) != null &&  
  30. (q = ch.find(h, k, kc)) != null) ||  
  31. ((ch = p.right) != null &&  
  32. (q = ch.find(h, k, kc)) != null))  
  33. return q;  
  34.  
  35. dir = tieBreakOrder(k, pk);  
  36.  
  37. TreeNode xp = p;  
  38. // 步骤②:终于从上向下遍历到了空的叶子节点,插入新节点  
  39. if ((p = (dir <= 0) ? p.left : p.right) == null) {  
  40. // 1.父节点指向新节点  
  41. Node xpn = xp.next 
  42. TreeNode x = map.newTreeNode(h, k, v, xpn);  
  43. // 新节点位置在左子树还是右子树  
  44. if (dir <= 0)  
  45. xp.left = x;  
  46. else  
  47. xp.right = x;  
  48. xp.next = x;  
  49. // 2.新节点指向父节点  
  50. x.parent = x.prev = xp;  
  51. if (xpn != null 
  52. // 如果有兄弟节点,兄弟节点的上一个节点指向新节点  
  53. ((TreeNode)xpn).prev = x;  
  54. // 最后平衡红黑树,并保证根节点在哈希桶的头部  
  55. moveRootToFront(tab, balanceInsertion(root, x));  
  56. return null 
  57.  
  58.  
  59. }

红黑树的插入与二叉查找树的不同之处,在于最后的平衡部分。

插入节点之后的红黑树平衡方法如下: 

  1. /**  
  2. * 平衡红黑树-当新增后  
  3. * x: 影响平衡的点(俗称:当前节点)  
  4. */  
  5. static TreeNode balanceInsertion(TreeNode root,  
  6. TreeNode x) {  
  7. // 新增节点x默认是红色  
  8. x.red = true 
  9. // xp父节点 xpp祖父节点 xppl祖父左节点 xppr 祖父右节点(p: parent, l: left, r: right 
  10. for (TreeNode xp, xpp, xppl, xppr;;) {  
  11. if ((xp = x.parent) == null) {  
  12. // 步骤①:如果插入的是根节点,根据红黑树性质之根节点是黑色,直接涂黑即可  
  13. x.red = false 
  14. return x;  
  15.  
  16. // 步骤②:插入节点的父节点是黑色或者父节点是根节点,红黑树没有被破坏,不需要调整  
  17. else if (!xp.red || (xpp = xp.parent) == null 
  18. return root;  
  19. // 父节点是祖父节点的左子节点  
  20. if (xp == (xppl = xpp.left)) {  
  21. // 步骤③:当前节点的父节点是红色,且叔叔节点也是红色  
  22. if ((xppr = xpp.right) != null && xppr.red) {  
  23. // 1.叔叔节点设置为黑色  
  24. xppr.red = false 
  25. // 2.父节点设置为黑色  
  26. xp.red = false 
  27. // 3.祖父节点设置为红色  
  28. xpp.red = true 
  29. // 4.将当前节点指向祖父节点,从新的当前节点重新开始算法  
  30. x = xpp;  
  31.  
  32. else {  
  33. // 步骤④:当前节点的父节点是红色,且叔叔节点是黑色,当前节点是其父右子节点  
  34. if (x == xp.right) {  
  35. // 当前节点的父节点作为新的当前节点,以新当节点为支点左旋  
  36. root = rotateLeft(root, x = xp);  
  37. // 重新赋值  
  38. xpp = (xp = x.parent) == null ? null : xp.parent;  
  39.  
  40. if (xp != null) {  
  41. // 父节点设置为黑色  
  42. xp.red = false 
  43. if (xpp != null) {  
  44. // 祖父节点设置为红色  
  45. xpp.red = true 
  46. // 以祖父节点为支点右旋  
  47. root = rotateRight(root, xpp);  
  48.  
  49.  
  50.  
  51.  
  52. // 父节点是祖父节点的右子节点  
  53. else {  
  54. // 步骤③:当前节点的父节点是红色,且叔叔节点也是红色  
  55. if (xppl != null && xppl.red) {  
  56. // 1.叔叔节点设置为黑色  
  57. xppl.red = false 
  58. // 2.父节点设置为黑色  
  59. xp.red = false 
  60. // 3.祖父节点设置为红色  
  61. xpp.red = true 
  62. // 4.将当前节点指向祖父节点,从新的当前节点重新开始算法  
  63. x = xpp;  
  64.  
  65. else {  
  66. // 步骤④:当前节点的父节点是红色,且叔叔节点是黑色,当前节点是其父左子节点  
  67. if (x == xp.left) {  
  68. // 当前节点的父节点作为新的当前节点,以新当节点为支点右旋  
  69. root = rotateRight(root, x = xp);  
  70. // 重新赋值  
  71. xpp = (xp = x.parent) == null ? null : xp.parent;  
  72.  
  73. if (xp != null) {  
  74. // 父节点设置为黑色  
  75. xp.red = false 
  76. if (xpp != null) {  
  77. // 祖父节点设置为红色  
  78. xpp.red = true 
  79. // 以祖父节点为支点左旋  
  80. root = rotateLeft(root, xpp);  
  81.  
  82.  
  83.  
  84.  
  85.  
  86. }

10. 红黑树的删除

红黑树的删除相比插入更加复杂,需要先从链表结构上进行删除,也就是处理当前节点的 next 指针与 prev 指针。如果当前树的节点太少,那么就将树退化为链表。然后,查找被删除节点的后继节点。所谓后继节点,就是当删除节点p后,如果p有子节点,p的子节点上移继承其位置。当找到后继节点,立即删除节点,如果被删除节点是红色,那么不需要平衡,否则平衡红黑树。

红黑树的删除方法代码如下: 

  1. final void removeTreeNode(HashMap map, Node[] tab,  
  2. boolean movable) {  
  3. int n;  
  4. if (tab == null || (n = tab.length) == 0) 
  5.  
  6. // 不处理空的哈希表 
  7.  
  8. return
  9.  
  10. // 第 1 部分: 处理链表结构  
  11. // 通过被删除的key的哈希值查找桶(红黑树)的位置 
  12.  
  13. int index = (n - 1) & hash;  
  14. // first、root: 红黑树的根节点  
  15. TreeNode first = (TreeNode)tab[index], root = first, rl; 
  16.  
  17. // succ: 当前节点(被删除节点)的下一个节点 
  18.  
  19. // pred: 当前节点(被删除节点)的上一个节点  
  20. TreeNode succ = (TreeNode)next, pred = prev; 
  21.  
  22. // 步骤①:从链表结构上删除当前节点(处理上一个节点的 next 指针与下一个节点的 prev 指针) 
  23.  
  24. if (pred == null 
  25. // 上一个节点为空,说明是要删除的是根节点,将红黑树的根节点索引改为当前节点的下一个节点  
  26. tab[index] = first = succ;  
  27. else  
  28. // 删除的不是根节点,就把当前节点的上一个节点的next指向当前节点的下一个节点 
  29. pred.next = succ; 
  30.  
  31. if (succ != null 
  32. // 当前节点的下一个节点不为空,就把下一个节点的prev指向当前节点的上一个节点  
  33. succ.prev = pred; 
  34.  
  35. // 步骤②:删除节点完毕后,红黑树为空,直接返回 
  36.  
  37. if (first == null 
  38. return
  39.  
  40. // 步骤③:更新root指针,并判断是否需要将树转为链表结构 
  41.  
  42. if (root.parent != null 
  43. root = root.root();  
  44. if (root == null || root.right == null ||  
  45. (rl = root.left) == null || rl.left == null) {  
  46. tab[index] = first.untreeify(map); // too small  
  47. return
  48.  
  49.  
  50. // 第 2 部分: 处理树结构  
  51. // p: 被删除的节点; replacement: 后继节点(删除节点p后,如果p有子节点,p的子节点上移继承其位置)  
  52. // pl: 被删除节点的左儿子; pr: 被删除节点的右儿子 
  53.  
  54. TreeNode p = this, pl = left, pr = right, replacement; 
  55.  
  56. // 步骤①:查找被删除节点的后继节点,分以下几种情况  
  57. // ⑴ 被删除节点有左子树和右子树 
  58.  
  59. if (pl != null && pr != null) {  
  60. TreeNode s = pr, sl; 
  61.  
  62. // 1.查找右子树最左叶子节点s与待删除节点p进行位置互换  
  63. while ((sl = s.left) != null) // find successor  
  64. s = sl;  
  65. // 2.交换最左叶子节点和待删除节点的颜色  
  66. boolean c = s.red; s.red = p.red; p.red = c; // swap colors  
  67. // sr:最左叶子节点的右儿子  
  68. TreeNode sr = s.right 
  69. // pp:被删除节点的父节点  
  70. TreeNode pp = p.parent; 
  71. // 3.交换被删除节点p和最左叶子节点s的位置  
  72. // 判断最左叶子节点是否是被删除节点的右儿子(即右子树是否只有一个节点)分别处理节点s.right和p.parent的引用  
  73. if (s == pr) { // p was s's direct parent  
  74. p.parent = s;  
  75. s.right = p;  
  76.  
  77. else {  
  78. TreeNode sp = s.parent;  
  79. if ((p.parent = sp) != null) {  
  80. if (s == sp.left 
  81. sp.left = p;  
  82. else  
  83. sp.right = p;  
  84.  
  85. if ((s.right = pr) != null 
  86. pr.parent = s;  
  87.  
  88. p.left = null; 
  89. // 处理p.right和sr.parent的引用  
  90. if ((p.right = sr) != null 
  91. sr.parent = p;  
  92. // 处理s.left和pl.parent的引用  
  93. if ((s.left = pl) != null 
  94. pl.parent = s;  
  95. // 处理s.parent的引用和pp.left或pp.right  
  96. if ((s.parent = pp) == null 
  97. root = s;  
  98. else if (p == pp.left 
  99. pp.left = s;  
  100. else  
  101. pp.right = s; 
  102.  
  103. // 4.交换最左叶子节点和被删除节点的位置完成  
  104. // 此时被删除节点p在原最左叶子节点的位置,现在被删除节点p没有左子树,如果有右子树,那么右儿子sr就是后继节点,否则后继节点指向自身 
  105.  
  106. if (sr != null 
  107. replacement = sr;  
  108. else  
  109. replacement = p;  
  110.  
  111. // ⑵ 被删除节点只有左子树,后继节点就是左儿子
  112.  
  113. else if (pl != null 
  114. replacement = pl;  
  115. // ⑶ 被删除节点只有右子树,后继节点就是右儿子  
  116. else if (pr != null 
  117. replacement = pr;  
  118. // ⑷ 被删除节点没有子树,那么后继节点就指向自身  
  119. else  
  120. replacement = p;  
  121. // 步骤②:已经找到删除节点后的后继节点,这一步将从树中彻底删除节点p。 
  122.  
  123. if (replacement != p) {  
  124. // 1.修改替代节点的父节点引用  
  125. TreeNode pp = replacement.parent = p.parent; 
  126.  
  127. // 2.将被删除节点的父节点对其的引用进行修改  
  128. if (pp == null 
  129. root = replacement;  
  130. else if (p == pp.left 
  131. pp.left = replacement;  
  132. else  
  133. pp.right = replacement;  
  134. // 3.彻底删除节点p  
  135. p.left = p.right = p.parent = null 
  136.  
  137. // 步骤③:删除节点完成,删除的是红色节点,不需要平衡;否则,平衡  
  138. TreeNode r = p.red ? root : balanceDeletion(root, replacement);  
  139. // 步骤④:若没有后继节点,直接删除节点p  
  140. if (replacement == p) { // detach  
  141. TreeNode pp = p.parent;  
  142. p.parent = null 
  143. if (pp != null) { 
  144.  
  145. if (p == pp.left 
  146. pp.left = null 
  147. else if (p == pp.right 
  148. pp.right = null 
  149.  
  150.  
  151. if (movable)  
  152. // 确保节点r是树根  
  153. moveRootToFront(tab, r);  
  154.  

删除红黑树节点之后的平衡代码如下,笔者认为,应当重点关注红黑树的变色、旋转逻辑。 

  1. /**  
  2. * 平衡红黑树-当删除后  
  3. * x: 影响平衡的点(俗称:当前节点)  
  4. */ 
  5.  
  6. static TreeNode balanceDeletion(TreeNode root,  
  7. TreeNode x) { 
  8.  
  9. // xp: 当前节点的父节点 
  10.  
  11. // xpl: 当前节点的父节点的左儿子 
  12.  
  13. // xpr: 当前节点的父节点的右儿子 
  14.  
  15. for (TreeNode xp, xpl, xpr;;) { 
  16.  
  17. // 步骤①:当前节点是null或者是根节点,不改变红黑树的结构,不需要改变 
  18.  
  19. if (x == null || x == root) 
  20.  
  21. return root; 
  22.  
  23. // 步骤②:当前节点成为根节点,置为黑色 
  24.  
  25. else if ((xp = x.parent) == null) { 
  26.  
  27. x.red = false
  28.  
  29. return x; 
  30.  
  31.  
  32. // 步骤③:当前节点为红色,改为黑色后,不影响路径上黑色的数量,不需要改变 
  33.  
  34. else if (x.red) { 
  35.  
  36. x.red = false
  37.  
  38. return root; 
  39.  
  40.  
  41. // 步骤④:当前节点为父节点的左儿子 
  42.  
  43. else if ((xpl = xp.left) == x) { 
  44.  
  45. // 如果兄弟节点是红色,那么父节点一定是黑色 
  46.  
  47. if ((xpr = xp.right) != null && xpr.red) { 
  48.  
  49. // 1.兄弟节点置为黑色 
  50.  
  51. xpr.red = false
  52.  
  53. // 2.父节点置为红色 
  54.  
  55. xp.red = true
  56.  
  57. // 3.以父节点为支点左旋  
  58. root = rotateLeft(root, xp); 
  59.  
  60. // 4.刷新兄弟节点  
  61. xpr = (xp = x.parent) == null ? null : xp.right
  62.  
  63.  
  64. // 如果兄弟节点为空,将当前节点向上调整为父节点,继续循环 
  65.  
  66. if (xpr == null
  67.  
  68. x = xp; 
  69.  
  70. else { 
  71.  
  72. // sl: 兄弟节点的左儿子  
  73. // sr: 兄弟节点的右儿子  
  74. TreeNode sl = xpr.left, sr = xpr.right 
  75. if ((sr == null || !sr.red) &&  
  76. (sl == null || !sl.red)) {  
  77. // 如果兄弟节点没有红色孩子,则兄弟节点置为红色  
  78. xpr.red = true 
  79. // 本轮结束,将当前节点向上调整为父节点,继续循环  
  80. x = xp;
  81.  
  82.  
  83. else { 
  84.  
  85. if (sr == null || !sr.red) {  
  86. // 如果兄弟节点的左儿子是红色就改为黑色  
  87. if (sl != null 
  88. sl.red = false 
  89. // 并将兄弟节点置为红色  
  90. xpr.red = true 
  91. // 以兄弟节点为支点右旋  
  92. root = rotateRight(root, xpr);  
  93. // 刷新兄弟节点  
  94. xpr = (xp = x.parent) == null ?  
  95. null : xp.right 
  96.  
  97. if (xpr != null) {  
  98. // 将兄弟节点的颜色染成和父节点一样  
  99. xpr.red = (xp == null) ? false : xp.red;
  100.  
  101. // 将兄弟节点的右儿子染成黑色,防止出现两个红色节点相连  
  102. if ((sr = xpr.right) != null 
  103. sr.red = false 
  104.  
  105. if (xp != null) {  
  106. // 父节点置为黑色,并对其左旋,这样就能保证被删除的节点所在的路径又多了一个黑色节点,从而达到恢复平衡的目的  
  107. xp.red = false 
  108. root = rotateLeft(root, xp);  
  109.  
  110. // 调整完毕,下一次循环直接退出  
  111. x = root;  
  112.  
  113.  
  114.  
  115. // 步骤⑤:当前节点为父节点的右儿子,同上 
  116.  
  117. else { // symmetric  
  118. if (xpl != null && xpl.red) {  
  119. xpl.red = false 
  120. xp.red = true 
  121. root = rotateRight(root, xp);  
  122. xpl = (xp = x.parent) == null ? null : xp.left 
  123.  
  124. if (xpl == null 
  125. x = xp; 
  126.  
  127. else {  
  128. TreeNode sl = xpl.left, sr = xpl.right 
  129. if ((sl == null || !sl.red) && 
  130.  
  131. (sr == null || !sr.red)) {  
  132. xpl.red = true 
  133. x = xp; 
  134.  
  135.  
  136. else {  
  137. if (sl == null || !sl.red) {  
  138. if (sr != null 
  139. sr.red = false 
  140. xpl.red = true 
  141. root = rotateLeft(root, xpl);  
  142. xpl = (xp = x.parent) == null ?  
  143. null : xp.left
  144.  
  145.  
  146. if (xpl != null) {  
  147. xpl.red = (xp == null) ? false : xp.red;  
  148. if ((sl = xpl.left) != null 
  149. sl.red = false 
  150.  
  151. if (xp != null) {  
  152. xp.red = false 
  153. root = rotateRight(root, xp);  
  154.  
  155. x = root; 
  156.  
  157.  
  158.  
  159.  

 

写在最后

这篇逐行解读 HashMap 源码的文章到此终于结束!笔者从开始写下此文的前半部分到现在,已经过了半年,期间由于功力不足、时间有限一直断断续续。不过所幸,笔者始终没有放弃,坚持要写出一篇高质量的解读 HashMap 源码的文章。但是,笔者虽已完成大部分核心源码的解读,但受限于自身功力浅薄,未能写的尽善尽美,本文仍旧存有许多遗憾之处。如果有读者想对本文进行补充的话,欢迎通过微信公众号“代码艺术”联系我!

学习资源

1. 红黑树在线演示网站:https://www.cs.usfca.edu/~galles/visualization/RedBlack.html 

作者介绍

薛勤,现就职于阿里巴巴,热衷于Java技术栈,对底层原理有独特的追求,个人在Github@ystcode上拥有多个开源项目。

【51CTO原创稿件,合作站点转载请注明原文作者和出处为51CTO.com】

 

责任编辑:庞桂玉 来源: 51CTO
相关推荐

2019-10-12 08:36:48

Java程序员数据结构

2020-01-17 09:00:00

HashMapJava编程语言

2020-09-17 07:37:09

红黑树数据结构

2022-07-19 13:51:47

数据库Hikari连接池

2020-08-19 16:36:53

HashMap红黑树阈值

2015-06-15 10:32:44

Java核心源码解读

2009-12-11 10:02:46

Linux内存管理

2019-09-23 11:35:23

数据结构设计红黑树

2021-03-19 07:59:33

红黑树面试数据

2016-12-08 11:01:39

红黑树Java

2020-10-09 06:56:55

红黑树动图二叉树

2020-11-05 09:03:32

红黑树面试数据

2023-11-21 16:13:38

C++代码

2019-08-22 09:22:44

数据结构二叉搜索树

2020-11-05 13:12:47

红黑树

2022-05-27 08:18:00

HashMapHash哈希表

2023-03-31 08:24:29

数据结构算法数目

2023-08-29 08:31:13

B+树数据索引

2020-05-06 16:41:36

红黑树二叉查找树

2020-03-11 08:40:51

红黑树平衡二叉B树
点赞
收藏

51CTO技术栈公众号