当前位置:   article > 正文

【数据结构】平衡树之红黑树_红黑树的平衡调整过程

红黑树的平衡调整过程

        AVL树解决了二叉搜索树退化为单支树而引发的效率问题,是一种绝对平衡的二叉搜索树,其性质(每个节点的左右子树高度差绝对值都不超过1)使其在对数据进行搜索时始终能保持高效(详见【数据结构】平衡树之AVL树)。但是,当涉及一些结构上的修改场景(例如增删),因为要频繁通过旋转来维护其绝对平衡的结构特性,代价较高,使得其在这样的场景中性能十分低下。

        为了继续优化效率问题,后来又不断有大佬提出新的解决方案。1972年,Rudolf Bayer发明了最初红黑树(当时被称为平衡二叉B树)。而后在1978年, Leo J. Guibas 和 Robert Sedgewick 将其修改成了如今的红黑树。 

       红黑树(RBTree)是一种特殊的二叉平衡搜索树,通过一种特殊的“手法”来控制着二叉搜索树的平衡,使其效率与AVL树相当,且在应用和维护等方面整体优于AVL树。

        对于平衡的控制,红黑树放弃了AVL树中调整平衡因子的方式(任一节点的左右子树高度差的绝对值不大于1),而是在树的每个节点上增加了一个用于表示节点的颜色(可以是Red或Black)的存储位,通过对任意一条从根到某个叶子节点的路径上,各个节点着色方式的限制,确保最长路径不超过最短路径的2倍,以此来使一整棵树接近平衡。

        红黑树的应用十分的广泛,例如Java的集合框架 (HashMap、TreeMap、TreeSet)、C++ 的 STL(map/set、mutil_map/mutil_set)、linux内核等等,都将红黑树作为底层结构来使用过。

        本篇博客将通过,对红黑树性质的梳理和主要功能的模拟实现,来帮助读者更加全面地了解红黑树。

目录

一、红黑树的性质 

二、红黑树的模拟实现

1 - 树的构建

2 - 插入

3 - 完整代码

补、一些迷思

1.控制节点的红和黑,怎么就做到了“最长路径不超过最短路径的2倍”?

2.由红黑树与AVL树的性能比较,来说明为什么放弃绝对平衡

3.为什么新创建的节点/新插入的节点默认为红色?

4.对两种需控制平衡的情景的更多说明

5.插入构建红黑树的过程图解


一、红黑树的性质 

        红黑树是因其维护平衡的手段而得名的,通过对树节点颜色的控制,来实现“最长路径不超过最短路径的2倍”的近似平衡。它主要具备以下性质(或者说是它的平衡规则),且这些性质与维护平衡息息相关:

  1. 任意一个树节点的颜色非红即黑;
  2. 根节点的颜色必为黑;  
  3. 任意一个颜色为红的树节点,其孩子节点均为黑,双亲节点为黑(这意味着任意路径上没有连续的红色节点);  
  4. 对于任意一条从(不为空的叶节点下的)空节点通往根节点的路径,每条路径上黑色节点数量均相同; 
  5. 每个空节点默认为黑色。

(ps:NIL节点指的是空节点 )

二、红黑树的模拟实现

1 - 树的构建

  1. //用枚举体来标识节点的颜色
  2. enum Colour
  3. {
  4. RED, //0
  5. BLACK, //1
  6. };
  7. //创建一个树节点(三叉链结构)
  8. template<class K, class V>
  9. struct RBTreeNode
  10. {
  11. RBTreeNode<K, V>* _left; //左孩子
  12. RBTreeNode<K, V>* _right; //右孩子
  13. RBTreeNode<K, V>* _parent; //双亲
  14. pair<K, V> _kv; //节点的值
  15. Colour _col; //放弃了AVL树的平衡因子,改用颜色标识来控制平衡
  16. //一个树节点的构造函数
  17. RBTreeNode(const pair<K, V>&kv)
  18. : _left(nullptr)
  19. , _right(nullptr)
  20. , _parent(nullptr)
  21. , _kv(kv)
  22. , _col(RED) //默认一个新节点为红色
  23. {}
  24. };
  25. //创建一棵红黑树
  26. template<class K, class V>
  27. class RBTree
  28. {
  29. typedef RBTreeNode<K, V>Node;
  30. private:
  31. //根节点
  32. Node*_root = nullptr;
  33. };

