手撕红黑树（map和set底层结构）（2）

news2024/9/21 7:32:23

@[TOC]红黑树

一 红黑树概念

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

简单来说红黑树通过了对颜色的控制进而对树的长度做出了限制，不再需要平衡因子。
红黑树性质

  1. 每个节点不是红色就是黑色
  2. 根节点为黑色
  3. 如果一个节点是红色，那么他的孩子节点就是黑色。没有连续的红色节点
  4. 对于每个节点，他的所以路径上的黑色节点的数量是相同的。
  5. 每个叶子节点都是黑色的（注意：这里的叶子节点指的是空节点）

红黑树通过这些性质保证了他的高度是合法的。
在这里插入图片描述
红黑树保证的是：最长路径不超过最短路径的两倍

二 红黑树的实现

2.1 红黑树的结构

enum Color
{
	Red,
	Black,
};


template<class K, class V>
struct RBTreeNode
{
	RBTreeNode<K, V>* _left;
	RBTreeNode<K, V>* _right;
	RBTreeNode<K, V>* _parent;
	Color _color;
	pair<K, V> _kv;

	RBTreeNode(const pair<K, V>& kv)
		:_left(nullptr)
		, _right(nullptr)
		, _parent(nullptr)
		, _color(Red)
		, _kv(kv)
	{}
};
template<class K,class V>
class RBtree
{
	typedef RBTreeNode<K, V> Node;
	private:
	Node* _node=nullptr;
}

2.2 红黑树的插入

在这里插入图片描述
我们要注意：我们在插入新节点的时候，默认插入元素的颜色为红色，（黑色节点不好控制，因为要保证全部的路径的黑色节点数量是相同的，插入了一个黑色节点，就不能保证这一原则了）
然后插入元素对整棵树的影响我们就要从局部开始看，

  1. 如果插入元素的父亲为黑色那就不用在变化了；
  2. 插入元素的父亲为红色，此时就出翔了连续的红色节点，我们就要对这颗树进行处理；

我们对第二种情况分类讨论（用上述的图片为例）

第一种：cur为红，p为红，g为黑，u存在且为红

把p和u变为黑色，再把g变成红色；
到这里就要继续分类讨论：

  1. g为根节点，那就把g再变成黑色。
  2. 如果不是根节点，就把g=c，向上继续处理
    注意（p/u是g的左还是右都不影响，同样cur是p的左还是右也不影响）

第二种: cur为红，p为红，g为黑，u不存在/u存在且为黑
（i）u不存在在这里插入图片描述
这里cur一定是新插入的节点，如果cur不是新插入的节点，cur和p一定有一个是黑色的。
（ii）u存在且为黑色在这里插入图片描述
这里的cur就不是新插入的节点，u是黑色那么，每个路径下黑色节点数量一定相同，所以cur原本一定是黑色的，是因为新插入的原因被变成了红色。
以上这两个情况本质上是一种。
这里就要用到旋转，旋转和前面AVL树的旋转一模一样

p为g的左孩子，cur为p的左孩子，则进行右单旋转；相反，
p为g的右孩子，cur为p的右孩子，则进行左单旋转
旋转完：p、g变色–p变黑，g变红。

参考下面的图，进一步理解
在这里插入图片描述
在这里插入图片描述
第三种 cur为红，p为红，g为黑，u不存在/u存在且为黑，但这里的左右位置不同

(i) p为g的左，cur为p的右
在这里插入图片描述
在这里插入图片描述
可以看到这里就变回了情况二，根据上面的我们再进行右单旋即可。
在这里插入图片描述
(i) p为g的右，cur为p的左

这个就是先右旋+左旋即可。这个图留个大家去完成。

2.3代码

