C++笔记19•数据结构:红黑树(RBTree)•

news2024/11/15 17:59:45

红黑树

1.简介：

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

  • 当搜索二叉树退化为单支树时，搜索效率极低，为了使搜索效率高，建立平衡搜索二叉树就需要"平衡树"来解决。上一篇博客介绍了AVL树，这篇博客介绍的红黑树和AVLTree作用是一样的。
  • 如果在一棵原本是平衡的树中插入一个新节点，可能造成不平衡，此时必须调整树的结构，使之平衡化，用AVLTree或RBTree。
  • RBTree相对AVLTree效果略微差一些，但是相比AVLTree实现更简单一些，不需要平衡因子的不断更新，而是用红&黑颜色替代，只用到了左单旋和右单旋（RBTree的双旋是调用左单旋和右单旋），现在的硬件设备运转非常快，CUP的高速运转下RBTree与AVLTree的差别已经显得微不足道。关联式容器map/set的底层就是用RBTree实现的。
  • 性质：
    1. 每个结点不是红色就是黑色
    2. 根节点是黑色的 
    3. 如果一个节点是红色的，则它的两个孩子结点是黑色的 ( 没有连续的红节点)
    4. 对于每个结点，从该结点到其所有后代叶结点的简单路径上，均 包含相同数目的黑色结点 
    5. 每个叶子结点都是黑色的 ( 此处的叶子结点指的是空结点)

满足以上性质就可以保证：最长路径中节点的个数不会超过最短路径节点个数的两倍。

总结一下：RBTree(①根是黑的②没有连续的红节点③每条路径有相同数量的黑节点)

举例：

做个比方，假设上面的图，最短路径是：全黑 时间复杂度（log（N））     

                                           最长路径是：一黑一红，时间复杂度（2*log（N）） 

                                           所以最多是2倍。

现在的硬件设备运转非常快，log（N）和2*log（N）对CPU来处理真的很快。比如N=10亿，log（10亿）大概等于30；2*log（10亿）大概等于60；最短路径需要搜索30次，最短路径需要搜索60次，这对CPU来处理真的是不值一提的，所以说不管在最短或者最长路径上搜索数据，都是非常快的。

  •  红黑树是近似平衡,高度控制没有AVL树那么严格，增删查改的性能基本差不多。
  • 红黑树高度可能会高一些，但是它旋转得少一些。实际中红黑树综合而言更优一点，实际中红黑树用得更多。

 2.代码实现

#define _CRT_SECURE_NO_WARNINGS 1
#include <iostream>
#include <string>
#include <assert.h>
using namespace std;

//平衡搜索树 RBTree
//1. 每个结点不是红色就是黑色
//2. 根节点是黑色的 
//3. 如果一个节点是红色的，则它的两个孩子结点是黑色的(意思就是红色节点不能连续)
//4. 对于每个结点，从该结点到其所有后代叶结点的简单路径上，均包含相同数目的黑色结点 
//5. 每个叶子结点都是黑色的(此处的叶子结点指的是空结点)

enum Colour
{
	RED,
	BLACK,
};

template<class K, class V>
struct RBTreeNode
{
	RBTreeNode<K, V>* _left;//一定不要写成RBTreeNode*<K>  _left;  这样编译器无法识别
	RBTreeNode<K, V>* _right;
	RBTreeNode<K, V>* _parent;
	pair<K, V> _kv;
	Colour _col;

	RBTreeNode(const pair<K, V>& kv)
		:_left(nullptr)
		, _right(nullptr)
		, _parent(nullptr)
		, _kv(kv)
		, _col(RED)//新插入的节点颜色默认设置为红色，原因是针对红黑树的组建规则，红色限制少,但是根节点必须是黑色。
	{}
};