2 - 插入

        红黑树可以看作是引入了颜色标识的二叉搜索树,与AVL树类似(详见【数据结构】平衡树之AVL树),它插入过程也大致分为两步:

  1. 按照二叉搜索树的方式插入新节点:利用插入的值创建一个新的树节点。树为空,就直接将新节点赋给根节点的指针;树不为空,就按二叉搜索树的性质查找到合适的插入位置,在合适位置插入新节点。
  2. 控制树的平衡:调整颜色+旋转。
  1. bool Insert(const pair<K, V>& kv)
  2. {
  3. //1.按照二叉搜索树的方式插入新节点
  4. if (_root == nullptr)
  5. {
  6. _root = new Node(kv);
  7. _root->_col = BLACK;
  8. return true;
  9. }
  10. Node* parent = nullptr;
  11. Node* cur = _root;
  12. while (cur)
  13. {
  14. if (cur->_kv.first < kv.first)
  15. {
  16. parent = cur;
  17. cur = cur->_right;
  18. }
  19. else if (cur->_kv.first > kv.first)
  20. {
  21. parent = cur;
  22. cur = cur->_left;
  23. }
  24. else
  25. {
  26. return false;
  27. }
  28. }
  29. cur = new Node(kv);
  30. cur->_col = RED;
  31. if (parent->_kv.first < kv.first)
  32. {
  33. parent->_right = cur;
  34. }
  35. else
  36. {
  37. parent->_left = cur;
  38. }
  39. cur->_parent = parent;
  40. //2.控制树的平衡
  41. //...
  42. }

        因为默认一个新插入的节点为红色(原因见下文“一些迷思”),所以插入后就存在两种情况:

  1. 新插入节点的双亲为黑色,则仍满足红黑树的平衡性质,无需控制平衡;
  2. 新插入节点的双亲为红色,则违背平衡性质三,需控制平衡。

        而当新插入节点的双亲为红色,该如何控制平衡呢?

       插入节点的双亲一定是红色的,双亲的双亲一定是黑色的,这两个节点的颜色是始终确定的。 插入的节点为红色,插入后出现了两个连续的红色节点,此时就需要调整节点的颜色。

        调整颜色既是控制树平衡的手段,也是检验是否需要旋转的途径调整颜色的对象不是新插入的节点,而是其双亲、双亲的兄弟和双亲的双亲

        详情见下图:

 【Tips】新节点插入后,平衡的控制具体取决于其双亲的兄弟节点:

  1. 双亲的兄弟节点存在且为红色,调整颜色后,继续向上更新;
  2. 双亲的兄弟节点不存在,或存在且为黑色,需先旋转,然后再调整颜色。

(关于将特殊情景下的结论推广到一般性结论的说明,以及旋转分类讨论的图解,见下文“一些迷思”)

 【Tips】旋转的分类讨论:

        设:c为当前节点,p为其双亲节点,g为其祖父节点,u为其双亲的兄弟节点,并且cur为红色,p为红色,g为黑色,u不存在或u存在且为黑色。

1、p为g的左孩子,c为p的左孩子,则针对g做右单旋转;

2、p为g的左孩子,c为p的右孩子,则针对p做左单旋转,再针对g做右单旋转;

3、p为g的右孩子,c为p的右孩子,则针对g做左单旋转;