bool Insert(const pair<K, V>& kv)
	{
		if (_root == nullptr)
		{
			_root = new Node(kv);
			_root->_color = 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->_color == Red)
		{
			Node* grandfather = parent->_parent;
			if (parent == grandfather->_left)
			{
				Node* uncle = grandfather->_right;
				//情况一：存在且为红
				if (uncle && uncle->_color == Red)
				{
					//变色
					parent->_color = Black;
					uncle->_color = Black;
					grandfather->_color = Red;
					//向上处理
					cur = grandfather;
					parent = cur->_parent;
				}
				//情况二：不存在or存在且为黑
				else
				{
					//旋转+变色
					//    g
					//  p    u
					//c
					if (cur == parent->_left)
					{
						RotateR(grandfather);
						parent->_color = Black;
						grandfather->_color = Red;
					}
					else
					{
						//情况三
						//    g
						//  p    u
						//     c  
						RotateL(parent);
						RotateR(grandfather);
						cur->_color = Black;
						grandfather->_color = Red;
					}
					break;
				}
			}
			else//p是g的右孩子
			{
				Node* uncle = grandfather->_left;
				//情况一：存在且为红
				if (uncle && uncle->_color == Red)
				{
					//变色
					parent->_color = Black;
					uncle->_color = Black;
					grandfather->_color = Red;
					//向上处理
					cur = grandfather;
					parent = cur->_parent;
				}
				//情况二：不存在or存在且为黑
				else
				{
					//旋转+变色
					//    g
					//  u   p
 					//        c
					if (cur == parent->_right)
					{
						RotateL(grandfather);
						parent->_color = Black;
						grandfather->_color = Red;
					}
					else
					{
						//情况三
						//    g
						//  u   p
						//     c
						RotateR(parent);
						RotateL(grandfather);
						cur->_color = Black;
						grandfather->_color = Red;
					}
					break;
				}
			}
		}
		_root->_color = Black;

		return true;
	}

三 检查函数

	bool Check(Node* cur, int BlackNum, int refBlackNum)
	{
		if (cur == nullptr)
		{
			if (BlackNum != refBlackNum)
			{
				cout << "黑色节点的数量不匹配" << endl;
				return false;
			}
			else
				return true;
		}
		if (cur->_color == Red && cur->_parent->_color == Red)
		{
			cout << "出现了连续的红色节点" << endl;
			return false;
		}
		if (cur->_color == Black)
			++BlackNum;
		return Check(cur->_left, BlackNum, refBlackNum) &&
			Check(cur->_right, BlackNum, refBlackNum);

	}
	bool IsBalance()
	{
		if (_root == nullptr || _root->_color == Red)
			return false;
		//给一个基准值，和其他的黑色节点去比较
		int refBlackNum = 0;
		Node* cur = _root;
		while (cur)
		{
			if (cur->_color == Black)
				++refBlackNum;
			cur = cur->_left;
		}
		return Check(_root, 0, refBlackNum);
	}

四 总结

总的来说红黑树的实现和AVL树非常像，他们就是兄弟，红黑树就是把考虑的因素从平衡因子转变成了颜色。

五 完整代码

#pragma once
#include<assert.h>
#include<vector>
enum Color
{
	Red,
	Black,
};


template<class K, class V>
struct RBTreeNode
{
	RBTreeNode<K, V>* _left;
	RBTreeNode<K, V>* _right;
	RBTreeNode<K, V>* _parent;
	Color _color;
	pair<K, V> _kv;

