红黑树介绍与模拟实现（insert+颜色调整精美图示超详解哦）

2024-04-04 21:28
文章标签 实现 详解 介绍 模拟 红黑树 调整 insert 颜色 图示 精美

本文主要是介绍红黑树介绍与模拟实现（insert+颜色调整精美图示超详解哦），希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

红黑树

  • 引言
  • 红黑树的介绍
  • 实现
    • 结点类
    • insert
      • 搜索插入位置
      • 插入
      • 调整
        • 当parent为gparent的左子结点
        • 当parent为gparent的右子结点
  • 参考源码
  • 测试红黑树是否合格
  • 总结

引言

在上一篇文章中我们认识了高度平衡的平衡二叉树AVL树：戳我看AVL树详解哦
（关于旋转调整的部分，在AVL树的时候已经详细介绍过了，如果大家对旋转调平衡的部分有疑惑的话，请移步至AVL树的详解）

由于AVL树的高度平衡，其平均搜索时间复杂度几乎可以达到严格的O(logN)。同样也因为平衡的程度很高，在维护平衡上时间的花费对于搜索上时间的提升是得不偿失的。
大多数情况下，我们并不需要很高的平衡程度，只需要达成一种接近平衡的状态，搜索的平均时间复杂度基本达到O(logN)即可。 红黑树就是这样的一种结构，它的最高子树的高度小于等于最低子树的二倍。通过减少调平衡时的时间成本来提高效率：

红黑树的介绍

红黑树是平衡二叉树的一种，他在满足二叉搜索树特性的基础上，给每个结点增加了一个颜色属性，包括Red与Black；并要求从根节点到一个叶子结点形成的任意一条路径中，通过对结点颜色的限制规则，没有任何一条路径回比其他的路径长一倍：
在这里插入图片描述
（红黑圣诞树）

对于红黑树结点颜色的限制规则如下：

  • 每个结点的颜色只有红色或黑色两种；
  • 根结点的颜色一定是黑色的；
  • 如果某一个结点的颜色是红色的，它的两个孩子结点的颜色一定是黑色（即不存在两个连续的红色结点）；
  • 对于任一结点，到叶子结点的任一路径上，包含的黑色结点的数量一定相等；
  • 叶子结点一定是黑色的（这里的叶子结点直最后的nullptr）

当满足上面的所有规则时，根节点到叶子结点的任一路径就都不可能比其他路径长一倍。由于不存在连续的红色结点，所以当黑色结点的数量 n 一定时，最长路径的长度为2 * n，最短路径的长度为 n ，所以不可能相差一倍以上。这样就达成了一种相对平衡的状态，并不需要经常去旋转调平了。

实现

红黑树的实现，在之前的二叉搜索树上增加了结点的颜色，以及对于结点的颜色调整的部分；
在本篇文章中依旧实现 K-V的模式 的树， 并且以非递归实现insert；
为防止命名冲突，将实现放在我们的命名空间qqq中：

结点的颜色我们使用 枚举常量enum Color 来表示；

RBTree是一个类模板，有两个模板参数，即K与V，表示其中存储的索引类型与值类型；
成员变量类型为Node*（由结点类RBTreeNode重命名），表示根结点的指针：

 //基本的代码结构
namespace qqq
{enum Color //枚举常量表示颜色{RED,BLUCK};template<class K, class V>struct RBTreeNode  //结点类{};template<class K, class V>class RBTree  //红黑树{typedef RBTreeNode<K, V> Node;public:bool insert(const pair<K, V>& kv){}protected:Node* _root = nullptr;};
}

结点类

首先，对红黑树的结点进行实现：

RBTreeNode是也一个类模板， 两个模板参数同样为K与V，表示索引与值的类型；
在结点中储存数据的结构为pair，其中first为K类型，second为V类型；

成员变量包括结点中的数据 _kv；
指向父亲结点的指针 _parent；
指向左右孩子结点的指针 _left 与 _right；
表示结点颜色的枚举常量 _col：