template<class K, class V>
class RBTree
{
	typedef struct RBTreeNode<K, V> Node;
public:
	bool insert(const pair<K, V>& kv)
	{
		if (_root == nullptr)
		{
			_root = new Node(kv);
			_root->_col = BLACK;
			return true;
		}
		Node* parent = nullptr;
		Node* cur = _root;

		while (cur)
		{
			if (cur->_kv.first < kv.first)
			{
				parent = cur;
				cur = cur->_right;
			}
			else if (cur->_kv.first > kv.first)
			{
				parent = cur;
				cur = cur->_left;
			}
			else
			{
				return false;
			}
		}
		//找到空了，开始插入
		cur = new Node(kv);
		cur->_col = RED;
		if (parent->_kv.first < kv.first)
		{
			parent->_right = cur;
		}
		else
		{
			parent->_left = cur;
		}
		cur->_parent = parent;

		//对RBTree进行颜色节点排查  检查是否存在连续的红色节点
		while (parent && parent->_col == RED)
		{
			Node* grandfater = parent->_parent;
			assert(grandfater);
			
			if (parent == grandfater->_left)
			{
				Node* uncle = grandfater->_right;

				//情况一：叔叔节点存在且颜色为红
				if (uncle && uncle->_col == RED)//叔叔节点存在且颜色为红
				{
					//变色：父亲和叔叔变黑，爷爷变红；
					parent->_col = BLACK;
					uncle->_col = BLACK;
					grandfater->_col = RED;

					//继续向上搜索
					cur = grandfater;
					parent = cur->_parent;
				}

				//情况二：叔叔节点不存在 或 叔叔节点存在且颜色为黑
				//(1)如果叔叔不存在，那么cur就是新增节点
				//(2)如果叔叔存在且颜色为黑，那么cur一定不是新增节点。

				else
				{
					if (cur == parent->_left)
					{
						//     右单旋
						//     grandfater
						//      /
						//   parent        ->    parent 
						//   /                    /  \
						// cur                  cur   grandfater
						RotateR(grandfater);
						parent->_col = BLACK;
						grandfater->_col = RED;
						//不用再向上搜索
					}
					else  //cur == parent->_right
					{
						//双旋
						//     g                  g              
						//   p        ->        c      ->       c
						//     c              p              p    g
						RotateL(parent);
						RotateR(grandfater);
						cur->_col = BLACK;
						grandfater->_col = RED;
					}
					break;
				}
			}
		

			else //parent == grandfater->_right
			{
				Node* uncle = grandfater->_left;

				//情况一：叔叔节点存在且颜色为红
				if (uncle&& uncle->_col == RED)//叔叔节点存在且颜色为红
				{
					//变色：父亲和叔叔变黑，爷爷变红；
					parent->_col = BLACK;
					uncle->_col = BLACK;
					grandfater->_col = RED;

					//继续向上搜索
					cur = grandfater;
					parent = cur->_parent;
				}

				//情况二：叔叔节点不存在 或 叔叔节点存在且颜色为黑
				//(1)如果叔叔不存在，那么cur就是新增节点
				//(2)如果叔叔存在且颜色为黑，那么cur一定不是新增节点。

				else
				{
					if (cur == parent->_right)
					{
						//     左单旋
						//     grandfater
						//         \
						//        parent        ->     parent 
						//            \                /     \
						//             cur        grandfater  cur 
						RotateL(grandfater);
						parent->_col = BLACK;
						grandfater->_col = RED;
						//不用再向上搜索
					}
					else  //cur == parent->_left
					{
						//双旋
						//     g                  g              
						//       p        ->        c      ->       c
						//     c                     p            g   p
						RotateR(parent);
						RotateL(grandfater);
						cur->_col = BLACK;
						grandfater->_col = RED;
					}
					break;
				}
			}
		}
		_root->_col = BLACK;

		return true;
	}
	void InOrder()
	{
		_InOrder(_root);
		cout << endl;
	}

	void Height()
	{
		cout << "最长路径:" << _maxHeight(_root) << endl;
		cout << "最短路径:" << _minHeight(_root) << endl;
	}


	bool IsBalanceTree()
	{
		// 检查红黑树几条规则

		Node* pRoot = _root;
		// 空树也是红黑树
		if (nullptr == pRoot)
			return true;

		// 检测根节点是否满足情况
		if (BLACK != pRoot->_col)
		{
			cout << "违反红黑树性质二：根节点必须为黑色" << endl;
			return false;
		}

		// 获取任意一条路径中黑色节点的个数 -- 比较基准值
		size_t blackCount = 0;
		Node* pCur = pRoot;
		while (pCur)
		{
			if (BLACK == pCur->_col)
				blackCount++;

			pCur = pCur->_left;
		}

		// 检测是否满足红黑树的性质，k用来记录路径中黑色节点的个数
		size_t k = 0;
		return _IsValidRBTree(pRoot, k, blackCount);
	}

private:
	Node* _root=nullptr;
	void _InOrder(Node* root)
	{
		if (root == nullptr)
		{
			return;
		}
		_InOrder(root->_left);
		cout << root->_kv.first << ":" << root->_kv.second<<endl;
		_InOrder(root->_right);
	}