	RBTreeNode(const pair<K, V>& kv)
		:_left(nullptr)
		, _right(nullptr)
		, _parent(nullptr)
		, _color(Red)
		, _kv(kv)
	{}
};

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->_color = 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->_color == Red)
		{
			Node* grandfather = parent->_parent;
			if (parent == grandfather->_left)
			{
				Node* uncle = grandfather->_right;
				//情况一：存在且为红
				if (uncle && uncle->_color == Red)
				{
					//变色
					parent->_color = Black;
					uncle->_color = Black;
					grandfather->_color = Red;
					//向上处理
					cur = grandfather;
					parent = cur->_parent;
				}
				//情况二：不存在or存在且为黑
				else
				{
					//旋转+变色
					//    g
					//  p    u
					//c
					if (cur == parent->_left)
					{
						RotateR(grandfather);
						parent->_color = Black;
						grandfather->_color = Red;
					}
					else
					{
						//情况三
						//    g
						//  p    u
						//     c  
						RotateL(parent);
						RotateR(grandfather);
						cur->_color = Black;
						grandfather->_color = Red;
					}
					break;
				}
			}
			else//p是g的右孩子
			{
				Node* uncle = grandfather->_left;
				//情况一：存在且为红
				if (uncle && uncle->_color == Red)
				{
					//变色
					parent->_color = Black;
					uncle->_color = Black;
					grandfather->_color = Red;
					//向上处理
					cur = grandfather;
					parent = cur->_parent;
				}
				//情况二：不存在or存在且为黑
				else
				{
					//旋转+变色
					//    g
					//  u   p
 					//        c
					if (cur == parent->_right)
					{
						RotateL(grandfather);
						parent->_color = Black;
						grandfather->_color = Red;
					}
					else
					{
						//情况三
						//    g
						//  u   p
						//     c
						RotateR(parent);
						RotateL(grandfather);
						cur->_color = Black;
						grandfather->_color = Red;
					}
					break;
				}
			}
		}
		_root->_color = Black;

		return true;
	}

	void RotateL(Node* parent)
	{
		//++rotateSize;

		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)
	{
		//++rotateSize;

		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);
	}

	///
	//检查函数
	bool Check(Node* cur, int BlackNum, int refBlackNum)
	{
		if (cur == nullptr)
		{
			if (BlackNum != refBlackNum)
			{
				cout << "黑色节点的数量不匹配" << endl;
				return false;
			}
			else
				return true;
		}
		if (cur->_color == Red && cur->_parent->_color == Red)
		{
			cout << "出现了连续的红色节点" << endl;
			return false;
		}
		if (cur->_color == Black)
			++BlackNum;
		return Check(cur->_left, BlackNum, refBlackNum) &&
			Check(cur->_right, BlackNum, refBlackNum);

	}
	bool IsBalance()
	{
		if (_root == nullptr || _root->_color == Red)
			return false;
		//给一个基准值，和其他的黑色节点去比较
		int refBlackNum = 0;
		Node* cur = _root;
		while (cur)
		{
			if (cur->_color == Black)
				++refBlackNum;
			cur = cur->_left;
		}
		return Check(_root, 0, refBlackNum);
	}
private:
	Node* _root = nullptr;
};


void TestRBTree1()
{
	//int a[] = { 16, 3, 7, 11, 9, 26, 18, 14, 15 };
	int a[] = { 4, 2, 6, 1, 14,16};
	RBTree<int, int> t;
	for (auto e : a)
	{
		if (e == 16)
		{
			int x = 0;
		}
		t.Insert(make_pair(e, e));

		// 1、先看是插入谁导致出现的问题
		// 2、打条件断点，画出插入前的树
		// 3、单步跟踪，对比图一一分析细节原因
		cout << e << "->" << t.IsBalance() << endl;
	}

	t.InOrder();
	cout << t.IsBalance() << endl;
}

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

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

相关文章

SuperMap GIS基础产品FAQ集锦(20240422)

SuperMap GIS基础产品FAQ集锦(20240422)

一、SuperMap iDesktopX 问题1&#xff1a;请问一下&#xff0c;执行“要素数据集平滑”算子报错“Could not find a SpatialRDDProvider to process the params” 11.1.1 【解决办法】使用大数据下的算子配置大数据环境不正确&#xff0c;客户数据量不多&#xff0c;可以使…
阅读更多...
双链向表专题

双链向表专题

1.链表的分类 链表的种类非常多组合起来就有 2 2 8种 链表说明&#xff1a; 虽然有这么多的链表的结构&#xff0c;但是我们实际中最常⽤还是两种结构&#xff1a; 单链表 和 双向带头循环链表 1. 无头单向⾮循环链表&#xff1a;结构简单&#xff0c;⼀般不会单独⽤来存数…
阅读更多...
bugku ezbypass

bugku ezbypass