template<class K, class V>
struct RBTreeNode  //结点类
{pair<K, V> _kv;RBTreeNode<K, V>* _parent;RBTreeNode<K, V>* _left;RBTreeNode<K, V>* _right;Color _col;
};

结点类的构造函数
我们需要实现一个默认构造函数：
参数类型为const pair<K, V>，缺省值为一个pair<K, V>的匿名对象；
对于父子结点的指针，在初始化列表中初始化为nullptr即可；
对于结点的颜色，在初始化列表中初始化为RED（要保证每条路上黑色结点的个数相等，就必须初始化为红色）：

	RBTreeNode(const pair<K, V> kv = pair<K, V>()): _kv(kv), _left(nullptr), _right(nullptr), _parent(nullptr), _col(RED){}

insert

红黑树的insert分为两个部分，即搜索并插入以及调整使其满足红黑树的性质：

搜索插入位置

  • 与二叉搜索树类似，搜索并插入时，首先用要插入的pair对象创建一个新结点newnode；
  • 与此同时，使用结点指针 cur来记录当前位置 ，parent来记录cur的父结点，便于后面插入；
  • 然后 while循环向下查找 插入的位置：当newnode小于cur的元素时，向左查找，当newnode大于cur的元素时，向右查找，相等时即该元素已经存在，返回false；
  • 当cur为nullptr时，表示找到了插入的位置，循环终止：
// 部分代码：搜索插入位置 //
Node* newnode = new Node(kv);
Node* parent = nullptr;
Node* cur = _root;while (cur != nullptr) //搜索
{if (newnode->_kv.first > cur->_kv.first){parent = cur;cur = parent->_right;}else if (newnode->_kv.first < cur->_kv.first){parent = cur;cur = parent->_left;}else  //相等即插入失败{return false;}
}

插入

  • 将newnode插入红黑树即将newnode与parent链接，这时就需要判断parent是否为空；
  • 当parent为空时，即cur就是根节点_root，即newnode是这棵红黑树中的第一个结点，将 其赋值给_root 即可；
  • 当parent不为空时， 还需要判断cur位于parent的左边还是右边，然后再插入：

在这里插入图片描述

//  局部代码：插入newnode //
if (parent == nullptr) //插入
{_root = newnode;
}
else if (newnode->_kv.first < parent->_kv.first)
{parent->_left = newnode;newnode->_parent = parent;cur = newnode;
}
else
{parent->_right = newnode;newnode->_parent = parent;cur = newnode;
}

调整

在完成插入后，首先需要判断的是parent指针的状态：

  • 当parent为空时，表明当前结点为根结点，将其颜色改为BLACK即可调整完毕；
  • 当parent指向结点的颜色为BLACK 时，如果是在刚插入时，新插入的结点颜色为RED，不会影响该路径的黑色节点个数，所以不再需要调整。如果是在向上调整的过程中parent指向的结点为BLACK，也意味着整棵树调整结束了（这一点在后面的调整部分会详细介绍）；
  • 除了上面不用调整的两种情况外，其余的情况，即 parent指向的结点存在且为RED 的情况就需要进行调整了。调整是自下而上的，循环向上调整，直到 parent为空或parent指向的结点为BLACK时循环结束，调整完成
// 部分代码：自下而上调整的循环框架 //
while (parent != nullptr && parent->_col != BLUCK) 
{}

开始调整时首先对parent为gparent（cur的祖父结点）的左子结点还是右子结点做一分类讨论（当parent不为黑时，由于根结点必须为黑，所以parent不是根结点，所以gparent一定存在）：

当parent为gparent的左子结点

