【高阶数据结构】红黑树详解

饿了我会自己捡代码吃2024-06-12 23:45

目录

前言

在前面我们学习了平衡二叉树，但是二叉搜索树有其自身的缺陷，假如往树中插入的元素有序或者接近有序，二叉搜索树就会退化成单支树，时间复杂度会退化成O(N)，因此map、set等关联式容器的底层结构是对二叉树进行了平衡处理，即采用平衡树来实现，除了AVL树，下面我们要学习的红黑树也是处理二叉树自身缺陷的一种方式

一、红黑树的概念

红黑树是一种二叉搜索树，它在每个结点上增加一个存储位表示结点的颜色，可以是红（Red）或黑（Black），通过对任何一条从根到叶子的路径上各个结点着色方式的限制，红黑树能够确保没有任何一条路径的长度会超出其他路径两倍可以（小于等于），所以是接近平衡的。

二、红黑树的性质

红黑树有几大性质，通过维持这些性质，来保证红黑树最长路径的节点个数不会超过最短路径节点个数的两倍。

  1. 每个节点不是黑色就是红色的
  2. 根节点是黑色的
  3. 如果一个节点是红色的，它的两个孩子节点一定是黑的
    （即：没有连续的红色节点）
  4. 对于每个节点，从该节点到其所有后代叶节点的所有路径上，均包含数量相同的黑色节点
    （即：每条路径都包含相同数量的黑色节点）
  5. 每个叶子节点是黑色的（此处的叶子结点指的是空节点）

我们可以得出：

最长路径为一黑一红

最短路径为全黑

三、红黑树节点的定义

cpp 复制代码 
enum Colour
{
	RED,
	BLACK
};

template<class K,class V>
struct RBTreeNode
{
	RBTreeNode<K, V>* _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)
	{}
};

这里我们将默认节点（新增节点）给为红节点，因为红黑树中的每条路径中黑色节点的路径必须相同，如果我们给黑节点，必定违反性质，但如果给红节点还有"一线生机"。

四、红黑树的插入

红黑树是加了平衡限制的二叉搜索树，因此我们可以分两步来进行插入操作
1.先按二叉搜索树的插入规则插入

cpp 复制代码 
template<class K,class V>
class 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);
		if (parent->_kv.first < kv.first)
		{
			parent->_right = cur;
		}
		else
		{
			parent->_left = cur;
		}
	}
private:
	Node* _root = nullptr;
};

2.检查一下插入新节点后有没有破坏红黑树的性质

我们的新增节点默认是红色

如果新增节点的双亲节点是黑色，则未违反红黑树性质，不需要调整；

如果新增节点的双亲节点是红的，则违反了红黑树的性质（不能有连续的红节点），此时我们需要分情况来调整

为了方便，我们假设cur为当前节点，parent为父节点，grandfather为爷爷节点，uncle为叔叔节点
在这些节点中，叔叔尤为重要，基本上是通过叔叔的状态来控制的

情况1：cur为红，parent为红，grandfather为黑，uncle为红

例如：
注：此处的数可能是一颗完整的树，也可能是一颗子树

我们的解决方法是：将parent，uncle改为黑，grandfather改为红，然后把grandfather当成cur，继续向上调整


如果grandfather是根节点的话，需要将grandfather变成黑色，因为红黑树的性质（根节点是黑色的）

如果grandfather是子树的话，给就会有父节点，且父节点为红色，然后就以grandfather为cur，继续向上调整

情况2： cur为红，parent为红，grandfather为黑，uncle不存在或者uncle存在且为黑

1.uncle不存在：这种情况稍微简单一点，如果uncle不存在，说明cur一定是新增节点，则cur和parent定有一个节点是黑色的，就违反了红黑树的性质（每条路径黑色节点个数相同 ）

2.uncle存在：如果uncle存在且为黑，那么cur原来就应该是黑的，至于为什么是红的，原因是cur的子树在调整过程中将cur的颜色变成了红色，换句话说，这种情况是一个中间态，是由其他情况调整而变来的

如果出现了上述情况，我们就需要进行旋转了