bugku ezbypass 1.代码审计 2.题目代码如下所示 <?php error_reporting(0); highlight_file(__FILE__);if (isset($_POST[code])) {$code $_POST[code];if (strlen($code) < 105){if (is_string($code)) {if (!preg_match("/[a-zA-Z0-9#%^&*:{}\-<\?>…
阅读更多...
运行Java或Python的时候，Git是必要的吗？

运行Java或Python的时候，Git是必要的吗？

在运行Java或Python代码时&#xff0c;Git并不是必需的&#xff0c;但它可以成为一个非常有用的工具&#xff0c;特别是在团队协作、版本控制和代码管理方面。 Git的作用和优势 版本控制&#xff1a; Git是一个分布式版本控制系统&#xff0c;可以跟踪文件的更改历史&#xff…
阅读更多...
从数据库中到处所有表的列、注释、类型、是否必填等信息

从数据库中到处所有表的列、注释、类型、是否必填等信息

从数据库中到处所有中文表名、英文表名、所有列、注释、类型、长度、是否必填等信息&#xff0c;效果如下&#xff1a; 要实现上面的表格可以直接用SQL实现&#xff0c;实现SQL如下&#xff1a; #查询SQL select* FROMinformation_schema.COLUMNS as columns left join (sele…
阅读更多...
ExcelVBA把当前工作表导出为PDF文档

ExcelVBA把当前工作表导出为PDF文档

我们先问问Kimi Excel导出为PDF的方法有多种&#xff0c;以下是一些常见的方法&#xff1a; 1 使用Excel软件的内置功能&#xff1a; 打开Excel文件&#xff0c;点击“文件”菜单。 选择“另存为”&#xff0c;在“保存类型”中选择“PDF”。 设置保存路径和文件名&#xff…
阅读更多...
Leetcode 119 杨辉三角 II

Leetcode 119 杨辉三角 II

目录 一、问题描述二、示例及约束三、代码方法一&#xff1a;递推方法二&#xff1a;线性递推 四、总结 一、问题描述 给定一个非负索引 rowIndex&#xff0c;返回「杨辉三角」的第 rowIndex 行。   在「杨辉三角」中&#xff0c;每个数是它左上方和右上方的数的和。   自我…
阅读更多...
SpringBoot+layuimini实现左侧菜单动态展示

SpringBoot+layuimini实现左侧菜单动态展示

layuimini左侧菜单动态显示 首先我们看一下layuimini的原有菜单显示格式 {"homeInfo": {"title": "首页","href": "page/welcome-2.html?t2"},"logoInfo": {"title": "LAYUI MINI","…
阅读更多...
苍穹外卖day8（2）用户下单、微信支付

苍穹外卖day8（2）用户下单、微信支付

文章目录 前言一、用户下单1. 业务流程2. 接口设计3. 数据库设计3.1 订单表orders3.2 订单明细表 order_detail 4. 代码实现 二、订单支付 前言 用户下单 因为订单信息中包含了其他业务中的数据&#xff0c;在逻辑处理中涉及了多个其他业务&#xff0c;比如要判断地址簿、购物…
阅读更多...
ThreeJs模拟工厂生产过程一

ThreeJs模拟工厂生产过程一

上节我们已经通过绘制两条长方体和多个圆柱体成功绘制出传送带&#xff0c;今天根据之前的传送带做个简单的工厂车间生产的demo&#xff0c;然后再不断完善它。 首先分析下工厂生产有哪些部分组成&#xff0c;工厂内是产线&#xff0c;产线需要有设备&#xff0c;传送带以及生产…
阅读更多...
要养生也要时尚，益百分满足你的所有需求

要养生也要时尚，益百分满足你的所有需求

要养生也要时尚&#xff0c;益百分满足你的所有需求 艾灸是个好东西&#xff0c;尤其是在近几年的时候&#xff0c;艾灸就像一阵浪潮席卷进了人们的日常生活之中&#xff0c;我们可以在街边看到大大小小的艾灸馆&#xff0c;有些评价比较高的艾灸馆门前甚至还排起了长长的队伍…
阅读更多...
Langchain入门到实战-第四弹

Langchain入门到实战-第四弹

Langchain入门到实战 Langchain中的提示词官网地址Langchain概述Langchain的提示词用法更新计划 Langchain中的提示词 语言模型提示模板是预定义的生成语言模型提示的方法。模板可能包括指令、少样本示例、特定任务的上下文和问题。LangChain 提供了创建和处理提示模板的工具。…
阅读更多...
TS-namespace（命名空间）#记录

TS-namespace（命名空间）#记录

一、命名空间 1、ts 中的 “命名空间” 就是之前的 “内部模块”&#xff0c;任何使用 module 关键字来声明一个内部模块的地方都应该使用 namespace 关键字来替换。 // ts 中的“内部模块” &#xff08;废弃&#xff09; module X { }// ts 中的“命名空间” &#xff08…
阅读更多...
系统启动修复和SYSTEM丢失损坏故障处理

系统启动修复和SYSTEM丢失损坏故障处理

系统启动修复和SYSTEM丢失损坏故障处理 一、问题描述 你的电脑/设备需要修复。无法加载应用程序或操作系统&#xff0c;原因是所需文件丢失或包含错误。 文件:\Windows\system32\winload.exe 错误代码: 0xc000000e 二、问题分析 1.查询winload.exe是win7或者win10以上系统…
阅读更多...
NX二次开发UF_LAYER（图层相关操作）常用函数

NX二次开发UF_LAYER（图层相关操作）常用函数

目录 一、概述 二、函数的介绍 2.1 UF_LAYER_ask_category_info &#xff08;查询图层类别信息&#xff09; 2.2 UF_LAYER_ask_category_tag&#xff08;查询图层类别TAG&#xff09; 2.3 UF_LAYER_ask_status&#xff08;查询图层的状态&#xff09; 2.4 UF_LAYER_ask_wo…
阅读更多...
护眼灯到底有用吗？实用护眼灯十大品牌推荐

护眼灯到底有用吗？实用护眼灯十大品牌推荐

护眼灯有用吗&#xff1f;答案无疑是肯定的。选购到一款合适的护眼台灯&#xff0c;益处多多。这些台灯经过精心设计&#xff0c;运用变频电子镇流器技术&#xff0c;显著提高了光线的频率&#xff0c;使之远超人眼的反应速度&#xff0c;从而有效缓解视觉疲劳。此外&#xff0…
阅读更多...
适用于手机蓝牙的热敏晶体FA1612AS

适用于手机蓝牙的热敏晶体FA1612AS

EPSON推出的一款1612小尺寸无源热敏晶体:FA1612AS。FA1612AS的额定频率为38.4Mhz的晶体单元&#xff0c;采用无铅材料&#xff0c;符合ROHS标准&#xff0c;内置热敏电阻&#xff0c;可用于移动电话&#xff0c;蓝牙等。热敏晶体FA1612AS的产品特性:额定频率:38.4MHZ外部尺寸规…
阅读更多...
Git 仓库内容操作

Git 仓库内容操作

Git 仓库内容操作 | CoderMast编程桅杆Git 仓库内容操作 添加文件到暂存区 使用如下指令将工作区的文件添加到暂存区&#xff0c;告诉 Git 在下次 commit 时哪些文件做出了修改。 commit 指令详看后续 添加一个或多个文件到暂存区&#xff1a; 添加指定目录到暂存区 添加当前目…
阅读更多...
山海鲸电力看板：运维数据一目了然

山海鲸电力看板：运维数据一目了然

在信息化高速发展的今天&#xff0c;电力行业的运维管理也迎来了前所未有的变革。山海鲸可视化智慧电力运维可视化看板&#xff0c;以其独特的数据整合能力和直观的可视化效果&#xff0c;成为了电力行业运维管理的得力助手&#xff0c;为电力的稳定运行提供了强大的技术支撑。…
阅读更多...
namesilo注册与域名购买教程

namesilo注册与域名购买教程

namesilo 是目前价格较便宜的国外域名平台&#xff0c;Paypal、Visa 等多种付款方式&#xff0c;还可以免费使用域名隐私保护&#xff0c;性价比非常之高。 1. 访问namesilo.com并注册用户账号。 邮箱可以填 QQ 邮箱&#xff0c;国家选择 China&#xff0c;注册信息尽量真实。…
阅读更多...
推荐文章
最新文章

Copyright @ 2022~2023