	void RotateL(Node* parent)//左单旋
	{
		Node* ppNode = parent->_parent;
		Node* subR = parent->_right;
		Node* subRL = subR->_left;

		parent->_right = subRL;
		if (subRL)
		{
			subRL->_parent = parent;
		}

		subR->_left = parent;
		parent->_parent = subR;

		if (parent == _root)
		{
			_root = subR;
			_root->_parent = nullptr;//subR->_parent = nullptr; //不可以只写这一句  如果parent是根 必须要更新_root; 加上 _root = subR;
		}
		else
		{
			if (parent == ppNode->_left)
			{
				ppNode->_left = subR;
			}
			else  //parent == ppNode->_right
			{
				ppNode->_right = subR;
			}
			subR->_parent = ppNode;
		}

	}
	void RotateR(Node* parent)//右单旋
	{
		Node* ppNode = parent->_parent;
		Node* subL = parent->_left;
		Node* subLR = subL->_right;

		parent->_left = subLR;
		if (subLR)
		{
			subLR->_parent = parent;
		}

		subL->_right = parent;
		parent->_parent = subL;

		if (parent == _root)
		{
			_root = subL;
			_root->_parent = nullptr;//subR->_parent = nullptr; //不可以只写这一句  如果parent是根 必须要更新_root; 加上 _root = subR;
		}
		else
		{
			if (parent == ppNode->_left)
			{
				ppNode->_left = subL;
			}
			else  //parent == ppNode->_right
			{
				ppNode->_right = subL;
			}
			subL->_parent = ppNode;
		}

	}

	int _maxHeight(Node* root)
	{
		if (root == nullptr)
			return 0;

		int lh = _maxHeight(root->_left);
		int rh = _maxHeight(root->_right);

		return lh > rh ? lh + 1 : rh + 1;
	}

	int _minHeight(Node* root)
	{
		if (root == nullptr)
			return 0;

		int lh = _minHeight(root->_left);
		int rh = _minHeight(root->_right);

		return lh < rh ? lh + 1 : rh + 1;
	}


	bool _IsValidRBTree(Node* pRoot, size_t k, const size_t blackCount)
	{
		//走到null之后，判断k和black是否相等
		if (nullptr == pRoot)
		{
			if (k != blackCount)
			{
				cout << "违反性质四：每条路径中黑色节点的个数必须相同" << endl;
				return false;
			}
			return true;
		}

		// 统计黑色节点的个数
		if (BLACK == pRoot->_col)
			k++;

		// 检测当前节点与其双亲是否都为红色
		if (RED == pRoot->_col && pRoot->_parent && pRoot->_parent->_col == RED)
		{
			cout << "违反性质三：存在连在一起的红色节点" << endl;
			return false;
		}

		return _IsValidRBTree(pRoot->_left, k, blackCount) &&
			_IsValidRBTree(pRoot->_right, k, blackCount);
	}
};

void TestAVLTree1()
{
	//int a[] = {10, 1, 2, 3, 4, 5, 6, 7, 8,9 };
	int a[] = { 40,50,30,29,28,27,0,26,25,24,11,8,7,6,5,4,3,2,1 };
	RBTree<int, int> t;
	for (auto e : a)
	{
		t.insert(make_pair(e, e));
	}
	t.InOrder();
	cout << t.IsBalanceTree() << endl;
}
void TestAVLTree2()
{
	int a[] = { 30,29,28,27,26,25,24,11,8,7,6,5,4,3,2,1 };
	RBTree<int, int> t;
	for (auto e : a)
	{
		bool res = t.insert(make_pair(e, e));
		if (res)
		{
			cout << "Inserted: " << e << endl;
		}
		else
		{
			cout << "Failed to insert: " << e << endl;
		}
	}
	t.InOrder();
	cout << t.IsBalanceTree() << endl;

}


int main()
{
	TestAVLTree1();
	//TestAVLTree2();
	return 0;
}

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

如若内容造成侵权/违法违规/事实不符，请联系多彩编程网进行投诉反馈，一经查实，立即删除！

相关文章

F12抓包05：Network接口测试（抓包篡改请求）

F12抓包05：Network接口测试（抓包篡改请求）

课程大纲 使用线上接口测试网站演示操作&#xff0c;浏览器F12检查工具如何进行简单的接口测试&#xff1a;抓包、复制请求、篡改数据、发送新请求。 测试地址&#xff1a;https://httpbin.org/forms/post ① 抓包&#xff1a;鼠标右键打开“检查”工具&#xff08;F12&#xf…
阅读更多...
卷积神经网络(一)