4、p为g的右孩子,c为p的左孩子,则针对p做右单旋转;

  1. bool Insert(const pair<K, V>& kv)
  2. {
  3. //1.按照二叉搜索树的方式插入新节点
  4. //...(见前文)
  5. //2.控制平衡
  6. while (parent && parent->_col == RED)//其双亲存在且双亲为红,才需要处理
  7. {
  8. Node* grandfather = parent->_parent;
  9. if (parent == grandfather->_left)//parent是grandfather的左孩子
  10. {
  11. Node* uncle = grandfather->_right;//则uncle是grandfather的右孩子
  12. //uncle存在且为红
  13. if (uncle && uncle->_col == RED)
  14. {
  15. //变色
  16. parent->_col = uncle->_col = BLACK;
  17. grandfather->_col = RED;
  18. //继续向上处理
  19. cur = grandfather;
  20. parent = cur->_parent;
  21. }
  22. //uncle不存在,或存在且为黑
  23. else
  24. {
  25. if (cur == parent->_left)
  26. {
  27. // g
  28. // p
  29. // c
  30. //旋转
  31. RotateR(grandfather);
  32. //变色
  33. parent->_col = BLACK;
  34. grandfather->_col = RED;
  35. }
  36. else
  37. {
  38. // g
  39. // p
  40. // c
  41. //旋转
  42. RotateL(parent);
  43. RotateR(grandfather);
  44. //变色
  45. cur->_col = BLACK;
  46. grandfather->_col = RED;
  47. }
  48. break;
  49. }
  50. }
  51. else //parent == grandfather->_right //parent是grandfather的右孩子
  52. {
  53. Node* uncle = grandfather->_left;//则uncle是grandfather的左孩子
  54. // uncle存在且为红
  55. if (uncle && uncle->_col == RED)
  56. {
  57. //变色
  58. parent->_col = uncle->_col = BLACK;
  59. grandfather->_col = RED;
  60. //继续向上处理
  61. cur = grandfather;
  62. parent = cur->_parent;
  63. }
  64. //uncle不存在,或存在且为黑
  65. else
  66. {
  67. if (cur == parent->_right)
  68. {
  69. // g
  70. // p
  71. // c
  72. //旋转
  73. RotateR(grandfather);
  74. //变色
  75. parent->_col = BLACK;
  76. grandfather->_col = RED;
  77. }
  78. else
  79. {
  80. // g
  81. // p
  82. // c
  83. //旋转
  84. RotateL(parent);
  85. RotateR(grandfather);
  86. //变色
  87. cur->_col = BLACK;
  88. grandfather->_col = RED;
  89. }
  90. break;
  91. }
  92. }
  93. }
  94. _root->_col = BLACK;//无论什么情况,根始终要置为黑
  95. return true;
  96. }

(旋转的细节详见【数据结构】平衡树之AVL树) 

  1. //左单旋
  2. void RotateL(Node* parent)
  3. {
  4. Node* cur = parent->_right; //cur是parent的右孩子
  5. Node* curleft = cur->_left; //curleft是cur的左孩子
  6. //将parent及其左子树整体旋转下来,并将parent与cur、curleft与parent、cur与ppnode一一正确链接
  7. parent->_right = curleft;
  8. if (curleft)
  9. {
  10. curleft->_parent = parent; //curleft可能为空,若为空则无需将curleft->_parent与parent链接
  11. }
  12. cur->_left = parent;
  13. //需要旋转的树,可能是一个局部的子树
  14. //cur可能需要跟parent的双亲节点ppnode链接
  15. Node* ppnode = parent->_parent;
  16. parent->_parent = cur;
  17. if (parent == _root /*ppnode==nullptr*/) //旋转点在根,cur无需跟parent的双亲节点链接
  18. {
  19. _root = cur;
  20. cur->_parent = nullptr;
  21. }
  22. else //旋转点不在根,cur需要跟parent的双亲节点链接
  23. {
  24. if (ppnode->_left == parent) //parent是ppnode的左孩子,就让cur代替其成为左孩子
  25. {
  26. ppnode->_left = cur;
  27. }
  28. else //parent是ppnode的右孩子,就让cur代替其成为右孩子
  29. {
  30. ppnode->_right = cur;
  31. }
  32. cur->_parent = ppnode; //将cur的双亲节点置为ppnode
  33. }
  34. }
  35. //右单旋
  36. void RotateR(Node* parent)
  37. {
  38. Node* cur = parent->_left; //cur是parent的左孩子
  39. Node* curright = cur->_right; //curright是cur的右孩子
  40. //将parent及其右子树整体旋转下来,并将parent与cur、curleft与parent、cur与ppnode一一正确链接
  41. //链接curright与parent
  42. parent->_left = curright;
  43. if (curright)
  44. {
  45. curright->_parent = parent;
  46. }
  47. //链接cur与parent、cur与parent的双亲ppnode
  48. Node* ppnode = parent->_parent;
  49. cur->_right = parent;
  50. if (ppnode == nullptr)
  51. {
  52. _root = cur;
  53. cur->_parent = nullptr;
  54. }
  55. else
  56. {
  57. if (ppnode->_left == parent)
  58. {
  59. ppnode->_left = cur;
  60. }
  61. else
  62. {
  63. ppnode->_right = cur;
  64. }
  65. cur->_parent = ppnode;
  66. }
  67. }