当parent为gparent的左子结点时，我们首先要考察cur叔叔结点的情况：

  • 当cur的叔叔结点为RED 时，不需要进行旋转调整，只需要将gparent指向结点设置为RED，将parent和cur的叔叔结点全部设置为BLACK即可：
    由于在调整完颜色后，gparent指向结点颜色就为RED了，这时如果gparent父结点的颜色正好为红，就出现了连续的两个红色结点。所以需要将cur向上移动两个结点，再对其parent指向的结点继续进行判断：
    在这里插入图片描述

  • 当cur的叔叔结点为BLACK或不存在时，就需要进行旋转调整了：
    旋转的逻辑与之前AVL树类似，当cur位于parent的左边时，即左左——单次右旋：
    在这里插入图片描述
    当cur位于parent的右边时，即左右——左右双旋：
    在这里插入图片描述
    在这里插入图片描述
    旋转调整后将gparent指向结点设置为RED，将子树顶部的结点设置为BLACK 即可（parent指向的结点或cur指向的结点）：
    在这里插入图片描述

经过旋转后的红黑树，子树顶部的颜色一定为黑色，再向上也就不会存在两个连续的红色结点的问题了，所以旋转之后直接终止循环即可。

需要注意的是：叔叔是黑色结点的情况一定是出现在调整过程中发生的，当叔叔结点为黑色时，cur下的路径中一定存在着与gparent右子树中黑色结点相同个数的黑色结点，parent下的路径中同样也存在着相同数目的黑色结点，这样在旋转调平衡后，这棵子树中的所有路径中的黑色结点数目与之前是不变的。

//部分代码：当parent为gparent左子结点的情况
if (parent == gparent->_left)   
{if (gparent->_right == nullptr || gparent->_right->_col == BLUCK)  //1.叔叔结点不存在或为黑，需要旋转并调色{if (cur == parent->_left)//左左->单次右旋{RotateR(gparent);parent->_col = BLUCK;gparent->_col = RED;}else//左右->左旋+右旋{RotateL(parent);RotateR(gparent);cur->_col = BLUCK;gparent->_col = RED;}break; //通过旋转调整后，该子树的根结点一定是黑，所以可以直接结束循环}else if (gparent->_right->_col == RED)  //2.叔叔结点为红，通过调色即可实现红黑树{//调色parent->_col = BLUCK;gparent->_right->_col = BLUCK;gparent->_col = RED;//继续向上cur = gparent;parent = cur->_parent;}else{assert(0);}
}
当parent为gparent的右子结点

当parent为gparent的右子结点时，与上面的情况一致，只是左右对调了，所以这里只给出图示与代码（如果在这种情况下遇到了问题，希望你在上面的情况中能够找到答案）：

  • 叔叔结点为RED，仅调整颜色：
    在这里插入图片描述

  • 叔叔结点为BLACK或不存在，左单旋或右左双旋:
    左单旋：
    在这里插入图片描述
    右左双旋：
    在这里插入图片描述在这里插入图片描述
    旋转后调整颜色：
    在这里插入图片描述

旋转后子树顶部的结点一定为BLACK，所以直接break即可。

//部分代码：当parent为gparent右子结点的情况 //
else //parent == gparent->_right
{if (gparent->_left == nullptr || gparent->_left->_col == BLUCK)  //1.叔叔结点不存在或为黑，需要旋转并调色{if (cur == parent->_right)//右右->单次左旋{RotateL(gparent);parent->_col = BLUCK;gparent->_col = RED;}else//右左->右旋+左旋{RotateR(parent);RotateL(gparent);cur->_col = BLUCK;gparent->_col = RED;}break; //通过旋转调整后，该子树的根结点一定是黑，所以可以直接结束循环}else if (gparent->_left->_col == RED)  //2.叔叔结点为红，通过调色即可实现红黑树{//调色parent->_col = BLUCK;gparent->_left->_col = BLUCK;gparent->_col = RED;//继续向上cur = gparent;parent = cur->_parent;}else{assert(0);}
}

在while循环调整结束之后，再将根结点_root的颜色改为BLACK，统一做处理（insert的整体代码在这里就不做展示了，大家跳转至参考源码部分查看即可）。

参考源码

