深入浅出C++ ——手撕红黑树

已于 2023-05-18 18:00:07 修改
阅读量467 收藏 3
点赞数
分类专栏： C++ 数据结构 文章标签： c++ 红黑树 STL 树
于 2023-02-22 19:21:54 首次发布
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/weixin_55051736/article/details/127971072
版权

文章目录

一、红黑树的概念

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

在这里插入图片描述

二、红黑树的性质

  1. 每个结点不是红色就是黑色
  2. 根节点是黑色的
  3. 如果一个节点是红色的，则它的两个孩子结点是黑色的，即树中没有连续的红色节点
  4. 对于每个结点，从该结点到其所有后代叶结点的简单路径上，均包含相同数目的黑色结点，即每条路径的黑色节点数量相等
  5. 每个叶子结点(这里的叶子结点指的是空结点，也叫做NIL节点 ）都是黑色的

  红黑树中第三和第四条规则构成互斥，极限的最短一定是全黑，极限的最长一定是一黑一红 。

三、红黑树节点的定义

enum Colour
{
	RED,
	BLACK
};

template<class K, class V>
struct RBTreeNode
{
	RBTreeNode(const pair<K, V>& kv)
		:_left(nullptr)
		, _right(nullptr)
		, _parent(nullptr)
		, _kv(kv)
	{}

	pair<K, V> _kv;		//节点的值域
	Colour _col;		//节点的颜色

	RBTreeNode<K, V>* _left;	//节点的左孩子
	RBTreeNode<K, V>* _right;	//节点的右孩子
	RBTreeNode<K, V>* _parent;	//节点的双亲
};

四、红黑树的插入操作

  红黑树是在二叉搜索树的基础上加上其平衡限制条件，因此红黑树的插入可分为两步：一、按照二叉搜索的树规则插入新节点。二、检测新节点插入后，红黑树的性质是否造到破坏。

  因为新节点的默认颜色是红色，如果其双亲节点的颜色是黑色，没有违反红黑树任何性质，则不需要调整；但当新插入节点的双亲节点颜色为红色时，就违反了性质三（不能有连续的红色节点），此时需要对红黑树分情况来讨论。

情况一: cur为红，p为红，g为黑，u存在且为红

  cur为当前节点，p为父节点，g为祖父节点，u为叔叔节点。此处看到的树，可能是一颗完整的树，但是也有可能是一颗子树。

在这里插入图片描述

  解决方式是将将p,u改为黑，g改为红。继续把g当成cur，g是根节点，需要将g改为黑色，如果g是子树且g的双亲颜色为红色，则需要继续向上调整。

情况二: cur为红，p为红，g为黑，u不存在/u存在且为黑

  如果u节点不存在，那么cur一定为新插入的节点，否则就违反性质4。如果u节点存在，则一定是黑色的，cur之前也是黑色的，因为新增节点是a、b节点，所以现在它是红色。
在这里插入图片描述

  解决方式：如果p为g的左孩子，cur为p的左孩子，则进行右单旋转；p为g的右孩子，cur为p的右孩子，则进行左单旋转。然后再将p变黑，g变红。

情况三: cur为红，p为红，g为黑，u不存在/u存在且为黑
在这里插入图片描述
  解决方式：如果p为g的左孩子，cur为p的右孩子，则针对p做左单旋转；如果p为g的右孩子，cur为p的左孩子，则针对p做右单旋转。旋转后就转变为了情况二。

总结

  在这三种情况中，都有cur为红，p为红，g为黑，可以看出红黑树的关键是叔叔。U存在且为红则变色继续往上处理，U不存在或为黑，则旋转加变色。

  情况二和情况三的主要区别是单旋和双旋的不同。从图中可以看出，当p g cur为一条直线的时候，也就是情况二，单旋即可；p g cur 为一条折线的时候，也就是情况三，需要双旋。

代码实现

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;
	while (parent && parent->_col == RED)
	{
		Node* grandfater = parent->_parent;
		assert(grandfater);
		assert(grandfater->_col == BLACK);
		// 关键看叔叔
		if (parent == grandfater->_left)
		{
			Node* uncle = grandfater->_right;
			// 情况一 : uncle存在且为红，变色+继续往上处理
			if (uncle && uncle->_col == RED)
			{
				parent->_col = uncle->_col = BLACK;
				grandfater->_col = RED;
				// 继续往上处理
				cur = grandfater;
				parent = cur->_parent;
			}// 情况二+三：uncle不存在 + 存在且为黑
			else
			{
				// 情况二：右单旋+变色
				//     g 
				//   p   u
				// c
				if (cur == parent->_left)
				{
					RotateR(grandfater);
					parent->_col = BLACK;
					grandfater->_col = RED;
				}
				else
				{
					// 情况三：左右单旋+变色
					//     g 
					//   p   u
					//     c
					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 = uncle->_col = BLACK;
				grandfater->_col = RED;
				// 继续往上处理
				cur = grandfater;
				parent = cur->_parent;
			}
			else
			{
				// 情况二：左单旋+变色
				//     g 
				//   u   p
				//         c
				if (cur == parent->_right)
				{
					RotateL(grandfater);
					parent->_col = BLACK;
					grandfater->_col = RED;
				}
				else
				{
					// 情况三：右左单旋+变色
					//     g 
					//   u   p
					//     c
					RotateR(parent);
					RotateL(grandfater);
					cur->_col = BLACK;
					grandfater->_col = RED;
				}
				break;
			}
		}

	}
	_root->_col = BLACK;
	return true;
}

五、红黑树的验证

  红黑树的检测分为两步：一、检测其是否满足二叉搜索树，这里只需要中序遍历是否为有序序列即可。二、检测其是否满足红黑树的性质。

public:
	bool IsBalance()
	{
		if (_root == nullptr)
		{
			return true;
		}
		if (_root->_col == RED)
		{
			cout << "根节点不是黑色" << endl;
			return false;
		}
		int benchmark = 0;
		return PrevCheck(_root, 0, benchmark);
	}
private:
	bool PrevCheck(Node* root, int blackNum, int& benchmark)
	{
		if (root == nullptr)
		{
			if (benchmark == 0)
			{
				benchmark = blackNum;
				return true;
			}
			if (blackNum != benchmark)
			{
				cout << "某条黑色节点的数量不相等" << endl;
				return false;
			}
			else
			{
				return true;
			}
		}
		if (root->_col == BLACK)
		{
			++blackNum;
		}
		if (root->_col == RED && root->_parent->_col == RED)
		{
			cout << "存在连续的红色节点" << endl;
			return false;
		}
		return PrevCheck(root->_left, blackNum, benchmark)
			&& PrevCheck(root->_right, blackNum, benchmark);
	}

五、红黑树的删除

  红黑树的删除过于复杂，可以参考 红黑树的删除 这篇博客

六、红黑树与AVL树的比较

  红黑树和AVL树都是高效的平衡二叉树，增删改查的时间复杂度都是O( l o g 2 N log_2 N log2​N)，红黑树不追求绝对平衡，其只需保证最长路径不超过最短路径的2倍，相对而言，降低了插入和旋转的次数，所以在经常进行增删的结构中性能比AVL树更优，而且红黑树实现比较简单，所以实际运用中红黑树更多。

七、红黑树的应用

  红黑树的应用场景建议观看这个视频讲解红黑树在linux中的3种应用场景

八、红黑树模拟实现

#pragma once
#include<iostream>
#include<assert.h>
using namespace std;

enum Colour
{
	RED,
	BLACK
};

template<class K, class V>
struct RBTreeNode
{
	RBTreeNode(const pair<K, V>& kv)
		:_left(nullptr)
		, _right(nullptr)
		, _parent(nullptr)
		, _kv(kv)
	{}

	pair<K, V> _kv;
	Colour _col;

	RBTreeNode<K, V>* _left;
	RBTreeNode<K, V>* _right;
	RBTreeNode<K, V>* _parent;
};

template<class K, class V>
struct RBTree
{
	typedef 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;
		while (parent && parent->_col == RED)
		{
			Node* grandfater = parent->_parent;
			assert(grandfater);
			assert(grandfater->_col == BLACK);
			// 关键看叔叔
			if (parent == grandfater->_left)
			{
				Node* uncle = grandfater->_right;
				// 情况一 : uncle存在且为红，变色+继续往上处理
				if (uncle && uncle->_col == RED)
				{
					parent->_col = uncle->_col = BLACK;
					grandfater->_col = RED;
					// 继续往上处理
					cur = grandfater;
					parent = cur->_parent;
				}// 情况二+三：uncle不存在 + 存在且为黑
				else
				{
					// 情况二：右单旋+变色
					//     g 
					//   p   u
					// c
					if (cur == parent->_left)
					{
						RotateR(grandfater);
						parent->_col = BLACK;
						grandfater->_col = RED;
					}
					else
					{
						// 情况三：左右单旋+变色
						//     g 
						//   p   u
						//     c
						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 = uncle->_col = BLACK;
					grandfater->_col = RED;
					// 继续往上处理
					cur = grandfater;
					parent = cur->_parent;
				}
				else
				{
					// 情况二：左单旋+变色
					//     g 
					//   u   p
					//         c
					if (cur == parent->_right)
					{
						RotateL(grandfater);
						parent->_col = BLACK;
						grandfater->_col = RED;
					}
					else
					{
						// 情况三：右左单旋+变色
						//     g 
						//   u   p
						//     c
						RotateR(parent);
						RotateL(grandfater);
						cur->_col = BLACK;
						grandfater->_col = RED;
					}

					break;
				}
			}

		}
		_root->_col = BLACK;
		return true;
	}

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

	bool IsBalance()
	{
		if (_root == nullptr)
		{
			return true;
		}
		if (_root->_col == RED)
		{
			cout << "根节点不是黑色" << endl;
			return false;
		}
		// 黑色节点数量基准值
		int benchmark = 0;
		return PrevCheck(_root, 0, benchmark);
	}
	
private:
	bool PrevCheck(Node* root, int blackNum, int& benchmark)
	{
		if (root == nullptr)
		{
			if (benchmark == 0)
			{
				benchmark = blackNum;
				return true;
			}

			if (blackNum != benchmark)
			{
				cout << "某条黑色节点的数量不相等" << endl;
				return false;
			}
			else
			{
				return true;
			}
		}

		if (root->_col == BLACK)
		{
			++blackNum;
		}

		if (root->_col == RED && root->_parent->_col == RED)
		{
			cout << "存在连续的红色节点" << endl;
			return false;
		}

		return PrevCheck(root->_left, blackNum, benchmark)
			&& PrevCheck(root->_right, blackNum, benchmark);
	}

	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* subR = parent->_right;
		Node* subRL = subR->_left;

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

		Node* ppNode = parent->_parent;

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

		if (_root == parent)
		{
			_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;

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

		Node* ppNode = parent->_parent;

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

		if (_root == parent)
		{
			_root = subL;
			subL->_parent = nullptr;
		}
		else
		{
			if (ppNode->_left == parent)
			{
				ppNode->_left = subL;
			}
			else
			{
				ppNode->_right = subL;
			}
			subL->_parent = ppNode;
		}
	}
private:
	Node* _root = nullptr;
};
程序员丶Alex
关注
  • 0
    点赞
  • 踩
  • 3
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
专栏目录
Python手撕算法Recursion
12-25
Python手撕算法Recursion
Python手撕算法mysingleton
12-25
Python手撕算法mysingleton
Python手撕算法RadixSort
12-25
Python手撕算法RadixSort
28.红黑树_1.wmv（目前数据结构最好的视频,共42集，需要哪一集的知识自己下，一集3积分）
03-13
C/C++实现 数据结构与算法视频教程 01交换算法 02冒泡排序 03选择排序 04顺序排序 05折半排序 06递归算法 07递归折半查找 08算法_perm 09插入排序 10快速排序 11归并排序 12顺序栈 13顺序队列 14链表的基本概率 15链表 16链表的迭代器 17循环链表 18双项链表 19链式栈 20链式队列 21STL_list类 22基数排序 23属 24二叉树 25二叉树找数 26红黑树 27红黑树_0 28红黑树_1 29红黑树_2 30红黑树_3 31红黑树_4 32红黑树_5 33红黑树_6 34堆 35堆排序 36哈希与映射的概述 37B树有什么用 38B树的概念 39图_邻接矩阵 40图_邻接表 41图_DFS 42图_BFS
Python手撕算法QuickSort
12-25
Python手撕算法QuickSort
【数据结构】手撕红黑树
学了就忘是正常现象
02-26 8612
红黑树和AVL树一样，因其逻辑复杂，面试时现场要求手撕就是纯纯刁难面试者。但某大厂面试官曾要求某些求职者现场手撕红黑树（我赌5毛，让面试官撕，他也撕不出来，而且你家员工上班手搓红黑树啊？），随后求职遭遇被发到网上吐槽，这便有了“手撕红黑树”的梗，也让红黑树成为了知名度最高的数据结构。（话虽如此，对于红黑树的性质、插入思想等概念还是需要掌握的）
保姆级别带你手撕红黑树BRTree
我的学习笔记分享
04-20 1524
一、前言 本文是自己在学习过程中，记录下的一些总结，以便日后复习，同时分享出来，方便与更多人学习交流，共同进步。鄙人水平有限，若有误，请不吝赐教😁😁😁。 二、从头介绍 红黑树树是一种二叉树，比较常用的数据结构，如果对二叉树不是很熟悉或者有遗忘的朋友，可以看看我之前做的笔记。 第一篇介绍二叉树的基础知识：数据结构与算法 树_振予的博客-CSDN博客 第二篇介绍二叉搜索树和二叉平衡树树：数据结构与算法 查找_振予的博客-CSDN博客 三、红黑树介绍 每当我们打开电脑查找一个文档、在教务系统输入我们
【高阶数据结构】手撕红黑树（超详细版本）
qq_42996461的博客
01-08 2028
但行好事，莫问前程
手撕红黑树很难？看完零基础都能手撕各种二叉树
qq_43778239的博客
01-14 545
从零基础通过图片和动图来讲解数据结构中二叉树的知识点，并实现常见的二叉树，附带相关leetcode题目，适合零基础或者刚刚入门算法小白阅读，考研的同学也可以来了解了解喔。
C++手撕红黑树
分享学习的知识，记录我的技术成长道路
08-07 614
红黑树在表意上就是一棵每个节点带有颜色的二叉搜索树，并通过对节点颜色的控制，使该二叉搜索树达到尽量平衡的状态。红黑树确保没有一条路径比其他路径长两倍。和AVL树不同的是，AVL树是一棵平衡树，而红黑树可能平衡也可能不平衡(因为是尽量平衡的状态)每一次插入都对根节点置为黑操作，因为第一种情况可能导致根节点不是黑。同时对根节点置黑也并不违反三条规定。...
Python手撕算法Queue
12-25
Python手撕算法Queue
深入浅出MFC—— C++ 经典 光盘
11-06
深入浅出MFC,学习MFC编程的人都有所耳闻的，这个是光盘的内容
红黑树-基于C++实现的红黑树数据结构.zip
03-21
红黑树 红黑树_基于C++实现的红黑树数据结构
C++数据结构 红黑树 实现全部功能
12-29
使用C++编写的红黑树源代码
使用C++实现的rbtree红黑树.zip
03-21
红黑树 使用C++实现的rbtree红黑树
红黑树的实现C++代码
07-27
用C++代码实现了一个红黑树，大家可以一起学习学习，有问题也可以直接给我留言
【C++第三阶段】list容器&排序
还有很多需要学习，现在还很菜
04-29 687
以下内容仅为当前认识，可能有不足之处，欢迎讨论！
C#(C Sharp)学习笔记_类【十五】
最新发布
追求细节，成就完美
05-01 465
类（Class）是面向对象程序设计（OOP，Object-Oriented Programming）实现信息封装的基础。类是一种用户定义的引用数据类型，也称类类型。每个类包含数据说明和一组操作数据或传递消息的函数。类的实例称为对象。
《21天学通C++》（第十章）实现继承（1）
立志成为炼丹师？
04-28 299
覆盖是一种极端情况，即派生类中的重载方法会隐藏基类中方法的所有重载版本，导致需要调用时无法调用，示例代码如下。解决方法3：在派生类中，把基类中的方法全部重载一次，并在派生类中就调用基类方法。只要派生类中实现了从基类中继承的函数，且返回值和标值相同，就可以覆盖。如果基类包含重载的构造函数，可以使用初始化列表传递参数。，避免不定义的情况下，随机数值产生影响。解决办法1：使用作用域解析符。解决方法2：在派生类中使用。同样使用作用域解析运算符。在构造一个派生类时，使用作用域解析运算符。若想调用基类中的方法。
C++ map set 红黑树
07-28
C++中的map和set容器都是基于红黑树实现的。 红黑树是一种自平衡的二叉搜索树，它通过对节点进行染色来满足一定的性质，从而保持树的平衡。红黑树的性质包括： 1. 每个节点只能是红色或黑色。 2. 根节点是黑色。 3...

“相关推荐”对你有帮助么？

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

程序员丶Alex

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付：¥1
获取中
扫码支付

您的余额不足，请更换扫码支付或充值

打赏作者

实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值