3 - 完整代码

  1. #pragma once
  2. #include<iostream>
  3. using namespace std;
  4. enum Colour
  5. {
  6. RED,
  7. BLACK
  8. };
  9. template<class K,class V>
  10. struct RBTreeNode
  11. {
  12. RBTreeNode<K, V>* _left;
  13. RBTreeNode<K, V>* _right;
  14. RBTreeNode<K, V>* _parent;
  15. pair<K, V> _kv;
  16. Colour _col;
  17. RBTreeNode(const pair<K, V>& kv)
  18. :_left(nullptr)
  19. ,_right(nullptr)
  20. ,_parent(nullptr)
  21. ,_kv(kv)
  22. ,_col(RED)
  23. {}
  24. };
  25. template<class K, class V>
  26. struct RBTree
  27. {
  28. typedef RBTreeNode<K, V> Node;
  29. public:
  30. bool Insert(const pair<K, V>& kv)
  31. {
  32. if (_root == nullptr)
  33. {
  34. _root = new Node(kv);
  35. _root->_col = BLACK;
  36. return true;
  37. }
  38. Node* parent = nullptr;
  39. Node* cur = _root;
  40. while (cur)
  41. {
  42. if (cur->_kv.first < kv.first)
  43. {
  44. parent = cur;
  45. cur = cur->_right;
  46. }
  47. else if (cur->_kv.first > kv.first)
  48. {
  49. parent = cur;
  50. cur = cur->_left;
  51. }
  52. else
  53. {
  54. return false;
  55. }
  56. }
  57. cur = new Node(kv);
  58. cur->_col = RED;
  59. if (parent->_kv.first < kv.first)
  60. {
  61. parent->_right = cur;
  62. }
  63. else
  64. {
  65. parent->_left = cur;
  66. }
  67. cur->_parent = parent;
  68. //红黑树插入的关键是uncle
  69. //1、uncle存在且为红,则变色+继续向上更新
  70. //2、uncle不存在,或uncle存在且为黑,则旋转+变色
  71. while (parent && parent->_col == RED)
  72. {
  73. Node* grandfather = parent->_parent;
  74. if (parent == grandfather->_left)
  75. {
  76. Node* uncle = grandfather->_right;
  77. //uncle存在且为红
  78. if (uncle && uncle->_col == RED)
  79. {
  80. //变色
  81. parent->_col = uncle->_col = BLACK;
  82. grandfather->_col = RED;
  83. //继续向上处理
  84. cur = grandfather;
  85. parent = cur->_parent;
  86. }
  87. //uncle不存在,或存在且为黑
  88. else
  89. {
  90. if (cur == parent->_left)
  91. {
  92. // g
  93. // p
  94. // c
  95. RotateR(grandfather);
  96. parent->_col = BLACK;
  97. grandfather->_col = RED;
  98. }
  99. else
  100. {
  101. // g
  102. // p
  103. // c
  104. RotateL(parent);
  105. RotateR(grandfather);
  106. cur->_col = BLACK;
  107. grandfather->_col = RED;
  108. }
  109. break;
  110. }
  111. }
  112. else // parent == grandfather->_right
  113. {
  114. Node* uncle = grandfather->_left;
  115. // uncle存在且为红
  116. if (uncle && uncle->_col == RED)
  117. {
  118. //变色
  119. parent->_col = uncle->_col = BLACK;
  120. grandfather->_col = RED;
  121. //继续向上处理
  122. cur = grandfather;
  123. parent = cur->_parent;
  124. }
  125. //uncle不存在,或存在且为黑
  126. else
  127. {
  128. if (cur == parent->_right)
  129. {
  130. // g
  131. // p
  132. // c
  133. //旋转+变色
  134. RotateR(grandfather);
  135. parent->_col = BLACK;
  136. grandfather->_col = RED;
  137. }
  138. else
  139. {
  140. // g
  141. // p
  142. // c
  143. //旋转+变色
  144. RotateL(parent);
  145. RotateR(grandfather);
  146. cur->_col = BLACK;
  147. grandfather->_col = RED;
  148. }
  149. break;
  150. }
  151. }
  152. }
  153. _root->_col = BLACK;
  154. return true;
  155. }
  156. //左单旋
  157. void RotateL(Node* parent)
  158. {
  159. ++_rotateCount;
  160. Node* cur = parent->_right;
  161. Node* curleft = cur->_left;
  162. parent->_right = curleft;
  163. if (curleft)
  164. {
  165. curleft->_parent = parent;
  166. }
  167. cur->_left = parent;
  168. Node* ppnode = parent->_parent;
  169. parent->_parent = cur;
  170. if (parent == _root)
  171. {
  172. _root = cur;
  173. cur->_parent = nullptr;
  174. }
  175. else
  176. {
  177. if (ppnode->_left == parent)
  178. {
  179. ppnode->_left = cur;
  180. }
  181. else
  182. {
  183. ppnode->_right = cur;
  184. }
  185. cur->_parent = ppnode;
  186. }
  187. }
  188. //右单旋
  189. void RotateR(Node* parent)
  190. {
  191. ++_rotateCount;
  192. Node* cur = parent->_left;
  193. Node* curright = cur->_right;
  194. parent->_left = curright;
  195. if (curright)
  196. curright->_parent = parent;
  197. Node* ppnode = parent->_parent;
  198. cur->_right = parent;
  199. parent->_parent = cur;
  200. if (ppnode == nullptr)
  201. {
  202. _root = cur;
  203. cur->_parent = nullptr;
  204. }
  205. else
  206. {
  207. if (ppnode->_left == parent)
  208. {
  209. ppnode->_left = cur;
  210. }
  211. else
  212. {
  213. ppnode->_right = cur;
  214. }
  215. cur->_parent = ppnode;
  216. }
  217. }
  218. //验树的平衡
  219. bool CheckColourNum(Node* root, int blackNum,int benchmark)//递归查询连续的红色节点和验证黑色节点数量
  220. {
  221. if (root == nullptr)
  222. {
  223. if (blackNum != benchmark)//若某一条路径上的黑色节点数量与基准值不符,则说明树不平衡(无论基准值本身是否正确)
  224. {
  225. return false;
  226. }
  227. return true;
  228. }
  229. //递归记录某一条路径上的黑色节点数量
  230. if (root->_col == BLACK)
  231. {
  232. ++blackNum;
  233. }
  234. //出现两个连续的红色节点,则说明树不平衡
  235. if (root->_col == RED && root->_parent && root->_parent->_col == RED)
  236. {
  237. cout << root->_kv.first << "RED show up" << endl;
  238. return false;
  239. }
  240. //递归至左子树和右子树里去验证平衡
  241. return CheckColourNum(root->_left,blackNum,benchmark)
  242. && CheckColourNum(root->_right, blackNum, benchmark);
  243. }
  244. bool IsBalance()//封装一次“bool IsBalance(Node* root)”
  245. {
  246. return IsBalance(_root);
  247. }
  248. bool IsBalance(Node* root)//验树的平衡
  249. {
  250. //空树是平衡的
  251. if (root == nullptr)
  252. return true;
  253. //根节点不是黑色的,则说明树不平衡
  254. if (root->_col != BLACK)
  255. {
  256. return false;
  257. }
  258. //取最左路径上的黑色节点数量作为基准值
  259. Node* cur = _root;
  260. int benchmark = 0;
  261. while (cur)
  262. {
  263. if (cur->_col == BLACK)
  264. {
  265. ++benchmark;
  266. }
  267. cur = cur->_left;
  268. }
  269. //查询连续的红色节点和验证黑色节点数量
  270. return CheckColourNum(root,0,benchmark);//参数:根节点,当前路径的黑节点数量,黑节点数量的基准值
  271. }
  272. //求树的高度
  273. int Height()
  274. {
  275. return Height(_root);
  276. }
  277. int Height(Node* root)
  278. {
  279. if (root == nullptr)
  280. return 0;
  281. int leftHeight = Height(root->_left);
  282. int rightHeight = Height(root->_right);
  283. return leftHeight > rightHeight ? leftHeight + 1 : rightHeight + 1;
  284. }
  285. private:
  286. Node* _root = nullptr;
  287. };