namespace qqq
{enum Color //枚举常量表示颜色{RED,BLUCK};template<class K, class V>struct RBTreeNode  //结点类{pair<K, V> _kv;RBTreeNode<K, V>* _parent;RBTreeNode<K, V>* _left;RBTreeNode<K, V>* _right;Color _col;RBTreeNode(const pair<K, V> kv = pair<K, V>()): _kv(kv), _left(nullptr), _right(nullptr), _parent(nullptr), _col(RED){}};template<class K, class V>class RBTree  //红黑树{typedef RBTreeNode<K, V> Node;public:bool insert(const pair<K, V>& kv){ //先插入Node* newnode = new Node(kv);Node* parent = nullptr;Node* cur = _root;while (cur != nullptr) //搜索{if (newnode->_kv.first > cur->_kv.first){parent = cur;cur = parent->_right;}else if (newnode->_kv.first < cur->_kv.first){parent = cur;cur = parent->_left;}else  //相等即插入失败{return false;}}if (parent == nullptr) //插入{_root = newnode;}else if (newnode->_kv.first < parent->_kv.first){parent->_left = newnode;newnode->_parent = parent;cur = newnode;}else{parent->_right = newnode;newnode->_parent = parent;cur = newnode;}//调整颜色以及旋转使满足红黑树while (parent != nullptr && parent->_col != BLUCK) //当parent不为黑时，由于根结点必须为黑，所以parent不是根结点，所以gparent一定存在{Node* gparent = parent->_parent;if (parent == gparent->_left)   {if (gparent->_right == nullptr || gparent->_right->_col == BLUCK)  //1.叔叔结点不存在或为黑，需要旋转并调色{if (cur == parent->_left)//左左->单次右旋{RotateR(gparent);parent->_col = BLUCK;gparent->_col = RED;}else//左右->左旋+右旋{RotateL(parent);RotateR(gparent);cur->_col = BLUCK;gparent->_col = RED;}break; //通过旋转调整后，该子树的根结点一定是黑，所以可以直接结束循环}else if (gparent->_right->_col == RED)  //2.叔叔结点为红，通过调色即可实现红黑树{//调色parent->_col = BLUCK;gparent->_right->_col = BLUCK;gparent->_col = RED;//继续向上cur = gparent;parent = cur->_parent;}else{assert(0);}}else{if (gparent->_left == nullptr || gparent->_left->_col == BLUCK)  //1.叔叔结点不存在或为黑，需要旋转并调色{if (cur == parent->_right)//右右->单次左旋{RotateL(gparent);parent->_col = BLUCK;gparent->_col = RED;}else//右左->右旋+左旋{RotateR(parent);RotateL(gparent);cur->_col = BLUCK;gparent->_col = RED;}break; //通过旋转调整后，该子树的根结点一定是黑，所以可以直接结束循环}else if (gparent->_left->_col == RED)  //2.叔叔结点为红，通过调色即可实现红黑树{//调色parent->_col = BLUCK;gparent->_left->_col = BLUCK;gparent->_col = RED;//继续向上cur = gparent;parent = cur->_parent;}else{assert(0);}}}_root->_col = BLUCK;return true;}void RotateL(Node* parent){Node* gparent = parent->_parent;Node* cur = parent->_right;Node* curleft = cur->_left;parent->_right = curleft;if (curleft != nullptr){curleft->_parent = parent;}cur->_left = parent;parent->_parent = cur;cur->_parent = gparent;if (gparent == nullptr){_root = cur;}else{if (cur->_kv.first < gparent->_kv.first){gparent->_left = cur;}else{gparent->_right = cur;}}}void RotateR(Node* parent){Node* gparent = parent->_parent;Node* cur = parent->_left;Node* curright = cur->_right;parent->_left = curright;if (curright != nullptr){curright->_parent = parent;}cur->_right = parent;parent->_parent = cur;cur->_parent = gparent;if (gparent == nullptr){_root = cur;}else{if (cur->_kv.first < gparent->_kv.first){gparent->_left = cur;}else{gparent->_right = cur;}}}bool isRBTree(){return isRBTree(_root);}protected:Node* _root = nullptr;bool checkColor(Node* root, int countBluck, int& baseBluck) //计算黑结点数量与红结点连续是否满足条件{if (root == nullptr){if (baseBluck == countBluck){return true;}return false;}if (root->_col == RED && root->_parent != nullptr && root->_parent->_col == RED){return false;}if (root->_col == BLUCK){countBluck++;}return checkColor(root->_left, countBluck, baseBluck) && checkColor(root->_right, countBluck, baseBluck);}bool isRBTree(Node* root){if (root == nullptr)return true;if (root->_col == RED)return false;//先计算一条路径中的黑色结点数量int baseBluck = 0;Node* cur = root;while (cur != nullptr){if (cur->_col == BLUCK){++baseBluck;}cur = cur->_right;}return checkColor(root, 0, baseBluck);}};
}

