当前位置:   article > 正文

红黑树(RBTree)的插入算法以及如何测试一棵树是否是红黑树?(详细图解说明)_怎么判断一棵树是不是红黑树

怎么判断一棵树是不是红黑树

1.什么叫红黑树?

             红黑树是一棵二叉搜索树,它在每个节点上增加了一个存储位来表示节点的颜色,可以是Red或Black。通过对任何一条从根到叶子简单路径上的颜色来约束,红黑树保证最长路径不超过最短路径的两倍,因而近似于平衡。


2.红黑树的性质

     

   1. 每个节点,不是红色就是黑色的

   2. 根节点是黑色的

   3. 如果一个节点是红色的,则它的两个子节点是黑色的(没有连续的红色结点

   4. 对每个节点,从该节点到其所有后代叶节点的简单路径上,均包含相同数目的黑色节点。(每条路径的黑色结点个数相等)

   5. 每个叶子节点都是黑色的(这里的叶子节点是指的NIL节点(空节点))


3.红黑树为什么通过颜色的约束能够保证最长路径不超过最短路径的两倍?




4.红黑树的结点的定义

  1. enum Colour
  2. {
  3. RED,
  4. BLACK
  5. };
  6. template<typename K,typename V>
  7. struct RBTreeNode
  8. {
  9. //结点
  10. RBTreeNode* _left;
  11. RBTreeNode* _right;
  12. RBTreeNode* _parent;
  13. //颜色
  14. Colour _col;
  15. //KV
  16. K _key;
  17. V _value;
  18. RBTreeNode(const K& key, const V& value)
  19. :_key(key)
  20. , _value(value)
  21. , _left(NULL)
  22. , _right(NULL)
  23. , _parent(NULL)
  24. , _col(RED) //把新增结点置成红色为了保持每条路径的黑色结点个数不变
  25. {}
  26. };


为什么把新增结点的颜色设置为红色而不是黑色呢?

    因为设置为红色,比较容易维护这棵树,设置为黑色,会导致当前这条路径多了一个黑色结点,为了满足每条路径的黑色结点个数相同,需要维护多条路径。



5.红黑树插入的几种情况


注:由于p结点(父结点)可能为g结点(祖父结点)的左孩子或者右孩子,下面的作图以左孩子为模板,右孩子与左孩子类似,只是在某些需要旋转的情况下不一样。

    (1)插入结点为根节点(只有一个结点的情况),将结点颜色置成黑色即可

    

    (2)插入结点的父结点或调整结点的父结点为黑色结点(不需要调整)



  

  (3)插入结点的父结点或向上调整的结点父结点为红色之——uncle结点存在为红色

  

  (4)插入结点的父结点为红色之——uncle结点不存在



  (5)插入结点的父结点为红色之——uncle结点存在为黑色





6.红黑树插入实现代码


  1. template<class K,class V>
  2. class RBTree
  3. {
  4. typedef RBTreeNode<K, V> Node;
  5. public:
  6. RBTree()
  7. :_root(NULL)
  8. {}
  9. bool Insert(const K& key, const V& value)
  10. {
  11. //1.插入结点为根节点,必须为黑色
  12. if (_root == NULL)
  13. {
  14. _root = new Node(key, value);
  15. _root->_col = BLACK;
  16. return true;
  17. }
  18. Node* cur = _root;
  19. Node* parent = NULL;
  20. while (cur != NULL)
  21. {
  22. if (cur->_key > key)
  23. {
  24. parent = cur;
  25. cur = cur->_left;
  26. }
  27. else if (cur->_key < key)
  28. {
  29. parent = cur;
  30. cur = cur->_right;
  31. }
  32. else
  33. {
  34. return false;
  35. }
  36. }
  37. cur = new Node(key, value);
  38. if (key > parent->_key)
  39. parent->_right = cur;
  40. else
  41. parent->_left = cur;
  42. cur->_parent = parent;
  43. while (parent != NULL)
  44. {
  45. Node* pparent = parent->_parent;
  46. Node* uncle = NULL;
  47. //2.如果插入节点或者调整结点的父结点为黑色结点,增加一个红色结点,对每条路径的黑色结点个数没影响
  48. if (parent->_col == BLACK)
  49. {
  50. break;
  51. }
  52. //判断叔叔结点是否存在,存在是在左还是右?
  53. if (pparent != NULL)
  54. {
  55. if (pparent->_right == parent)
  56. uncle = pparent->_left;
  57. else
  58. uncle = pparent->_right;
  59. }
  60. //3.插入结点或调整结点父结点为红色且叔叔结点为红色
  61. if (uncle != NULL&&uncle->_col == RED)
  62. {
  63. parent->_col = BLACK;
  64. uncle->_col = BLACK;
  65. if (pparent == _root)
  66. return true;
  67. pparent->_col = RED;
  68. //继续向上调整
  69. //假如调整到根结点,只需将根结点的颜色置为黑色即可
  70. if (parent == _root)
  71. {
  72. parent->_col = BLACK;
  73. return true;
  74. }
  75. cur = parent->_parent;
  76. parent = cur->_parent;
  77. }
  78. //4.插入结点或调整结点父结点为红色且叔叔结点不存在或
  79. // 插入结点或调整结点父结点为红色且叔叔结点存在为黑色二者逻辑一样。
  80. else
  81. //剩下的情况 (uncle == NULL || uncle->_col == BLACK)
  82. {
  83. //判断父结点在左还是在有
  84. if (pparent->_left == parent)
  85. {
  86. //左--左
  87. if (parent->_left == cur)
  88. {
  89. _RotateR(pparent);
  90. pparent->_col = RED;
  91. parent->_col = BLACK;
  92. if (pparent == _root)
  93. {
  94. _root = parent;
  95. }
  96. }
  97. //左--右
  98. else
  99. {
  100. _RotateL(parent);
  101. _RotateR(pparent);
  102. cur->_col = BLACK;
  103. parent->_col = RED;
  104. pparent->_col = RED;
  105. if (pparent == _root)
  106. _root = cur;
  107. }
  108. }
  109. //父结点在右
  110. else
  111. {
  112. //右--右
  113. if (parent->_right == cur)
  114. {
  115. _RotateL(pparent);
  116. pparent->_col = RED;
  117. parent->_col = BLACK;
  118. if (pparent == _root)
  119. _root = parent;
  120. }
  121. //右--左
  122. else
  123. {
  124. _RotateR(parent);
  125. _RotateL(pparent);
  126. parent->_col = RED;
  127. pparent->_col = RED;
  128. cur->_col = BLACK;
  129. if (pparent == _root)
  130. _root = parent;
  131. }
  132. }
  133. break;
  134. }
  135. }
  136. return true;
  137. }
  138. void InOrder()
  139. {
  140. _InOrder(_root);
  141. cout << endl;
  142. }
  143. protected:
  144. void _InOrder(Node* root)
  145. {
  146. if (root == NULL)
  147. return;
  148. _InOrder(root->_left);
  149. cout << root->_key << " ";
  150. _InOrder(root->_right);
  151. }
  152. void _RotateR(Node* parent)
  153. {
  154. Node* SubL = parent->_left;
  155. Node* SubLR = SubL->_right;
  156. //链上SubLR
  157. parent->_left = SubLR;
  158. if (SubLR != NULL)
  159. SubLR->_parent = parent;
  160. Node* pparent = parent->_parent;
  161. //链上SubL
  162. SubL->_parent = pparent;
  163. if (pparent != NULL)
  164. {
  165. if (pparent->_left == parent)
  166. pparent->_left = SubL;
  167. else
  168. pparent->_right = SubL;
  169. }
  170. //链上parent
  171. parent->_parent = SubL;
  172. SubL->_right = parent;
  173. }
  174. void _RotateL(Node* parent)
  175. {
  176. Node* SubR = parent->_right;
  177. Node* SubRL= SubR->_left;
  178. //链上SubLR
  179. parent->_right= SubRL;
  180. if (SubRL != NULL)
  181. SubRL->_parent = parent;
  182. Node* pparent = parent->_parent;
  183. //链上SubL
  184. SubR->_parent = pparent;
  185. if (pparent != NULL)
  186. {
  187. if (pparent->_left == parent)
  188. pparent->_left = SubR;
  189. else
  190. pparent->_right = SubR;
  191. }
  192. //链上parent
  193. parent->_parent = SubR;
  194. SubR->_left = parent;
  195. }
  196. Node* _root;
  197. };

7.如何测试一棵树是不是红黑树?

(1)测试它的根节点是不是黑色

(2)测试是否有连续的红结点

(3)测试每条路径上的黑节点的数量是否相等?

以某一条路径(测试用例用的是左子树)上的黑色结点的个数为判断标准,如果有一条不满足相等,证明不是红黑树。

实现代码:

    

  1. bool Isbalance()
  2. {
  3. //1.空树认为是RBTree
  4. if (_root == NULL)
  5. return true;
  6. //2.根结点非黑不是RBTree
  7. if (_root->_col == RED)
  8. return false;
  9. int count = 0;
  10. //以左子树上的黑色结点作为一个基准,如果一样就证明是,不一样就证明不是RBTree
  11. Node* cur = _root;
  12. while (cur != NULL)
  13. {
  14. if (cur->_col == BLACK)
  15. count++;
  16. cur = cur->_left;
  17. }
  18. //num用来标记每条路径上的黑色结点的个数
  19. //count用来标记左子树上的黑色结点的个数
  20. int num = 0;
  21. return _Isbalance(_root, count, num);
  22. }
  23. bool _Isbalance(Node* root, int count, int num)
  24. {
  25. if (root == NULL)
  26. return true;
  27. //3.连续红结点不是RBTree,root结点一定有父结点
  28. if (root->_col == RED&&root->_parent->_col == RED)
  29. {
  30. cout << root->_key << " 有连续的红结点" << endl;
  31. return false;
  32. }
  33. if (root->_col == BLACK)
  34. num++;
  35. if (root->_left == NULL&&root->_right == NULL)
  36. {
  37. if (num != count)
  38. {
  39. cout << root->_key << " 黑色结点个数不一样" << endl;
  40. return false;
  41. }
  42. }
  43. return _Isbalance(root->_left, count, num) && _Isbalance(root->_right, count, num);
  44. }

  




8.测试用例

测试用例:

  1. void TestRBTree()
  2. {
  3. int a[] = { 16, 3, 7, 11, 9, 26, 18, 14, 15 };
  4. RBTree<int, int> t;
  5. for (size_t i = 0; i < sizeof(a) / sizeof(a[0]); i++)
  6. {
  7. t.Insert(a[i], i);
  8. }
  9. t.InOrder();
  10. cout << "Isbalance---->" << t.Isbalance()<< endl;
  11. }


       运行结果和测试用例红黑树模型:





9.红黑树和AVL树比较

      (1)红黑树和AVL树都是高效的平衡二叉树,增删查改的时间复杂度都是O(lg(N))。            

         (2)红黑树的不追求完全平衡,保证最长路径不超过最短路径的2倍,相对而言,降低了旋转的要求,所以性能会优于AVL树,所以实际运用中红黑树更多。(效率差不多要求还低)。

注:

AVL旋转因子的调节(详细图解)



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

闽ICP备14008679号