卷积神经网络(一)

目录 一.卷积神经网络的组成 二.卷积层 目的： 参数： 计算公式 卷积运算过程 三.padding-零填充 1.Valid and Same卷积 2.奇数维度的过滤器 四.stride步长 五.多通道卷积 1.多卷积核(多个Filter) 六.卷积总结 七.池化层(Pooling) 八.全连接层 都看到这里了，点个…
阅读更多...
Lanenet - 实时车道线检测系统

Lanenet - 实时车道线检测系统

基于深度学习的实时车道线检测&#xff0c;lanenet,tensorflow框架&#xff0c;有界面&#xff0c;可以检测图像和视频等。 项目名称&#xff1a;Lanenet - 实时车道线检测系统 项目概述 Lanenet 是一个利用深度学习技术进行实时车道线检测的系统。该系统旨在帮助驾驶员在行驶…
阅读更多...
分享7款实现社会实践报告AI生成论文网站

分享7款实现社会实践报告AI生成论文网站

在当今社会&#xff0c;AI技术的快速发展极大地改变了我们的生活方式和工作方式。特别是在学术研究和写作领域&#xff0c;AI工具的应用已经变得越来越普遍。本文将详细介绍7款实现社会实践报告AI生成的论文网站&#xff0c;并重点推荐千笔-AIPassPaper。 1. 千笔-AIPassPaper…
阅读更多...
【Pytorch】加载数据

【Pytorch】加载数据

数据集获取&#xff1a;链接: https://pan.baidu.com/s/1jZoTmoFzaTLWh4lKBHVbEA 密码: 5suq 本文基于P5. PyTorch加载数据初认识_哔哩哔哩_bilibili dataset&#xff1a;提供一种方式去获取数据及其label值&#xff0c;解释&#xff1a;Pytorch中的dataset类——创建适应任意…
阅读更多...
使用rsyslog转发自定义日志到指定服务器

使用rsyslog转发自定义日志到指定服务器

rsyslog简介 rsyslog 是一个高度可配置的、功能强大的系统日志守护进程&#xff0c;广泛用于 UNIX 和 Linux 系统中。它是 syslog 的一个扩展版本&#xff0c;提供了许多额外的功能和改进。能够收集、过滤、存储和转发日志数据。它的灵活性和扩展性使其成为现代 Linux 系统中日…
阅读更多...
剪辑视频，这四大工具助你一臂之力！

剪辑视频，这四大工具助你一臂之力！

在这个数字化的时代&#xff0c;视频已成为一种重要的表达手段。无论您是专业视频制作者还是只是偶尔想要编辑一些个人视频&#xff0c;一款优秀的视频剪辑软件都将是您不可或缺的好帮手。以下是几款值得推荐的视频剪辑软件。 福昕视频剪辑 直达链接&#xff1a;www.pdf365.c…
阅读更多...
巧用智能表单高效收集客户信息

巧用智能表单高效收集客户信息

企客宝企微版的智能表单功能&#xff0c;方便企微好友提交信息&#xff0c;直接入库管理&#xff0c;确保了一些类似身份证号等敏感信息&#xff0c;在传递过程中的数据安全 前言 很多企业在与企微好友沟通时&#xff0c;有时会有收集客户信息的需求&#xff0c;比如客户报名的…
阅读更多...
PID控制算法（一）

PID控制算法（一）

PID算法控制&#xff1a; PID算法应用十分广泛&#xff0c;包括温度、气压控制&#xff0c;流速、液位控制&#xff0c;无人机悬停&#xff0c;小球摆动受力以及姿态调整等等。 此时&#xff0c;假定有固定时间间隔t&#xff0c;对应有不同t时刻的输出值x&#xff1b;另外&…
阅读更多...
【机器学习】生成对抗网络(Generative Adversarial Networks, GANs)详解

【机器学习】生成对抗网络(Generative Adversarial Networks, GANs)详解

&#x1f308;个人主页: 鑫宝Code &#x1f525;热门专栏: 闲话杂谈&#xff5c; 炫酷HTML | JavaScript基础 ​&#x1f4ab;个人格言: "如无必要&#xff0c;勿增实体" 文章目录 生成对抗网络(Generative Adversarial Networks, GANs)详解GANs的基本原理GANs的…
阅读更多...
基于人工智能的情感分析系统

基于人工智能的情感分析系统