测试红黑树是否合格

在写完红黑树的insert之后，我们可以再编写一个测试模块来测试一棵树是否满足红黑树的特性：

我们其实只需要判断两点即可：

  • 任一路径上的黑色结点个数是否相等；
  • 是否存在连续的两个红色结点。

我们使用递归的方式来判断：

判断任一路径上的黑色结点个数是否相等时，必须要先计算某条路径上的黑色结点个数，可以通过 while循环计算最右侧一条路径上的黑色结点个数。将cur从_root开始一直向右子节点遍历即可，当cur为空时即该路径结束，终止循环。

然后使用checkColor函数（参数为当前结点指针Node*，计算中的当前路径黑色结点个数int，以及先前计算的最外层黑色结点个数int ）递归计算每条路径上的黑色结点个数，顺便判断是否存在连续的红色结点。
在递归过程中：
当_root为空时，即当前路径已经结束，判断countBluck与baseBlack的值是否相等，若相等返回true，否则返回false；
若当前结点为红色，则判断其父结点是否为红色，是就返回false；
若当前结点为黑色，countBluck加1，并继续向左右子结点递归，返回左子结点的结果&&右子结点的结果:
在这里插入图片描述

测试红黑树是否合格的代码这里就不赘述了，大家可以在参考源码部分查找。这里来展示一下测试结果：

namespace qqq
{void testfunc(){const int N = 10000000;vector<int> v;v.reserve(N);srand(time(0));for (size_t i = 0; i < N; i++){v.push_back(i);}RBTree<int, int> rbt;for (auto e : v){rbt.insert(make_pair(e, e));cout << "insert:" << e << "->" << rbt.isRBTree() << endl;}cout << rbt.isRBTree() << endl;}
}int main()
{qqq::testfunc();return 0;
}

在这里插入图片描述
因为这段测试代码中，存在大量I/O，所以运行速度很慢，大家可以将cout注释掉，只打印最后的结果。

总结

到此，关于红黑树的知识就介绍完了
在接下来的文章中，将会对红黑树进行封装，即map与set，尽情期待哦

如果大家认为我对某一部分没有介绍清楚或者某一部分出了问题，欢迎大家在评论区提出

如果本文对你有帮助，希望一键三连哦

希望与大家共同进步哦

这篇关于红黑树介绍与模拟实现（insert+颜色调整精美图示超详解哦）的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！


http://www.chinasem.cn/article/876864。

相关文章

C#借助Spire.XLS for .NET实现在Excel中添加文档属性

C#借助Spire.XLS for .NET实现在Excel中添加文档属性

《C#借助Spire.XLSfor.NET实现在Excel中添加文档属性》在日常的数据处理和项目管理中,Excel文档扮演着举足轻重的角色,本文将深入探讨如何在C#中借助强大的第三方库Spire.... 目录为什么需要程序化添加Excel文档属性使用Spire.XLS for .NET库实现文档属性管理Sp
阅读更多...
C++ move 的作用详解及陷阱最佳实践

C++ move 的作用详解及陷阱最佳实践