旋转的思路我在AVL树中有详细讲解，对旋转不熟悉的可以看看（https://blog.csdn.net/liuty0125/article/details/139563426?spm=1001.2014.3001.5502），下面我就只给出具体代码了

1.右单旋

parent为grandfather的左孩子，cur为parent的左孩子

cpp 复制代码 
if (parent == gradfather->_left)
{
	Node* uncle = gradfather->_right;	
	if (uncle && uncle->_col == RED)//情况1：叔叔存在且为红
	{
		//......
	}
	else//情况2：叔叔不存在或者叔叔存在且为黑
	{
		//右单旋
		if (cur == parent->_left)
		{
			RotateR(gradfather);
			parent->_col = BLACK;
			gradfather->_col = RED;
		}
		//左右双旋
		else
		{
			//......
		}
		break;
	}
}
else
{
	//......
}

2.左单旋

parent为grandfather的右孩子，cur为parent的右孩子

cpp 复制代码 
if (parent == gradfather->_left)
{
	//......
}
else
{
	Node* uncle = gradfather->_left;
	//情况1：叔叔存在且为红
	if (uncle && uncle->_col == RED)
	{
		//......
	}
	else//情况2：叔叔不存在或者叔叔存在且为黑
	{
		//左单旋
		if (cur == parent->_right)
		{
			RotateL(gradfather);
			parent->_col = BLACK;
			gradfather->_col = RED;
		}
		//右左双旋
		else
		{
			//......
		}
		break;
	}
}

3.左右单旋

parent为grandfather的左孩子，cur为parent的右孩子

cpp 复制代码 
if (parent == gradfather->_left)
{
	Node* uncle = gradfather->_right;	
	if (uncle && uncle->_col == RED)//情况1：叔叔存在且为红
	{
		//......
	}
	else//情况2：叔叔不存在或者叔叔存在且为黑
	{
		//右单旋
		if (cur == parent->_left)
		{
			//.......
		}
		//左右双旋
		else
		{
			RotateL(parent);
			RotateR(grandfather);
			cur->_col = BLACK;
			gradfather->_col = RED;
		}
		break;
	}
}
else
{
	//......
}

4.右左单旋

parent为grandfather的右孩子，cur为parent的左孩子

cpp 复制代码 
if (parent == gradfather->_left)
{
	//......
}
else
{
	Node* uncle = gradfather->_left;
	//情况1：叔叔存在且为红
	if (uncle && uncle->_col == RED)
	{
		//......
	}
	else//情况2：叔叔不存在或者叔叔存在且为黑
	{
		//左单旋
		if (cur == parent->_right)
		{
			//......
		}
		//右左双旋
		else
		{
			RotateR(parent);
			RotateL(grandfather);
			cur->_col = BLACK;
			gradfather->_col = RED;
		}
		break;
	}
}

五、红黑树与AVL树的比较

红黑树和AVL树都是性能十分不错的二叉搜索树，它们的增删查改的时间复杂度都是O(log_n)，

但是红黑树并不像AVL树一样追求绝对平衡，只需要保证最长路径不超过最短路径的两倍，所以相对的降低了插入和旋转的次数，因此在经常需要修改的结构中，红黑树的性能要比AVL树更佳，在实际中红黑树的运用更多。

六、完整代码

cpp 复制代码 
#pragma once

enum Colour
{
	RED,
	BLACK
};

