数据结构：红黑树讲解（C++）

红黑树

    • 1.前言
    • 2.红黑树简述
      • 2.1概念
      • 2.2性质
    • 3.红黑树的插入
      • 3.1关于新插入节点的颜色
      • 3.2节点的定义
      • 3.3插入新节点
      • 3.4判断插入后是否需要调整
      • 3.5插入后维持红黑树结构（重点）
        • 3.5.1cur、p、u为红，g为黑
        • 3.5.2cur、p为红，g为黑，u为空/u存在为黑
    • 4.一些简单的测试接口
    • 5.完整代码

1.前言

  • 本文旨在理解红黑树基本概念以及变色旋转规则，以理解C++map和set的底层原理，不会讲红黑树的删除操作。
  • 对于基本的旋转操作(单旋和双旋)，本文不会展开讲，详细讲解在这里：
    AVL树旋转讲解。



2.红黑树简述

2.1概念

红黑树，是一种二叉搜索树，但在每个结点上增加一个存储位表示结点的颜色，可以是Red或Black。 通过对任何一条从根到叶子的路径上各个结点着色方式的限制，红黑树确保最长路径不超过最短路径两倍，因而是接近平衡的。


2.2性质

  1. 每个节点不是红色就是黑色。
  2. 根部节点是黑色的。(为了减少旋转次数，后面讲旋转大家就明白了）
  3. 对于一个红节点，它的孩子只能是黑色。(即一条路径上不能出现连续的红色节点)
  4. 每条路径都必须包含相同数量的黑色节点。

通过上面规则的限制，红黑树最长路径一定不会超过最短路径两倍，也就维持了高度的相对平衡。
结合3、4来看下面的两条路径：
最长：黑、红、黑、红、黑、红…………
最短：黑、黑、黑…………



3.红黑树的插入

3.1关于新插入节点的颜色

对于新插入节点，我们设置为红色，原因是红黑树每条路径都必须包含相同数量的黑色节点（性质4），新插入红节点不一定破坏红黑树的结构，新插入黑色节点一定不符合性质4而且很难调整。
在这里插入图片描述


3.2节点的定义

//用枚举来定义颜色
enum Color
{RED,BLACK
};//这里直接实现key_value模型
template<class K, class V>
struct RBTreeNode
{RBTreeNode<K, V>* _left;RBTreeNode<K, V>* _right;RBTreeNode<K, V>* _parent;  //涉及到旋转，多加父亲指针来简化操作pair<K, V> _kv;  //存储键值对Color _col; //颜色RBTreeNode(const pair<K, V>& kv):_left(nullptr), _right(nullptr), _parent(nullptr), _kv(kv),_col(RED)  //新节点颜色为红色{}
};

3.3插入新节点

这里比较简单，按二叉搜索树的规则插入即可：

bool Insert(const pair<K, V>& kv)
{if (_root == nullptr){_root = new Node(kv);_root->_col = BLACK;return true;}Node* cur = _root;Node* parent = nullptr;while (cur){if (kv.first > cur->_kv.first) //待插入节点在右子树{parent = cur;cur = cur->_right;}else if (kv.first < cur->_kv.first)  //待插入节点在左子树{parent = cur;cur = cur->_left;}else  //相同{return false;}}cur = new Node(kv);if (kv.first > parent->_kv.first) //新节点在父亲右子树{parent->_right = cur;}else  //新节点在父亲左子树{parent->_left = cur;}cur->_parent = parent;  //记得更新父亲指针/// 变色旋转维持红黑树结构（暂时省略）  //_root->_col = BLACK; //可能改变根部颜色，保持根部为黑色return true;
}

3.4判断插入后是否需要调整

其实红黑树插入后只需要看当前节点和父亲的颜色即可，其中新节点一定为红。

  1. 父亲为黑，符合规则，不需要调整。
  2. 父亲为红，此时出现红红的连续节点，需要进行调整。

3.5插入后维持红黑树结构（重点）

为了方便叙述，我们做如下定义：

  1. cur表示当前节点
  2. p表示cur父亲节点
  3. u表示叔叔节点
  4. g表示祖父(p和u的父亲)节点
3.5.1cur、p、u为红，g为黑

在这里插入图片描述
代码：

while (parent && parent->_col == RED)  //父亲为红就调整，调整到根部要结束
{Node* granderfather = parent->_parent;  //祖父//需要对叔叔进行操作，需要判断叔叔是祖父的左还是右if (parent == granderfather->_left)  //父亲是祖父的左子树{Node* uncle = granderfather->_right;if (uncle && uncle->_col == RED) //叔叔不为空并且叔叔为红，变色即可{uncle->_col = parent->_col = BLACK;granderfather->_col = RED; //当前子树可能为部分，继续向上调整cur = granderfather;parent = cur->_parent;}else  //叔叔为空或为黑色{ //先省略}}else  //父亲是祖父的右子树{Node* uncle = granderfather->_left;if (uncle && uncle->_col == RED)  //叔叔不空并且为红{parent->_col = uncle->_col = BLACK;granderfather->_col = RED;  //当前可能为部分子树，需要继续上调cur = granderfather;parent = cur->_parent;}else  //叔叔为空或为黑色{// 先省略}}
}

3.5.2cur、p为红，g为黑，u为空/u存在为黑

下面是一会要用到的旋转接口：

void RotateL(Node* parent)  //左单旋,rotate->旋转
{Node* SubR = parent->_right;Node* SubRL = SubR->_left;  //这个有可能为空Node* ppnode = parent->_parent;  //原来父亲的父亲parent->_right = SubRL;if (SubRL)  SubRL->_parent = parent;SubR->_left = parent;parent->_parent = SubR;if (ppnode == nullptr)  //旋转的是整颗树{_root = SubR;SubR->_parent = nullptr;}else  //旋转的是部分{if (ppnode->_left == parent) //是左子树{ppnode->_left = SubR;}else  //是右子树{ppnode->_right = SubR;}SubR->_parent = ppnode;}
}void RotateR(Node* parent)  //右单旋细节处理和左单旋差不多
{Node* SubL = parent->_left;Node* SubLR = SubL->_right;  //这个有可能为空Node* ppnode = parent->_parent;parent->_left = SubLR;if (SubLR)  SubLR->_parent = parent;SubL->_right = parent;parent->_parent = SubL;if (ppnode == nullptr)  //旋转的是整颗树{_root = SubL;SubL->_parent = nullptr;}else  //旋转部分{if (ppnode->_left == parent)  //是左子树{ppnode->_left = SubL;}else  //右子树{ppnode->_right = SubL;}SubL->_parent = ppnode;}
}

涉及旋转情况比较复杂，分开讨论：

（1）p为g的左孩子，cur为p的左孩子
在这里插入图片描述


（2）p为g的左孩子，cur为p的右孩子

在这里插入图片描述


（3）p为g的右孩子，cur为p的右孩子

在这里插入图片描述


（4）p为g的右孩子，cur为p的左孩子

在这里插入图片描述

整合一下（1、2、3、4）得到下面的调整代码：

//到这里插入新节点的工作完成，下面进行结构调整：
while (parent && parent->_col == RED)  //父亲为红就调整，调整到根部要结束
{Node* granderfather = parent->_parent;  //祖父if (parent == granderfather->_left)  //父亲是祖父的左子树，p为g的左孩子{Node* uncle = granderfather->_right;if (uncle && uncle->_col == RED) //叔叔不为空并且叔叔为红，变色即可{uncle->_col = parent->_col = BLACK;granderfather->_col = RED; //当前子树可能为部分，继续向上调整cur = granderfather;parent = cur->_parent;}else  //叔叔为空或为黑色{ //     g//   p   u// cif (cur == parent->_left)  //当前为父亲的左子树，cur为p的左孩子{RotateR(granderfather);granderfather->_col = RED;parent->_col = BLACK;}else   //当前为父亲的右子树，cur为p的右孩子{//    g//  p   u//    c//左右双旋RotateL(parent);RotateR(granderfather);granderfather->_col = RED;cur->_col = BLACK;}break;  //这两种情况调整完可以结束}}else  //父亲是祖父的右子树，p为g的右孩子{Node* uncle = granderfather->_left;if (uncle && uncle->_col == RED)  //叔叔不空并且为红{parent->_col = uncle->_col = BLACK;granderfather->_col = RED;  //当前可能为部分子树，需要继续上调cur = granderfather;parent = cur->_parent;}else  //叔叔为空或为黑色{if (cur == parent->_right)  //当前为父亲的右，cur为p的右孩子{//    g//  u   p//        c//左旋RotateL(granderfather);parent->_col = BLACK;granderfather->_col = RED;}else  //当前为父亲的左，cur为p的左孩子{//   g// u   p//   c//右左双旋RotateR(parent);RotateL(granderfather);cur->_col = BLACK;granderfather->_col = RED;	}break;  //这两种情况调整完可以结束}}
}
_root->_col = BLACK; //保持根部为黑色



4.一些简单的测试接口

void InOrder()   //中序遍历，验证是否为二叉搜索树
{_InOrder(_root);cout << endl;
}void _InOrder(Node* root)
{if (root == nullptr)return;_InOrder(root->_left);cout << root->_kv.first << " ";_InOrder(root->_right);
}// 根节点->当前节点这条路径的黑色节点的数量
bool Check(Node* root, int blacknum, const int refVal)  
{if (root == nullptr)  //到根部看看当前路径黑色节点和标准值是否一致{//cout << balcknum << endl;if (blacknum != refVal){cout << "存在黑色节点数量不相等的路径" << endl;return false;}return true;}/检查子比较复杂，可以反过来去检查红节点父是否为黑色if (root->_col == RED && root->_parent->_col == RED)  {cout << "有连续的红色节点" << endl;return false;}if (root->_col == BLACK){++blacknum;  //为黑节点加一}return Check(root->_left, blacknum, refVal)&& Check(root->_right, blacknum, refVal);
}bool IsBalance()
{if (_root == nullptr)return true;if (_root->_col == RED)return false;//参考值，即先算出一条路径的黑色节点数int refVal = 0;Node* cur = _root;while (cur){if (cur->_col == BLACK){++refVal;}cur = cur->_left;}int blacknum = 0;return Check(_root, blacknum, refVal);
}



5.完整代码

#pragma once
#include <iostream>
#include <utility>
using namespace std;//用枚举来定义颜色
enum Color
{RED,BLACK
};//这里直接实现key_value模型
template<class K, class V>
struct RBTreeNode
{RBTreeNode<K, V>* _left;RBTreeNode<K, V>* _right;RBTreeNode<K, V>* _parent;  //涉及到旋转，多加父亲指针来简化操作pair<K, V> _kv;  //存储键值对Color _col; //颜色RBTreeNode(const pair<K, V>& kv):_left(nullptr), _right(nullptr), _parent(nullptr), _kv(kv),_col(RED)  //新节点颜色为红色{}
};template<class K, class V>
class RBTree
{
public:typedef RBTreeNode<K, V> Node;bool Insert(const pair<K, V>& kv){if (_root == nullptr){_root = new Node(kv);_root->_col = BLACK;return true;}Node* cur = _root;Node* parent = nullptr;while (cur){if (kv.first > cur->_kv.first) //待插入节点在右子树{parent = cur;cur = cur->_right;}else if (kv.first < cur->_kv.first)  //待插入节点在左子树{parent = cur;cur = cur->_left;}else  //相同{return false;}}cur = new Node(kv);if (kv.first > parent->_kv.first) //在右子树{parent->_right = cur;}else{parent->_left = cur;}cur->_parent = parent;while (parent && parent->_col == RED)  //父亲为红就调整，调整到根部要结束{Node* granderfather = parent->_parent;  //祖父if (parent == granderfather->_left)  //父亲是祖父的左子树{Node* uncle = granderfather->_right;if (uncle && uncle->_col == RED) //叔叔不为空并且叔叔为红，变色即可{uncle->_col = parent->_col = BLACK;granderfather->_col = RED; //当前子树可能为部分，继续向上调整cur = granderfather;parent = cur->_parent;}else  //叔叔为空或为黑色{ //     g//   p   u// cif (cur == parent->_left)  //当前为父亲的左子树{RotateR(granderfather);granderfather->_col = RED;parent->_col = BLACK;}else   //当前为父亲的右子树{//    g//  p   u//    c//左右双旋RotateL(parent);RotateR(granderfather);granderfather->_col = RED;cur->_col = BLACK;}break;}}else  //父亲是祖父的右子树{Node* uncle = granderfather->_left;if (uncle && uncle->_col == RED)  //叔叔不空并且为红{parent->_col = uncle->_col = BLACK;granderfather->_col = RED;  //当前可能为部分子树，需要继续上调cur = granderfather;parent = cur->_parent;}else  //叔叔为空或为黑色{if (cur == parent->_right)  //当前为父亲的右{//    g//  u   p//        c//左旋RotateL(granderfather);parent->_col = BLACK;granderfather->_col = RED;}else  //当前为父亲的左{//   g// u   p//   c//右左双旋RotateR(parent);RotateL(granderfather);cur->_col = BLACK;granderfather->_col = RED;	}break;}}}_root->_col = BLACK; //保持根部为黑色return true;}/// //
/// /
/// 	测试代码void InOrder()   //中序遍历，验证是否为二叉搜索树{_InOrder(_root);cout << endl;}void _InOrder(Node* root){if (root == nullptr)return;_InOrder(root->_left);cout << root->_kv.first << " ";_InOrder(root->_right);}// 根节点->当前节点这条路径的黑色节点的数量bool Check(Node* root, int blacknum, const int refVal)  {if (root == nullptr)  //到根部看看当前路径黑色节点和标准值是否一致{//cout << balcknum << endl;if (blacknum != refVal){cout << "存在黑色节点数量不相等的路径" << endl;return false;}return true;}/检查子比较复杂，可以反过来去检查红节点父是否为黑色if (root->_col == RED && root->_parent->_col == RED)  {cout << "有连续的红色节点" << endl;return false;}if (root->_col == BLACK){++blacknum;  //为黑节点加一}return Check(root->_left, blacknum, refVal)&& Check(root->_right, blacknum, refVal);}bool IsBalance(){if (_root == nullptr)return true;if (_root->_col == RED)return false;//参考值，即先算出一条路径的黑色节点数int refVal = 0;Node* cur = _root;while (cur){if (cur->_col == BLACK){++refVal;}cur = cur->_left;}int blacknum = 0;return Check(_root, blacknum, refVal);}int Height(){return _Height(_root);}int _Height(Node* root)  //求高度的{if (root == nullptr)return 0;int leftHeight = _Height(root->_left);int rightHeight = _Height(root->_right);return leftHeight > rightHeight ? leftHeight + 1 : rightHeight + 1;}Node* Find(K key){return _Find(key, _root);}Node* _Find(K key, Node* root){if (root == nullptr)return nullptr;if (key > root->_kv.first) //在右子树{return _Find(key, root->_right);}else if (key < root->_kv.first) //在左子树{return _Find(key, root->_left);}else  //找到了{return root;}}private:Node* _root = nullptr;void RotateL(Node* parent)  //左单旋,rotate->旋转{Node* SubR = parent->_right;Node* SubRL = SubR->_left;  //这个有可能为空Node* ppnode = parent->_parent;  //原来父亲的父亲parent->_right = SubRL;if (SubRL)  SubRL->_parent = parent;SubR->_left = parent;parent->_parent = SubR;if (ppnode == nullptr)  //旋转的是整颗树{_root = SubR;SubR->_parent = nullptr;}else  //旋转的是部分{if (ppnode->_left == parent) //是左子树{ppnode->_left = SubR;}else  //是右子树{ppnode->_right = SubR;}SubR->_parent = ppnode;}}void RotateR(Node* parent)  //右单旋细节处理和左单旋差不多{Node* SubL = parent->_left;Node* SubLR = SubL->_right;  //这个有可能为空Node* ppnode = parent->_parent;parent->_left = SubLR;if (SubLR)  SubLR->_parent = parent;SubL->_right = parent;parent->_parent = SubL;if (ppnode == nullptr)  //旋转的是整颗树{_root = SubL;SubL->_parent = nullptr;}else  //旋转部分{if (ppnode->_left == parent)  //是左子树{ppnode->_left = SubL;}else  //右子树{ppnode->_right = SubL;}SubL->_parent = ppnode;}}
};

本文来自互联网用户投稿，该文观点仅代表作者本人，不代表本站立场。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如若转载，请注明出处：http://www.hqwc.cn/news/195131.html

如若内容造成侵权/违法违规/事实不符，请联系编程知识网进行投诉反馈email:809451989@qq.com，一经查实，立即删除！

相关文章

【设计模式】聊聊模板模式

【设计模式】聊聊模板模式

原理和实现 设计模式的原理和实现是比较简单的&#xff0c;难的是掌握具体的应用场景和解决什么问题。而模板模式是为来解决复用和拓展两个问题。 模板模式在一个方法中定义好一个算法框架&#xff0c;然后将某些步骤推迟到子类中实现&#xff0c;子类可以在不修改父类流程的时…
阅读更多...
单片机实验（二）

单片机实验（二）

前言 实验一&#xff1a;用AT89C51单片机控制LCD 1602&#xff0c;使其显示两行文字&#xff0c;分别显示自己的学号和姓名拼音。 实验二&#xff1a;设计一个中断嵌套程序。要求K1和K2都未按下时&#xff0c;单片机控制8只数码管&#xff0c;滚动输出完整的学号。当按一下K1…
阅读更多...
UE 视差材质 学习笔记

UE 视差材质 学习笔记

视差材质节点&#xff1a; 第一个是高度图&#xff0c; Heightmap Channel就是高度图的灰色通道&#xff0c;在RGBA哪个上面&#xff0c;例如在R上就连接(1,0,0,0)&#xff0c;G上就连接&#xff08;0,1,0,0&#xff09;逐次类推 去看看对比效果&#xff1a; 这个是有视差效果…
阅读更多...
对vb.net 打印条形码code39、code128A、code128C、code128Auto（picturebox和打印机）封装类一文的补充

对vb.net 打印条形码code39、code128A、code128C、code128Auto（picturebox和打印机）封装类一文的补充

在【精选】vb.net 打印条形码code39、code128A、code128C、code128Auto&#xff08;picturebox和打印机&#xff09;封装类_vb.net打印标签_WormJan的博客-CSDN博客 这篇文章中&#xff0c;没有对含有字母的编码进行处理。这里另开一篇帖子&#xff0c;处理这种情况。 在那篇文…
阅读更多...
汽车虚拟仿真视频数据理解--CLIP模型原理

汽车虚拟仿真视频数据理解--CLIP模型原理

CLIP模型原理 CLIP的全称是Contrastive Language-Image Pre-Training&#xff0c;中文是对比语言-图像预训练&#xff0c;是一个预训练模型&#xff0c;简称为CLIP。该模型是 OpenAI 在 2021 年发布的&#xff0c;最初用于匹配图像和文本的预训练神经网络模型&#xff0c;这个任…
阅读更多...
基于Vue+SpringBoot的超市账单管理系统 开源项目

基于Vue+SpringBoot的超市账单管理系统 开源项目

项目编号&#xff1a; S 032 &#xff0c;文末获取源码。 \color{red}{项目编号&#xff1a;S032&#xff0c;文末获取源码。} 项目编号&#xff1a;S032&#xff0c;文末获取源码。 目录 一、摘要1.1 项目介绍1.2 项目录屏 二、功能模块三、系统设计3.1 总体设计3.2 前端设计3…
阅读更多...
三天吃透Redis面试八股文

三天吃透Redis面试八股文

目录&#xff1a; Redis是什么&#xff1f;Redis优缺点&#xff1f;Redis为什么这么快&#xff1f;讲讲Redis的线程模型&#xff1f;Redis应用场景有哪些&#xff1f;Memcached和Redis的区别&#xff1f;为什么要用 Redis 而不用 map/guava 做缓存?Redis 数据类型有哪些&…
阅读更多...
qsort使用举例和qsort函数的模拟实现

qsort使用举例和qsort函数的模拟实现

qsort使用举例 qsort是C语言中的一个标准库函数&#xff0c;用于对数组或者其他数据结构中的元素进行排序。它的原型如下&#xff1a; void qsort(void *base, size_t nmemb, size_t size, int (*compar)(const void *, const void *)); 我们可以去官网搜来看一看&#xff1a;…
阅读更多...
YOLOv8 加持 MobileNetv3，目标检测新篇章

YOLOv8 加持 MobileNetv3，目标检测新篇章

🗝️YOLOv8实战宝典--星级指南：从入门到精通，您不可错过的技巧   -- 聚焦于YOLO的 最新版本， 对颈部网络改进、添加局部注意力、增加检测头部，实测涨点 💡 深入浅出YOLOv8：我的专业笔记与技术总结   -- YOLOv8轻松上手， 适用技术小白，文章代码齐全，仅需 …
阅读更多...
golang中的并发模型

golang中的并发模型

并发模型 传统的编程语言&#xff08;如C、Java、Python等&#xff09;并非为并发而生的&#xff0c;因此它们面对并发的逻辑多是基于操作系统的线程。其并发的执行单元&#xff08;线程&#xff09;之间的通信利用的也是操作系统提供的线程或进程间通信的原语&#xff0c;比如…
阅读更多...
闭眼检测实现

闭眼检测实现

引言 这段代码是一个实时眼睛状态监测程序&#xff0c;可以用于监测摄像头捕获的人脸图像中的眼睛状态&#xff0c;判断眼睛是否闭合。具体应用实现作用说明如下&#xff1a; 1. 实时监测眼睛状态 通过摄像头捕获的实时视频流&#xff0c;检测人脸关键点并计算眼睛的 EAR&a…
阅读更多...
基于灰狼算法(GWO)优化的VMD参数（GWO-VMD）

基于灰狼算法(GWO)优化的VMD参数（GWO-VMD）

代码的使用说明 基于灰狼算法优化的VMD参数 代码的原理 基于灰狼算法&#xff08;Grey Wolf Optimizer, GWO&#xff09;优化的VMD参数&#xff08;GWO-VMD&#xff09;是一种结合了GWO和VMD算法的优化方法&#xff0c;用于信号分解和特征提取。 GWO是一种基于群体智能的优化…
阅读更多...
推荐文章
最新文章

Copyright @ 2022~2023