《C++move的作用详解及陷阱最佳实践》文章详细介绍了C++中的`std::move`函数的作用,包括为什么需要它、它的本质、典型使用场景、以及一些常见陷阱和最佳实践,感兴趣的朋友跟随小编一起看... 目录C++ move 的作用详解一、一句话总结二、为什么需要 move？C++98/03 的痛点⚡C++
阅读更多...
Python+FFmpeg实现视频自动化处理的完整指南

Python+FFmpeg实现视频自动化处理的完整指南

《Python+FFmpeg实现视频自动化处理的完整指南》本文总结了一套在Python中使用subprocess.run调用FFmpeg进行视频自动化处理的解决方案,涵盖了跨平台硬件加速、中间素材处理... 目录一、 跨平台硬件加速：统一接口设计1. 核心映射逻辑2. python 实现代码二、 中间素材处
阅读更多...
MySQL中between and的基本用法、范围查询示例详解

MySQL中between and的基本用法、范围查询示例详解

《MySQL中betweenand的基本用法、范围查询示例详解》BETWEENAND操作符在MySQL中用于选择在两个值之间的数据,包括边界值,它支持数值和日期类型,示例展示了如何使用BETWEEN... 目录一、between and语法二、使用示例2.1、betwphpeen and数值查询2.2、be
阅读更多...
python中的flask_sqlalchemy的使用及示例详解

python中的flask_sqlalchemy的使用及示例详解

《python中的flask_sqlalchemy的使用及示例详解》文章主要介绍了在使用SQLAlchemy创建模型实例时,通过元类动态创建实例的方式,并说明了如何在实例化时执行__init__方法,... 目录@orm.reconstructorSQLAlchemy的回滚关联其他模型数据库基本操作将数据添
阅读更多...
Java数组动态扩容的实现示例

Java数组动态扩容的实现示例

《Java数组动态扩容的实现示例》本文主要介绍了Java数组动态扩容的实现示例,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧... 目录1 问题2 方法3 结语1 问题实现动态的给数组添加元素效果，实现对数组扩容，原始数组使用静态分配
阅读更多...
Java中ArrayList与顺序表示例详解

Java中ArrayList与顺序表示例详解

《Java中ArrayList与顺序表示例详解》顺序表是在计算机内存中以数组的形式保存的线性表,是指用一组地址连续的存储单元依次存储数据元素的线性结构,：本文主要介绍Java中ArrayList与... 目录前言一、Java集合框架核心接口与分类ArrayList二、顺序表数据结构中的顺序表三、常用代码手动
阅读更多...
Python实现快速扫描目标主机的开放端口和服务

Python实现快速扫描目标主机的开放端口和服务

《Python实现快速扫描目标主机的开放端口和服务》这篇文章主要为大家详细介绍了如何使用Python编写一个功能强大的端口扫描器脚本,实现快速扫描目标主机的开放端口和服务,感兴趣的小伙伴可以了解下... 目录功能介绍场景应用1. 网络安全审计2. 系统管理维护3. 网络故障排查4. 合规性检查报错处理1.
阅读更多...
JAVA线程的周期及调度机制详解

JAVA线程的周期及调度机制详解

《JAVA线程的周期及调度机制详解》Java线程的生命周期包括NEW、RUNNABLE、BLOCKED、WAITING、TIMED_WAITING和TERMINATED,线程调度依赖操作系统,采用抢占... 目录Java线程的生命周期线程状态转换示例代码JAVA线程调度机制优先级设置示例注意事项JAVA线程
阅读更多...
Python轻松实现Word到Markdown的转换

Python轻松实现Word到Markdown的转换

《Python轻松实现Word到Markdown的转换》在文档管理、内容发布等场景中,将Word转换为Markdown格式是常见需求,本文将介绍如何使用FreeSpire.DocforPython实现... 目录一、工具简介二、核心转换实现1. 基础单文件转换2. 批量转换Word文件三、工具特性分析优点局
阅读更多...

Copyright China编程 23002807@qq.com

沪ICP备2023033072号-2