补、一些迷思

1.控制节点的红和黑,怎么就做到了“最长路径不超过最短路径的2倍”?

        如下图,是一棵满足“最长路径不超过最短路径的2倍”的合法红黑树,它的最长路径为4(例如13->17->25->22),最短路径为3(例如13->8->1)。

         同样的,下图中是一种极端情况下的红黑树,但它仍是合法的。它的最长路径为4(例如13->17->25->22),最短路径为2(例如13->8)。

        由此可见,控制节点的红和黑,保证最短路径全黑、最长路径是一黑一红相间的(或每条路径上黑色节点的总数占整棵树节点总数的1/2),就满足了“最长路径不超过最短路径的2倍”。

2.由红黑树与AVL树的性能比较,来说明为什么放弃绝对平衡

平衡树特点查找效率大致为创建一棵树的数据量创建后树的高度所需时间约为
AVL树高度差的绝对值不超过1O(logN)1000个值100.000001s
红黑树最长路径不超过最短路径的2倍O(2*logN)1000个值200.000002s

        对于红黑树与AVL树,两者的性能在同一量级上,效率差距并不大,虽然在维护平衡时均需旋转,但AVL树因控制平衡更加严格,需在插入和删除时频繁通过旋转维护其结构,会付出较大的代价,故红黑树整体优于AVL树。