template<class K,class V>
struct RBTreeNode
{
	RBTreeNode<K, V>* _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 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);
		if (parent->_kv.first < kv.first)
		{
			parent->_right = cur;
		}
		else
		{
			parent->_left = cur;
		}
		cur->_parent = parent;
		while (parent && parent->_col == RED)
		{
			Node* gradfather = parent->_parent;
			if (parent == gradfather->_left)
			{
				Node* uncle = gradfather->_right;	
				if (uncle && uncle->_col == RED)//情况1：叔叔存在且为红
				{
					parent->_col = uncle->_col = BLACK;
					gradfather->_col = RED;

					//继续往上处理
					cur = gradfather;
					parent = cur->_parent;
				}
				else//情况2：叔叔不存在或者叔叔存在且为黑
				{
					//右单旋
					if (cur == parent->_left)
					{
						RotateR(gradfather);
						parent->_col = BLACK;
						gradfather->_col = RED;
					}
					//左右双旋
					else
					{
						RotateL(parent);
						RotateR(grandfather);
						cur->_col = BLACK;
						gradfather->_col = RED;
					}
					break;
				}
			}
			else
			{
				Node* uncle = gradfather->_left;
				//情况1：叔叔存在且为红
				if (uncle && uncle->_col == RED)
				{
					parent->_col = uncle->_col = BLACK;
					gradfather->_col = RED;

					//继续往上处理
					cur = gradfather;
					parent = cur->_parent;
				}
				else//情况2：叔叔不存在或者叔叔存在且为黑
				{
					//左单旋
					if (cur == parent->_right)
					{
						RotateL(gradfather);
						parent->_col = BLACK;
						gradfather->_col = RED;
					}
					//右左双旋
					else
					{
						RotateR(parent);
						RotateL(grandfather);
						cur->_col = BLACK;
						gradfather->_col = RED;
					}
					break;
				}
			}
		}
		_root = BLACK;

		return true;
	}
	void RotateL(Node* parent)
	{
		Node* subR = parent->_right;
		Node* subRL = subR->_left;

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

		subR->_left = parent;
		Node* ppnode = parent->_parent;
		parent->_parent = subR;

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

		subL->_right = parent;

		Node* ppnode = parent->_parent;
		parent->_parent = subL;

		if (parent == _root)
		{
			_root = subL;
			subL->_parent = nullptr;
		}
		else
		{
			if (ppnode->_left == parent)
			{
				ppnode->_left = subL;
			}
			else
			{
				ppnode->_right = subL;
			}
			subL->_parent = ppnode;
		}
	}
	void _InOrder(Node* root)
	{
		if (root == nullptr)
			return;

		_InOrder(root->_left);
		cout << root->_kv.first << endl;
		_InOrder(root->_right);
	}

	void InOrder()
	{
		_InOrder(_root);
	}
private:
	Node* _root = nullptr;
};
相关推荐
韭菜盖饭1 小时前
LeetCode每日一题3261---统计满足 K 约束的子字符串数量 II
数据结构·算法·leetcode
♡喜欢做梦2 小时前
【数据结构】ArrayList与LinkedList详解！！！——Java
java·开发语言·数据结构·链表
好心的小明2 小时前
【深圳大学】数据结构A+攻略（计软版）
数据结构
熬夜学编程的小王2 小时前
【初阶数据结构篇】插入、希尔、选择、堆排序
数据结构·c++·插入排序·选择排序·希尔排序
三小尛2 小时前
快速排序（C语言）
数据结构·算法·排序算法
椅子哥2 小时前
数据结构--排序算法
java·数据结构·算法·排序算法
DDDiccc2 小时前
JAVA学习日记（十五） 数据结构
数据结构·学习
Nydia.J2 小时前
【学习笔记】数据结构（七）
数据结构·考研
2301_799084672 小时前
超全排序C语言实现
c语言·数据结构·算法·排序算法
苏唱！！！！2 小时前
二叉树-堆
数据结构
热门推荐
01软件工程从理论到实践客观题汇总(头歌第一章至第八章)02【HarmonyOS】HUAWEI DevEco Studio 下载地址汇总03组基轨迹建模 GBTM的介绍与实现（Stata 或 R）04VSCode插件 —— Cody AI （免费AI助手！）05全面解析：构建基于深度学习的安全帽检测系统（UI界面+YOLO代码+数据集）06Windows10安装PCL1.14.0及点云配准07Ubuntu 20.04使用Livox mid 360 测试 FAST_LIO08文件或文件夹名称中有空格如何批量去除09基于YOLOv10深度学习的CT扫描图像肾结石智能检测系统【python源码+Pyqt5界面+数据集+训练代码】深度学习实战、目标检测10【QuikGraph】C#调用第三方库实现迪杰斯特拉(Dijkstra)算法功能