• 【数据结构】红黑树(C++实现)


    👀樊梓慕:个人主页

     🎥个人专栏:《C语言》《数据结构》《蓝桥杯试题》《LeetCode刷题笔记》《实训项目》《C++》《Linux》《算法》

    🌝每一个不曾起舞的日子,都是对生命的辜负


    目录

    前言

    1.概念

    2.性质

    3.节点的定义 

    4.插入

    4.1情况1:叔叔u存在且为红

    4.2情况2:叔叔u不存在或者叔叔u存在且为黑

    4.3代码实现

    5.验证

    6.红黑树完整源码

    7.AVL树与红黑树的性能比较


    前言

    如果没有现在的红黑树的话,那么可能set与map底层的数据结构就是AVL树了,那么红黑树的设计为什么能够取代AVL树的地位呢,红黑树的设计又有哪些奥秘,今天让我们一同来探索一下吧!


    欢迎大家📂收藏📂以便未来做题时可以快速找到思路,巧妙的方法可以事半功倍。 

    =========================================================================

    GITEE相关代码:🌟樊飞 (fanfei_c) - Gitee.com🌟

    =========================================================================


    1.概念

    『 红黑树』,是一种二叉搜索树,但在每个结点上增加一个存储位表示结点的颜色,可以是Red或
    Black。 通过对任何一条从根到叶子的路径上各个结点着色方式的『 限制』,红黑树确保没有一条路径会比其他路径长出俩倍,因而是『 接近平衡』的。

    红黑树不像AVL树那样绝对的平衡,而是接近平衡,

    但是这个影响是常数级别的,接着往下看。


    2.性质

    了解了红黑树的性质,你就明白了为什么红黑树是如何控制平衡的。

    • 每个结点不是红色就是黑色
    • 根节点是黑色的
    • 如果一个节点是红色的,则它的两个孩子结点必须是黑色的(没有连续的红色节点)。
    • 对于每个结点,从该结点到其所有后代叶结点的简单路径上,均包含相同数目的黑色结点。
    • 每个叶子结点都是黑色的(此处的叶子结点指的是空结点-NIL节点,因为路径指的是从根节点走到空,而不是从根节点走到叶子节点,所以设计者专门将空节点定义为NIL节点默认为黑色,就是为了防止人们数错路径)。

    接下来我们构建极端场景来分析下为什么最长路径中节点的个数不会超过最短路径节点个数的两倍?

    • 最短路径情况:节点为全黑。
    • 最长路径情况:一黑一红排列。
    • 大前提:每条路径都包含相同数量的黑色节点。

    举例:

    而且红黑树不是每次搜索都是2*logN。 


     AVL树与红黑树搜索效率基本平齐,但是AVL树实现更小的高度的方法是频繁地旋转,而旋转的开销还是比较大的,所以AVL树有点『 得不偿失』。

    红黑树虽然也需要旋转来控制高度,但是引入了颜色控制,可以减少旋转的次数。


    3.节点的定义 

    1. enum Colour
    2. {
    3. RED,
    4. BLACK
    5. };
    6. template<class K, class V>
    7. struct RBTreeNode
    8. {
    9. RBTreeNode* _left;
    10. RBTreeNode* _right;
    11. RBTreeNode* _parent;
    12. pair _kv;
    13. Colour _col;
    14. RBTreeNode(const pair& kv)
    15. : _left(nullptr)
    16. , _right(nullptr)
    17. , _parent(nullptr)
    18. , _kv(kv)
    19. , _col(RED)
    20. {}
    21. };

    思考:节点的默认颜色为什么是红色? 

    因为如果插入黑色节点就一定会破坏规则(每条路径上的黑色节点数相同),并且会影响其他路径。

    所以我们需要设置节点的默认颜色为红色


    4.插入

    因为新节点的默认颜色是红色,因此

    • 如果其双亲节点的颜色是黑色,没有违反红黑树任何性质,则不需要调整;
    • 但当新插入节点的双亲节点颜色为红色时,就违反了性质三不能有连在一起的红色节点,此时需要对红黑树分情况来讨论:

    所以以下都是当新插入节点的父节点为红色时的情况: 

    4.1情况1:叔叔u存在且为红

     g的颜色是确定为黑的,因为如果g不为黑,那么就证明上一层就已经不满足红黑树的规则了。

    此时c一定是新插入的节点,并且已经破坏了没有连续的红色节点的规则,所以我们需要进行调整。

    调整的办法:p/u变黑,g变红。

     g变红的目的是为了保持当前分支黑色节点的数目不变。

    但这里注意:

    • 如果g为根节点,那么需要再将g变黑(规则:根节点必须是黑色),即将所有路径的黑色节点数都+1,仍然符合规则。
    • 如果g不为根节点,那么g为红色,此时将g视为cur,继续向上调整。

    4.2情况2:叔叔u不存在或者叔叔u存在且为黑

     同样的g的颜色是确定为黑的,因为如果g不为黑,那么就证明上一层就已经不满足红黑树的规则了。

    • 如果叔叔u不存在,那么c一定是新插入的节点,因为叔叔u不存在说明在parent的下面不可能再挂黑色结点了。
    • 如果叔叔u存在且为黑,那么c一定不是新插入的节点,因为之前已经不满足规则了,所以c一定是刚刚变红的,也就是说有可能是情况1调整过后,将c变红的。 

    此时单纯地变色已经无法处理了,所以此时我们需要进行旋转。

    很明显根据之前学习的AVL树,面对如上图情况时我们需要对g进行右旋。 

    即当p为g的左,cur为p的左时对g进行右旋:

    注意:之前分析由于cur是因为之前的调整才变红的,所以cur向下的路径中一定有一个黑色节点与叔叔抵消,此时达到平衡。

    如果叔叔不存在也是一样的对g进行右旋即可:

    那么假如cur在p的右子树上呢?此时则需要先左旋再右旋。

    即当p为g左,cur为p的右时进行左右双旋,然后变色。

      同样的如果叔叔不存在也是一样的进行左右双旋,然后变色。

    总结下来就以上这些种情况,相信你掌握了左面的这些种情况,右面也非常简单。

    4.3代码实现

    1. bool Insert(const pair& kv)
    2. {
    3. if (_root == nullptr)
    4. {
    5. _root = new Node(kv);
    6. _root->_col = BLACK;
    7. return true;
    8. }
    9. Node* parent = nullptr;
    10. Node* cur = _root;
    11. while (cur)
    12. {
    13. if (cur->_kv.first < kv.first)
    14. {
    15. parent = cur;
    16. cur = cur->_right;
    17. }
    18. else if (cur->_kv.first > kv.first)
    19. {
    20. parent = cur;
    21. cur = cur->_left;
    22. }
    23. else
    24. {
    25. return false;
    26. }
    27. }
    28. cur = new Node(kv); // 红色的
    29. if (parent->_kv.first < kv.first)
    30. {
    31. parent->_right = cur;
    32. }
    33. else
    34. {
    35. parent->_left = cur;
    36. }
    37. cur->_parent = parent;
    38. while (parent && parent->_col == RED)
    39. {
    40. Node* grandfather = parent->_parent;
    41. if (parent == grandfather->_left)
    42. {
    43. Node* uncle = grandfather->_right;
    44. // 情况一:叔叔存在且为红
    45. if (uncle && uncle->_col == RED)
    46. {
    47. // 变色
    48. parent->_col = uncle->_col = BLACK;
    49. grandfather->_col = RED;
    50. // 继续往上处理
    51. cur = grandfather;
    52. parent = cur->_parent;
    53. }
    54. else
    55. {
    56. // 情况二:叔叔不存在或者存在且为黑
    57. // 旋转+变色
    58. if (cur == parent->_left)
    59. {
    60. // g
    61. // p u
    62. // c
    63. RotateR(grandfather);
    64. parent->_col = BLACK;
    65. grandfather->_col = RED;
    66. }
    67. else
    68. {
    69. // g
    70. // p u
    71. // c
    72. RotateL(parent);
    73. RotateR(grandfather);
    74. cur->_col = BLACK;
    75. grandfather->_col = RED;
    76. }
    77. break;
    78. }
    79. }
    80. else
    81. {
    82. Node* uncle = grandfather->_left;
    83. // 情况一:叔叔存在且为红
    84. if (uncle && uncle->_col == RED)
    85. {
    86. // 变色
    87. parent->_col = uncle->_col = BLACK;
    88. grandfather->_col = RED;
    89. // 继续往上处理
    90. cur = grandfather;
    91. parent = cur->_parent;
    92. }
    93. else
    94. {
    95. // 情况二:叔叔不存在或者存在且为黑
    96. // 旋转+变色
    97. // g
    98. // u p
    99. // c
    100. if (cur == parent->_right)
    101. {
    102. RotateL(grandfather);
    103. parent->_col = BLACK;
    104. grandfather->_col = RED;
    105. }
    106. else
    107. {
    108. // g
    109. // u p
    110. // c
    111. RotateR(parent);
    112. RotateL(grandfather);
    113. cur->_col = BLACK;
    114. grandfather->_col = RED;
    115. }
    116. break;
    117. }
    118. }
    119. }
    120. _root->_col = BLACK;
    121. return true;
    122. }
    123. void RotateL(Node* parent)
    124. {
    125. Node* subR = parent->_right;
    126. Node* subRL = subR->_left;
    127. parent->_right = subRL;
    128. if (subRL)
    129. subRL->_parent = parent;
    130. subR->_left = parent;
    131. Node* ppnode = parent->_parent;
    132. parent->_parent = subR;
    133. if (parent == _root)
    134. {
    135. _root = subR;
    136. subR->_parent = nullptr;
    137. }
    138. else
    139. {
    140. if (ppnode->_left == parent)
    141. {
    142. ppnode->_left = subR;
    143. }
    144. else
    145. {
    146. ppnode->_right = subR;
    147. }
    148. subR->_parent = ppnode;
    149. }
    150. }
    151. void RotateR(Node* parent)
    152. {
    153. Node* subL = parent->_left;
    154. Node* subLR = subL->_right;
    155. parent->_left = subLR;
    156. if (subLR)
    157. subLR->_parent = parent;
    158. subL->_right = parent;
    159. Node* ppnode = parent->_parent;
    160. parent->_parent = subL;
    161. if (parent == _root)
    162. {
    163. _root = subL;
    164. subL->_parent = nullptr;
    165. }
    166. else
    167. {
    168. if (ppnode->_left == parent)
    169. {
    170. ppnode->_left = subL;
    171. }
    172. else
    173. {
    174. ppnode->_right = subL;
    175. }
    176. subL->_parent = ppnode;
    177. }
    178. }

    5.验证

    红黑树的验证分为两步:

    首先要对树是否满足搜索特性进行检测,这一部分我们只需要对树进行中序遍历,观察得到的序列是否为有序序列即可。

    其次就是如何判断该树满足红黑树的特性呢?

    其实就是看是否满足红黑树的规则即可。

    即:

    • 根节点是黑色的;
    • 没有连续的红色节点;
    • 每条路径均包含相同数目的黑色结点。

    首先根节点是否为黑很好检测。

    其次是否有连续的红色节点,由于我们采用的是三叉链式结构,我们维护了父亲节点,所以只需要判断当前节点和父亲节点是否同时为红即可。

    最后就是如何判断每条路径的黑色节点数目是否相同,其实最简单的做法就是先统计一下最左路径的黑色节点数目,然后将该值作为参考,检验其他路径是否与该值相同即可。

    1. bool Check(Node* cur, int blackNum, int refBlackNum)
    2. {
    3. if (cur == nullptr)
    4. {
    5. if (refBlackNum != blackNum)
    6. {
    7. cout << "黑色节点的数量不相等" << endl;
    8. return false;
    9. }
    10. return true;
    11. }
    12. if (cur->_col == RED && cur->_parent->_col == RED)
    13. {
    14. cout << cur->_kv.first << "存在连续的红色节点" << endl;
    15. return false;
    16. }
    17. if (cur->_col == BLACK)
    18. ++blackNum;
    19. return Check(cur->_left, blackNum, refBlackNum)
    20. && Check(cur->_right, blackNum, refBlackNum);
    21. }
    22. bool IsBalance()
    23. {
    24. if (_root && _root->_col == RED)
    25. return false;
    26. int refBlackNum = 0;
    27. Node* cur = _root;
    28. while (cur)
    29. {
    30. if (cur->_col == BLACK)
    31. refBlackNum++;
    32. cur = cur->_left;
    33. }
    34. return Check(_root, 0, refBlackNum);
    35. }

    6.红黑树完整源码

    1. enum Colour
    2. {
    3. RED,
    4. BLACK
    5. };
    6. template<class K, class V>
    7. struct RBTreeNode
    8. {
    9. RBTreeNode* _left;
    10. RBTreeNode* _right;
    11. RBTreeNode* _parent;
    12. pair _kv;
    13. Colour _col;
    14. RBTreeNode(const pair& kv)
    15. :_left(nullptr)
    16. , _right(nullptr)
    17. , _parent(nullptr)
    18. , _kv(kv)
    19. , _col(RED)
    20. {}
    21. };
    22. template<class K, class V>
    23. class RBTree
    24. {
    25. typedef RBTreeNode Node;
    26. public:
    27. bool Insert(const pair& kv)
    28. {
    29. if (_root == nullptr)
    30. {
    31. _root = new Node(kv);
    32. _root->_col = BLACK;
    33. return true;
    34. }
    35. Node* parent = nullptr;
    36. Node* cur = _root;
    37. while (cur)
    38. {
    39. if (cur->_kv.first < kv.first)
    40. {
    41. parent = cur;
    42. cur = cur->_right;
    43. }
    44. else if (cur->_kv.first > kv.first)
    45. {
    46. parent = cur;
    47. cur = cur->_left;
    48. }
    49. else
    50. {
    51. return false;
    52. }
    53. }
    54. cur = new Node(kv); // 红色的
    55. if (parent->_kv.first < kv.first)
    56. {
    57. parent->_right = cur;
    58. }
    59. else
    60. {
    61. parent->_left = cur;
    62. }
    63. cur->_parent = parent;
    64. while (parent && parent->_col == RED)
    65. {
    66. Node* grandfather = parent->_parent;
    67. if (parent == grandfather->_left)
    68. {
    69. Node* uncle = grandfather->_right;
    70. // 情况一:叔叔存在且为红
    71. if (uncle && uncle->_col == RED)
    72. {
    73. // 变色
    74. parent->_col = uncle->_col = BLACK;
    75. grandfather->_col = RED;
    76. // 继续往上处理
    77. cur = grandfather;
    78. parent = cur->_parent;
    79. }
    80. else
    81. {
    82. // 情况二:叔叔不存在或者存在且为黑
    83. // 旋转+变色
    84. if (cur == parent->_left)
    85. {
    86. // g
    87. // p u
    88. // c
    89. RotateR(grandfather);
    90. parent->_col = BLACK;
    91. grandfather->_col = RED;
    92. }
    93. else
    94. {
    95. // g
    96. // p u
    97. // c
    98. RotateL(parent);
    99. RotateR(grandfather);
    100. cur->_col = BLACK;
    101. grandfather->_col = RED;
    102. }
    103. break;
    104. }
    105. }
    106. else
    107. {
    108. Node* uncle = grandfather->_left;
    109. // 情况一:叔叔存在且为红
    110. if (uncle && uncle->_col == RED)
    111. {
    112. // 变色
    113. parent->_col = uncle->_col = BLACK;
    114. grandfather->_col = RED;
    115. // 继续往上处理
    116. cur = grandfather;
    117. parent = cur->_parent;
    118. }
    119. else
    120. {
    121. // 情况二:叔叔不存在或者存在且为黑
    122. // 旋转+变色
    123. // g
    124. // u p
    125. // c
    126. if (cur == parent->_right)
    127. {
    128. RotateL(grandfather);
    129. parent->_col = BLACK;
    130. grandfather->_col = RED;
    131. }
    132. else
    133. {
    134. // g
    135. // u p
    136. // c
    137. RotateR(parent);
    138. RotateL(grandfather);
    139. cur->_col = BLACK;
    140. grandfather->_col = RED;
    141. }
    142. break;
    143. }
    144. }
    145. }
    146. _root->_col = BLACK;
    147. return true;
    148. }
    149. void RotateL(Node* parent)
    150. {
    151. ++rotateSize;
    152. Node* subR = parent->_right;
    153. Node* subRL = subR->_left;
    154. parent->_right = subRL;
    155. if (subRL)
    156. subRL->_parent = parent;
    157. subR->_left = parent;
    158. Node* ppnode = parent->_parent;
    159. parent->_parent = subR;
    160. if (parent == _root)
    161. {
    162. _root = subR;
    163. subR->_parent = nullptr;
    164. }
    165. else
    166. {
    167. if (ppnode->_left == parent)
    168. {
    169. ppnode->_left = subR;
    170. }
    171. else
    172. {
    173. ppnode->_right = subR;
    174. }
    175. subR->_parent = ppnode;
    176. }
    177. }
    178. void RotateR(Node* parent)
    179. {
    180. ++rotateSize;
    181. Node* subL = parent->_left;
    182. Node* subLR = subL->_right;
    183. parent->_left = subLR;
    184. if (subLR)
    185. subLR->_parent = parent;
    186. subL->_right = parent;
    187. Node* ppnode = parent->_parent;
    188. parent->_parent = subL;
    189. if (parent == _root)
    190. {
    191. _root = subL;
    192. subL->_parent = nullptr;
    193. }
    194. else
    195. {
    196. if (ppnode->_left == parent)
    197. {
    198. ppnode->_left = subL;
    199. }
    200. else
    201. {
    202. ppnode->_right = subL;
    203. }
    204. subL->_parent = ppnode;
    205. }
    206. }
    207. void _InOrder(Node* root)
    208. {
    209. if (root == nullptr)
    210. return;
    211. _InOrder(root->_left);
    212. cout << root->_kv.first << endl;
    213. _InOrder(root->_right);
    214. }
    215. void InOrder()
    216. {
    217. _InOrder(_root);
    218. }
    219. bool Check(Node* cur, int blackNum, int refBlackNum)
    220. {
    221. if (cur == nullptr)
    222. {
    223. if (refBlackNum != blackNum)
    224. {
    225. cout << "黑色节点的数量不相等" << endl;
    226. return false;
    227. }
    228. //cout << blackNum << endl;
    229. return true;
    230. }
    231. if (cur->_col == RED && cur->_parent->_col == RED)
    232. {
    233. cout << cur->_kv.first << "存在连续的红色节点" << endl;
    234. return false;
    235. }
    236. if (cur->_col == BLACK)
    237. ++blackNum;
    238. return Check(cur->_left, blackNum, refBlackNum)
    239. && Check(cur->_right, blackNum, refBlackNum);
    240. }
    241. bool IsBalance()
    242. {
    243. if (_root && _root->_col == RED)
    244. return false;
    245. int refBlackNum = 0;
    246. Node* cur = _root;
    247. while (cur)
    248. {
    249. if (cur->_col == BLACK)
    250. refBlackNum++;
    251. cur = cur->_left;
    252. }
    253. return Check(_root, 0, refBlackNum);
    254. }
    255. size_t Size()
    256. {
    257. return _Size(_root);
    258. }
    259. size_t _Size(Node* root)
    260. {
    261. if (root == NULL)
    262. return 0;
    263. return _Size(root->_left)
    264. + _Size(root->_right) + 1;
    265. }
    266. Node* Find(const K& key)
    267. {
    268. Node* cur = _root;
    269. while (cur)
    270. {
    271. if (cur->_kv.first < key)
    272. {
    273. cur = cur->_right;
    274. }
    275. else if (cur->_kv.first > key)
    276. {
    277. cur = cur->_left;
    278. }
    279. else
    280. {
    281. return cur;
    282. }
    283. }
    284. return NULL;
    285. }
    286. int _Height(Node* root)
    287. {
    288. if (root == nullptr)
    289. return 0;
    290. int leftHeight = _Height(root->_left);
    291. int rightHeight = _Height(root->_right);
    292. return leftHeight > rightHeight ? leftHeight + 1 : rightHeight + 1;
    293. }
    294. int Height()
    295. {
    296. return _Height(_root);
    297. }
    298. int GetRotateSize()
    299. {
    300. return rotateSize;
    301. }
    302. private:
    303. Node* _root = nullptr;
    304. int rotateSize = 0;
    305. };

    注:rotateSize成员是用来记录旋转次数的,若不记录可以删除。 


    7.AVL树与红黑树的性能比较

     如上图,这是在一千万数据下得到的测试结果。

    结论:红黑树和AVL树都是高效的平衡二叉树,增删改查的时间复杂度都是logN,红黑树不追求绝对平衡,其只需保证最长路径不超过最短路径的2倍,相对而言,红黑树降低了插入和旋转的次数,所以在经常进行增删的结构中性能比AVL树更优,而且红黑树实现比较简单,所以实际运用中红黑树更多(比如STL库中的set和map容器,linux内核等等)。而AVL树虽然在高度和搜索上优于红黑树,但基本可以忽略。


    =========================================================================

    如果你对该系列文章有兴趣的话,欢迎持续关注博主动态,博主会持续输出优质内容

    🍎博主很需要大家的支持,你的支持是我创作的不竭动力🍎

    🌟~ 点赞收藏+关注 ~🌟

    =========================================================================

  • 相关阅读:
    一万小时真的能成为专家吗?
    据说是中国电信的java编程面试题
    ChatGPT生产力|实用指令(prompt)
    (附源码)ssm高校志愿者服务系统 毕业设计 011648
    【毕业设计】基于php+mysql+smarttemplate的图片共享系统设计与实现(毕业论文+程序源码)——图片共享系统
    SQL Server 全角半角设置梳理 Chinese_PRC_CI_AS_WS
    分享一个使用三目优先级遇到的优先级问题
    平板摄像头+算力搞定3D空间实时重建和理解,清华和禾多科技新成果入选CVPR 2022 Oral...
    【最优化方法】实验四 约束最优化方法的MATLAB实现
    直销系统开发是找开发公司还是外包团队?
  • 原文地址:https://blog.csdn.net/2301_77112634/article/details/136523992