3.为什么新创建的节点/新插入的节点默认为红色?

        当插入的节点为黑色,此时在所插入的路径上,黑色节点的数量比其他路径上多1,就违背了红黑树的性质四(对于任意一条从空节点通往根节点的路径,每条路径上黑色节点数量均相同)。

        当插入的节点为红色,若此时所插入位置的双亲为红色,即出现了连续的红色节点,就违背了红黑树的性质三(任意路径上没有连续的红色节点)。

        也就是说,插入的节点无论是什么颜色的,都会对红黑树的平衡有潜在的影响。

        不过具体而言,若插入的节点为黑色,则是“一定会因违背性质四而破坏平衡”,此时是必须对红黑树进行调整的,且要对多条路径进行调整;而插入的节点为红色,则是“可能会因性质三而破坏平衡”,此时既可能要进行调整,也可能不进行调整,且调整时仅对一条路径进行调整。

        基于这样的利弊,插入节点为红色时维护平衡的代价在理论上更小,故使插入的新节点/新创建的节点默认为红色。

4.对两种需控制平衡的情景的更多说明

        设:cur为当前节点,p为其双亲节点,g为其祖父节点,u为其双亲的兄弟节点。

        情景一:cur为红,p为红,g为黑,u存在且为红。cur和p均为红,违背了性质三。解决方法为,将p,u改为黑,g改为红,然后把g当成cur,继续向上调整。

        当a/b/c/d/e为空树,cur是新插入的节点,情况较为简单,上述解决方法是适用的。 

        而当a/b/c/d/e不为空树,cur不是新插入的节点,尽管情况较为复杂,但上述解决方法仍适用。

        更复杂的情况又例如,c/d/e是每条路径都有两个黑色节点的子树...

        情景二:cur为红,p为红,g为黑,u不存在/u存在且为黑。 

        u不存在,做旋转的处理。

        u存在且为黑,做旋转和变色的处理。

【Tips】旋转的分类讨论:

1、p为g的左孩子,cur为p的左孩子,则针对g做右单旋转;

2、p为g的左孩子,cur为p的右孩子,则针对p做左单旋转,再针对g做右单旋转;

3、p为g的右孩子,cur为p的右孩子,则针对g做左单旋转;

4、p为g的右孩子,cur为p的左孩子,则针对p做右单旋转;

 

5.插入构建红黑树的过程图解

        升序构建红黑树

        降序构建红黑树

        随机插入构建红黑树 

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/Guff_9hys/article/detail/762480
推荐阅读
相关标签
  

闽ICP备14008679号