数据结构:红黑树讲解(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++mapset的底层原理,不会讲红黑树的删除操作。
  • 对于基本的旋转操作(单旋和双旋),本文不会展开讲,详细讲解在这里:
    AVL树旋转讲解。



2.红黑树简述

2.1概念

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


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.mzph.cn/news/148293.shtml

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

相关文章

【【VDMA彩条显示实验之三 之 RGB LCD 彩条显示实验 】】

VDMA彩条显示实验之三 之 RGB LCD 彩条显示实验 VDMA彩条显示实验之三 之 RGB LCD 彩条显示实验 LCD 的构造是在两片平行的玻璃基板当中放置液晶盒&#xff0c;下基板玻璃上设置 TFT&#xff08;薄膜晶体管&#xff09;&#xff0c;上基板玻璃上设置彩色滤光片&#xff0c;通…

Flutter 应用启动从闪屏页短暂黑屏再到第一个页面

由于应用初始状态启动会有白屏现象&#xff0c;便使用 flutter_native_splash 2.3.5 插件生成了启动相关的配置&#xff0c;并且按照示例使用了 import package:flutter_native_splash/flutter_native_splash.dart;void main() {WidgetsBinding widgetsBinding WidgetsFlutte…

如何进行跨平台开发和移植性处理?

跨平台开发和移植性设计是在不同操作系统和硬件平台上开发软件的关键考虑因素。这通常涉及使用如C或C这样的通用编程语言&#xff0c;以及利用跨平台库和工具来处理平台特定的差异。 在具体实施时&#xff0c;需要考虑以下几点&#xff1a; 代码的可移植性&#xff1a;编写代码…

牛客 —— 链表中倒数第k个结点(C语言,快慢指针,配图)

目录 1. 思路1&#xff1a;倒数第K个节点&#xff0c;就是整数第N-K1的节点 2. 思路2&#xff1a;快慢指针 1. 思路1&#xff1a;倒数第K个节点&#xff0c;就是整数第N-K1的节点 链表中&#xff0c;一共有N个节点&#xff0c;如果我们想要得出倒数第K个节点&#xff0c;我们…

图像倾斜角度求取-Radon变换

Radon算法 Radon&#xff08;拉东&#xff09;算法是一种通过定方向投影叠加&#xff0c;找到最大投影值时角度&#xff0c;从而确定图像倾斜角度的算法。具体过程如图所示 图1 Radon变换算法 Radon计算示例 对于纹理方向明显的图像&#xff0c;如图2所示&#xff0c;可以通…

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

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

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

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

golang学习笔记——生成随机整数的函数Int31n

介绍 rand.31n()函数是Golang标准库中用来生成介于0到n之间的随机整数的函数。其中n是指定的范围&#xff0c;可以是任意正整数。该函数的具体实现方式是在go语言内部使用C语言的随机数生成函数来生成随机数。 具体来说&#xff0c;rand.Int31n()函数的参数n表示要生成的随机…

Ubuntu20.04安装Beyond Compare 4.4.7

参考链接&#xff1a; 1.Ubuntu20.04 Beyond Compare 4.3.7 安装 2.Ubuntu20.04安装Beyond Compare 4.3.7

MySQL 定时计划任务 事件的使用

目录 查看事件是否开启 开启事件 1&#xff09;通过设置全局参数修改 2&#xff09;更改配置文件 MySQL如何创建并执行事件&#xff1f; 例 1 MySQL查看事件状态信息 MySQL修改和删除事件 例 1 例 2 删除事件 例 3 在数据库管理中&#xff0c;经常要周期性的执行某…

lxml基本使用

lxml是python的一个解析库&#xff0c;支持HTML和XML的解析&#xff0c;支持XPath解析方式&#xff0c;而且解析效率非常高 XPath&#xff0c;全称XML Path Language&#xff0c;即XML路径语言&#xff0c;它是一门在XML文档中查找信息的语言&#xff0c;它最初是用来搜寻XML文…

【C/PTA】数组进阶练习(二)

本文结合PTA专项练习带领读者掌握数组&#xff0c;刷题为主注释为辅&#xff0c;在代码中理解思路&#xff0c;其它不做过多叙述。 目录 7-1 字符串逆序7-2 字符串替换7-3 统计字符出现次数7-4 IP地址转换7-1 删除重复字符7-2 说反话-加强版7-3 数组-回文串7-4 数组-无聊的菇菇…

2024全网最新最全的Pytest接口自动化测试框架教程

pytest编写的规则&#xff1a; 1、测试文件以test_开头&#xff08;以_test结尾也可以&#xff09; 2、测试类以Test开头&#xff0c;并且不能带有__init__方法 3、测试函数以test_开头 4、断言必须使用assert pytest.main([-s,-v]) &#xff1a;用来执行测试用例 -s 打印prin…

卷积神经网络(CNN)天气识别

文章目录 前期工作1. 设置GPU&#xff08;如果使用的是CPU可以忽略这步&#xff09;我的环境&#xff1a; 2. 导入数据3. 查看数据 二、数据预处理1. 加载数据2. 可视化数据3. 再次检查数据4. 配置数据集 三、构建CNN网络四、编译五、训练模型六、模型评估 前期工作 1. 设置GP…

11.10 校招 实习 内推 面经

绿*泡*泡&#xff1a; neituijunsir 交流裙 &#xff0c;内推/实习/校招汇总表格 1、校招丨海康机器人2024校招3大类岗位补录&#xff08;内推&#xff09; 校招丨海康机器人2024校招3大类岗位补录&#xff08;内推&#xff09; 2、校招&实习 | 华为数据存储研究部招聘…

SELinux零知识学习十九、SELinux策略语言之类型强制(4)

接前一篇文章&#xff1a;SELinux零知识学习十八、SELinux策略语言之类型强制&#xff08;3&#xff09; 二、SELinux策略语言之类型强制 2. 类型、属性和别名 &#xff08;4&#xff09;别名 别名是引用类型时的一个备选的名字&#xff0c;能够使用类型名的地方就可以使用别…

EEPROM与Flash的区别

EEPROM与Flash的区别 EEPROMEEPROM内部功能框图实现写入数据内部结构存储管在充电或放电状态下有着不同的阈值电压 问题点EEPROM是如何失效的呢&#xff1f;为何EEPROM不能做大呢&#xff1f; ------------------------------------------------------------------------------…

HTTP请求详解

HTTP请求格式 请求报文通常包含以下部分: 请求行(Request Line): 包括请求方法、请求的URL和协议版本。 示例:GET /index.html HTTP/1.1 请求头(Request Headers): 包含了一系列的键值对,用来描述客户端请求的相关信息,比如Accept(告诉服务器客户端能够处理的MIME类型…

Java多线程(3)

Java多线程(3) 深入剖析Java线程的生命周期&#xff0c;探秘JVM的线程状态&#xff01; 线程的生命周期 Java 线程的生命周期主要包括五个阶段&#xff1a;新建、就绪、运行、阻塞和销毁。 **新建&#xff08;New&#xff09;&#xff1a;**线程对象通过 new 关键字创建&…

tamarin运行

首先我们找到安装tamarin的文件位置&#xff0c;找到以后进入该文件夹下 ubuntuubuntu:~$ sudo find / -name tamarin-prover /home/linuxbrew/.linuxbrew/var/homebrew/linked/tamarin-prover /home/linuxbrew/.linuxbrew/Cellar/tamarin-prover /home/linuxbrew/.linuxbrew/…