目录 引言项目背景环境准备 硬件要求软件安装与配置系统设计 系统架构关键技术代码示例 数据预处理模型训练模型预测应用场景结论 1. 引言 情感分析系统是自然语言处理中的重要应用之一&#xff0c;用于从文本中自动识别和分析用户情感&#xff0c;如“积极”、“消极”或“…
阅读更多...
UDP通信实现

UDP通信实现

目录 前言 一、基础知识 1、跨主机传输 1、字节序 2、主机字节序和网络字节序 3、IP转换 2、套接字 3、什么是UDP通信 二、如何实现UDP通信 1、socket():创建套接字 2、bind():绑定套接字 3、sendto():发送指定套接字文件数据 4、recvfrom():接收指定地址信息的数据 三…
阅读更多...
C语言操作符汇总（上）

C语言操作符汇总（上）

目录 前言 一、操作符的分类 二、⼆进制和进制转换 1. 二进制转10进制 2. 10进制转2进制数字 3. 2进制转8进制和16进制 3.1 2进制转8进制 3.2 二进制转16进制 三、原码、反码、补码 四、移位操作符 1. 左移操作符 2. 右移操作符 五、位操作符&#xff1a;&…
阅读更多...
10-1RT-Thread动态内存管理

10-1RT-Thread动态内存管理

10-1RT-Thread动态内存管理 在嵌入式系统中&#xff0c;变量和中间数据一般存放在系统存储空间中。只有在实际使用时&#xff0c;才将它们从存储空间读取到CPU进行运算。存储空间可分为两种&#xff0c;内部存储空间rem和外部存储空间rome或flash。其中ram或称之为内存&…
阅读更多...
【Linux详解】命令行参数｜环境变量

【Linux详解】命令行参数｜环境变量

目录 一、命令行参数 二、环境变量 1.环境变量的基本概念 2.查看环境变量的方法 3.环境变量相关命令 4.环境变量的组织方式以及获取环境变量的三种方法 环境变量具有全局属性 一、命令行参数 【示例1】main函数也是函数&#xff0c;main函数可以带参吗&#xff1f; 没…
阅读更多...
Python教程（二十） : 十分钟入门【PyQt6】

Python教程（二十） : 十分钟入门【PyQt6】

文章目录 专栏列表环境准备1 安装 Python2 安装 PyQt6 PyQt6 中的模块使用模块创建一个窗体&#xff1a; PyQt6 常用的控件1. QPushButton&#xff08;按钮&#xff09;2. QLabel&#xff08;标签&#xff09;3. QLineEdit&#xff08;文本输入框&#xff09;4. QTextEdit&…
阅读更多...
(4)SVG-path中的椭圆弧A(绝对)或a(相对)

(4)SVG-path中的椭圆弧A(绝对)或a(相对)

1、概念 表示经过起始点(即上一条命令的结束点)&#xff0c;到结束点之间画一段椭圆弧 2、7个参数 rx&#xff0c;ry&#xff0c;x-axis-rotation&#xff0c;large-arc-flag&#xff0c;sweep-flag&#xff0c;x&#xff0c;y &#xff08;1&#xff09;和&#xff08;2&a…
阅读更多...
FFMpeg环境搭建(WIN10)

FFMpeg环境搭建(WIN10)

0、前期准备 软件环境&#xff1a;Win10 qtcreator 软件准备&#xff1a;MSYS2 安装包、 FFmpeg源码 1、软件安装 通过MSYS2安装编译工具 1、打开MSYS2安装包&#xff0c;一路next即可 &#xff08;注&#xff1a;如果需要更改路径可以自行更改&#xff09; 2、安装完成…
阅读更多...
虚拟现实辅助工程技术助力多学科协同评估

虚拟现实辅助工程技术助力多学科协同评估

在当今高速发展的经济环境中&#xff0c;制造业面临着多重挑战&#xff0c;包括提高产品性能、压缩设计周期、实现轻量化设计和降低成本。为了有效应对这些挑战&#xff0c;多学科协同评估成为缩短研发周期和提升研制质量的关键手段。 传统的多学科评估面临着数据孤立与融合困难…
阅读更多...
Android 系统源码项目加载预编好的so库

Android 系统源码项目加载预编好的so库

Android 系统源码项目加载预编好的so库 文章目录 Android 系统源码项目加载预编好的so库一、前言二、源码中加载so1、Android.mk加载so加载so的主要相关代码&#xff1a; 2、Android.bp加载so&#xff08;1&#xff09;Android.mk使用源码命令编译成Android.bp&#xff08;2&am…
阅读更多...
推荐文章
最新文章

Copyright @ 2022~2023