带你手撕红黑树! c++实现 带源码

目录

一、概念

二、特性

三、接口实现

1、插入

情况一:p为黑,结束

情况二:p为红

1)叔叔存在且为红色

2)u不存在/u存在且为黑色

(1)p在左,u在右

(2)u在左,p在右

2、检查平衡

四、对红黑树的理解

五、原码


一、概念

红黑树:AVL树不好控制(严格平衡),所以推出了红黑树
不用高度控制平衡,用颜色
最长路径<= 最短路径*2
红黑树是近似平衡

二、特性

1、每个节点不是红色就是黑色
2、根节点是黑色的
3、如果一个节点是红色的,则他的两个孩子是黑色的(不存在连续的红色节点)
4、如果对于每个节点,从该节点到其后代节点的简单路径上,均包含相同数目的黑色节点(每条路径的黑色节点数量相等)
5、每个叶子节点都是黑色的(叶子节点指的是空节点)

最短路径:全黑
最长路径:一黑一红

三、接口实现

1、插入

说明:p为父,cur为当前节点,u为叔叔节点,g为祖父节点

情况一:p为黑,结束

情况二:p为红

1)叔叔存在且为红色

怎么办?
将p和u变黑,g变红    
    (1)如果g是根节点,再变为黑色


    (2)如果g不是根,继续往上调整 g变cur, parent = g->parent


        有可能会一路更新到根节点,即父节点不存在
        c++库内部的处理是,直接将parent作为while循环条件之一
        然后,在单次数循环结束位置将parent置为黑
        保证parent为黑

2)u不存在/u存在且为黑色

p改黑,g改红,再旋转

(1)p在左,u在右

情况1
//        g
//      p          u
//  c
//
p在左,u在右:以g进行右单旋
p变黑,g变红
 
情况2
//        g
//      p          u
//            c
//
c在右:以p进行左单旋变为情况1
//        g
//      c          u
//  p
//
需修改p和c位置
//        g
//      p          u
//  c
//
再以g右单旋转
p变黑,g变红

(2)u在左,p在右

情况1
//        g
//      u          p
//                  c
//
以g进行左单旋
p变黑,g变红
 


情况2
//        g
//      u          p
//                   c
//
以p进行右单旋变为情况1
//        g
//      u          c
//                p
//
需修改p和c位置
//        g
//      u          p
//                 c
//
再以g右单旋转
p变黑,g变红

2、检查平衡


计算每一条路径的黑色节点的个数
检查是否有连续红色节点
递归
走到空的时候,说明该路径走到头了


p为红之后,就要判断p是左边还是右边
即:u在左,p在右 和 u在右,p在左
就是在p为红色的同时话要细分为两种大情况
然后对于内部还要进行u的判断

四、对红黑树的理解

红黑树的核心,是保证最长路径的长度不超过最短路径的两倍
怎么做到整个特性呢?
通过维持其四个特性
尤其是特性三和特性四
所以,在插入的时候,就要考虑不能打破特性3和特性4
特性3是不能有连续的红色节点
特性4是所有路径的黑色节点个数相同
相比之下,维持特性4明显要比特性3更加严格
维持成本更高,同时也更加难以控制
所以,在插入的时候,为了便于控制和成本
我们选择插入红色节点
剩下的问题,就是要怎么避免出现连续两个红色节点
如果插入的时候,父节点就已经是一个黑色节点
那么,直接插入,此时不会出现连续两个红色节点
同时,这个条路径的黑色节点个数也没有发生变化
但是,如果插入的父节点是一个红色节点呢?
问题来了
怎么办?
父节点是红色,插入的也是红色
只能有一个变黑色
谁变?
新插入节点吗?
如果新插入节点是黑色,那么插入路径黑色节点个数就增大了
就要去维护其他的所有路径的
何其恐怖
所以,只能父节点变黑色
同时,如果父亲有兄弟,就是叔叔节点存在
父节节点变为黑色,父亲节点的路径黑色节点多了一个
那么作为另外一条路径的叔叔节点,也必须变为黑色,也增加一个黑色节点,才能保持
而,父节点的父节点,即祖父节点一定存在且为黑色
因为父节点和叔叔节点(如果存在)已经变为黑色
那么,对于祖父作为根节点的这课子树来说,多了一个黑色节点
因此,祖父节点必须变为红色
以保持平衡
如此,以祖父节点作为根节点的这棵子树已经保持了黑色节点数量不变
但是因为祖父节点已经变为了红色,需要继续往上更改颜色

五、原码

#pragma once
#include<vector>
#include<iostream>
using namespace std;enum Colour{BLACK,RED};template<class K, class V>struct BRTreeNode{BRTreeNode<K, V>* _parent;BRTreeNode<K, V>* _right;BRTreeNode<K, V>* _left;pair<K, V> _kv;Colour _col;BRTreeNode(const pair<K, V>& kv):_parent(nullptr), _right(nullptr), _left(nullptr), _kv(kv), _col(RED){}};template<class K, class V>class BRTree{typedef BRTreeNode<K, V> Node;public:bool Insert(const pair<K, V>& kv){if (_root == nullptr){_root = new Node(kv);_root->_col = BLACK;return true;}Node* cur = _root;Node* parent = nullptr;while (cur){if (kv.first < cur->_kv.first){parent = cur;cur = cur->_left;}else if (kv.first > cur->_kv.first){parent = cur;cur = cur->_right;}else//找到相等key{return false;}}cur = new Node(kv);cur->_col = RED;if (kv.first < parent->_kv.first)//插入左{parent->_left = cur;}else //插入右{parent->_right = cur;}cur->_parent = parent;//插入之后,要进行颜色调整while (parent && parent->_col == RED)//如果为空/黑色节点,直接结束{//Node* grandfather = parent->_parent;if (parent == grandfather->_left)//p为左,u为右{Node* uncle = grandfather->_right;//如果叔叔存在,且为红色if (uncle && uncle->_col == RED){//修改颜色parent->_col = uncle->_col = BLACK;grandfather->_col = RED;//向上更新cur = grandfather;parent = cur->_parent;}else//叔叔不存在/叔叔存在且为黑色{if (cur == parent->_left){//		   g//	   p      u//  c//RotateR(grandfather);parent->_col = BLACK;grandfather->_col = RED;}else{//		   g//	   p      u//      c//RotateL(parent);//		   g//	   c      u//  p//RotateR(grandfather);cur->_col = BLACK;grandfather->_col = RED;}break;}}else//p为右,u为左{Node* uncle = grandfather->_left;//如果叔叔存在,且为红色if (uncle && uncle->_col == RED){//修改颜色parent->_col = uncle->_col = BLACK;grandfather->_col = RED;//向上更新cur = grandfather;parent = cur->_parent;}else//叔叔不存在/叔叔存在且为黑色{if (cur == parent->_right){//		   g//	   u      p//					c//RotateL(grandfather);parent->_col = BLACK;grandfather->_col = RED;}else{//		   g//	   u      p//          c//RotateR(parent);//		   g//	   u      c//  				p//RotateL(grandfather);cur->_col = BLACK;grandfather->_col = RED;}break;}}}_root->_col = BLACK;return true;}//右旋void RotateR(Node* parent){Node* subL = parent->_left;Node* subLR = subL->_right;parent->_left = subLR;if (subLR)//subLR可能为空{subLR->_parent = parent;}subL->_right = parent;Node* ppNode = parent->_parent;parent->_parent = subL;//注意修改顺序if (parent == _root){_root = subL;_root->_parent = nullptr;}else{if (ppNode->_left == parent){ppNode->_left = subL;}else{ppNode->_right = subL;}subL->_parent = ppNode;}}//左旋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;_root->_parent = nullptr;}else{if (ppNode->_left == parent){ppNode->_left = subR;}else{ppNode->_right = subR;}subR->_parent = ppNode;}}//检查平衡bool isBalance(){if (_root->_col == RED){return false;}//找到任意一条路黑色节点个数Node* cur = _root;int refNum = 0;while (cur){if (cur->_col == BLACK){refNum++;}cur = cur->_left;}return Check(_root, 0,  refNum);return 1;}void Inoder(){_Inoder(_root);cout << endl;}private:bool Check(Node* root,int blackNum,const int refNum){//到路径结束位置检查黑色节点if (root == nullptr){if (blackNum != refNum){cout << "黑色节点不相等" << endl;return false;}// << blackNum << endl;return true;}//检查红色节点if (root->_col == RED && root->_parent->_col == RED){cout << root->_kv.first << "连续红节点" << endl;return false;}if (root->_col == BLACK){blackNum++;}return Check(root->_left, blackNum, refNum)&& Check(root->_right, blackNum, refNum);}void _Inoder(const Node* root){if (root == nullptr){return;}_Inoder(root->_left);cout << root->_kv.first << ":" << _root->_kv.second << endl;_Inoder(root->_right);}private:Node* _root = nullptr;};void BRTreeTest1(){int a[] = { 8, 3, 1, 10, 6, 4, 7, 14, 13 };int b[] = { 4, 2, 6, 1, 3, 5, 15, 7, 16, 14,8, 3, 1, 10, 6, 4, 7, 14, 13 };BRTree<int, int> t;for (auto e : b){t.Insert({ e,e });}t.Inoder();int ret = t.isBalance();cout << ret << endl;}void BRTreeTest2(){int n = 10000000;//1000万个节点进行测试srand(time(0));vector<int> v;v.reserve(n);for (int i = 0; i < n; ++i){v.push_back(rand() + i);}size_t T1 = clock();BRTree<int, int> t;for (auto e : v){t.Insert(make_pair(e, e));}size_t T2 = clock();cout << "insert:" << T2 - T1 << endl;int ret = t.isBalance();cout << ret << endl;}

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

如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈email:809451989@qq.com,一经查实,立即删除!

相关文章

MySQL中JOIN连接的实现算法

目录 嵌套循环算法&#xff08;NLJ&#xff09; 简单嵌套循环&#xff08;SNLJ&#xff09; 索引嵌套循环&#xff08;INLJ&#xff09; 块嵌套循环&#xff08;BNLJ&#xff09; 三种算法比较 哈希连接算法&#xff08;Hash Join&#xff09; 注意事项&#xff1a; 工…

异常处理/CC++ 中 assert 断言 应用实践和注意事项

文章目录 概述assert 本质浅析Release版本下的assert是否生效默认设置下 QtCreator环境 assert 过程默认配置下 VS环境 assert 过程配置VS发布模式下的断言生效VS环境Release版本的UI程序Release下请当我不生效 请勿滥用assert导致逻辑错误再强调不要在assert内执行逻辑功能怎敢…

【UnityRPG游戏制作】Unity_RPG项目_PureMVC框架应用

&#x1f468;‍&#x1f4bb;个人主页&#xff1a;元宇宙-秩沅 &#x1f468;‍&#x1f4bb; hallo 欢迎 点赞&#x1f44d; 收藏⭐ 留言&#x1f4dd; 加关注✅! &#x1f468;‍&#x1f4bb; 本文由 秩沅 原创 &#x1f468;‍&#x1f4bb; 收录于专栏&#xff1a;就业…

Vue-watch监听器

监听器 watch侦听器&#xff08;监视器&#xff09;简单写法完整写法 watch侦听器&#xff08;监视器&#xff09; 作用&#xff1a;监视数据变化&#xff0c;执行一些业务逻辑或异步操作 语法&#xff1a; watch同样声明在跟data同级的配置项中简单写法&#xff1a; 简单类型…

C++ 中的 lambda 表达式

1.概念 lambda表达式实际上是一个匿名类的成员函数&#xff0c;该类由编译器为lambda创建&#xff0c;该函数被隐式地定义为内联。因此&#xff0c;调用lambda表达式相当于直接调用匿名类的operator()函数&#xff0c;这个函数可以被编译器内联优化&#xff08;建议&#xff0…

地图涟漪效果

参考API echarts图表集 useEcharts.js import { onBeforeUnmount, onDeactivated } from "vue"; // import * as echarts from "echarts";/*** description 使用 Echarts (只是为了添加图表响应式)* param {Element} myChart Echarts实例 (必传)* param …

AcWing-168生日蛋糕-搜索/剪枝

题目 思路 表面积和体积公式&#xff1a;以下分析参考自&#xff1a;AcWing 168. 生日蛋糕【图解推导】 - AcWing&#xff1b;AcWing 168. 关于四个剪枝的最清楚解释和再次优化 - AcWing 代码 #include<iostream> #include<cmath> using namespace std;const in…

【爬虫基础1.1课】——requests模块上

目录索引 requests模块的作用&#xff1a;实例引入&#xff1a; 特殊情况&#xff1a;锦囊1&#xff1a;锦囊2: 这一个栏目&#xff0c;我会给出我从零开始学习爬虫的全过程。感兴趣的小伙伴可以关注一波&#xff0c;用于复习和新学都是不错的选择。 那么废话不多说&#xff0c…

C语言学习(九)多文件编程 存储类型 结构体

目录 一、多文件编程&#xff08;一&#xff09;不写头文件的方方式进行多文件编程 &#xff08;二&#xff09;通过头文件方式进行多文件编程&#xff08;1&#xff09;方法&#xff08;2&#xff09;头文件守卫 &#xff08;三&#xff09; 使用多文件编程实现 - * / 功能 二…

HC-06 蓝牙串口从机 AT 命令详解

HC-06 蓝牙串口从机 AT 命令详解 要使用 AT 命令&#xff0c;首先要知道 HC-06 的波特率&#xff0c;然后要进入 AT 命令模式。 使用串口一定要知道三要素&#xff0c;一是波特率&#xff0c;二是串口号&#xff0c;三是数据格式, HC-06只支持一种数据格式: 数据位8 位&#…

HTTP 连接详解

概述 世界上几乎所有的 HTTP 通信都是由 TCP/IP 承载的&#xff0c;客户端可以打开一条TCP/IP连接&#xff0c;连接到任何地方的服务器。一旦连接建立&#xff0c;客户端和服务器之间交换的报文就永远不会丢失、受损或失序 TCP&#xff08;Transmission Control Protocol&…

97. 交错字符串-----回溯、动态规划

题目链接 97. 交错字符串 - 力扣&#xff08;LeetCode&#xff09; 解答 递归回溯 题目所述为两个字符串交替组成第三个字符串&#xff0c;之前好像做过相似的题目&#xff0c;直接联想到可以考虑使用递归回溯的做法&#xff0c;让字符串s1和字符串s2分别作为起始字符串&…

Mybatis-Plus大批量插入数据到MySQL

MyBatis-Plus的saveBatch方法 GetMapping("/save1") public void save1() {// 数据准备List<MallOrder> orderList getMallOrderList();// mybatis-pluslong start System.currentTimeMillis();mallOrderService.saveBatch(orderList);System.out.println(&…

计算机服务器中了360后缀勒索病毒怎么解密,360后缀勒索病毒恢复

计算机网络技术的不断发展与应用&#xff0c;为企业的生产运营提供了极大便利&#xff0c;大大提高了企业的办公效率&#xff0c;为企业的生产运营注入了新的动力&#xff0c;但网络是一把双刃剑&#xff0c;在为企业提供便利的同时&#xff0c;也为企业的数据安全带来严重威胁…

google test 使用指南

目录 测试项目 calculator.h calculator.cpp test01.cpp 创建新项目 选择Google Test 选择要测试的项目 pch.cpp 加入依赖 设为启动项目 ​编辑 运行 ​编辑 关键点 测试项目 calculator.h #ifndef __CALCULATOR_H__ #define __CALCULATOR_H__#include <i…

Linux操作系统中管理磁盘的另外一种操作方式。即LVM——逻辑卷管理操作

在Linux操作系统中管理磁盘的一种方法名称——LVM&#xff0c;这种管理磁盘的优势。 1.使用LVM去管理磁盘可以在不影响原来数据的前提下去扩容磁盘空间或者是缩减磁盘空间。 在LVM中除了上层逻辑券可以扩容&#xff0c;下层的券组也可以扩容。 2.使用LVM管理的磁盘支持快照功…

MySQL中的子查询

子查询,在一个查询语句中又出现了查询语句 子查询可以出现在from和where后面 from 表子查询(结果一般为多行多列)把查询结果继续当一张表对待 where 标量子查询(结果集只有一行一列)查询身高最高的学生,查询到一个最高身高 列子查询(结果集只有一行多列) 对上表进行如下操作 …

韩顺平0基础学Java——第10天

p202-233 类与对象&#xff08;第七章&#xff09; 成员方法 person类中的speak方法&#xff1a; 1.public表示方法是公开的 2.void表示方法没有返回值 3.speak&#xff08;&#xff09;中&#xff0c;speak表示方法名&#xff0c;括号是形参列表。 4.大括号为方法体&am…

WPF之多种视图切换

1&#xff0c;View切换&#xff0c;效果呈现 视图1 视图2 视图3 2&#xff0c;在Xaml中添加Listview控件&#xff0c;Combobox控件。 <Grid ><Grid.RowDefinitions><RowDefinition Height"143*"/><RowDefinition Height"30"/>&l…

Leetcode经典题目之用队列实现栈

P. S.&#xff1a;以下代码均在VS2019环境下测试&#xff0c;不代表所有编译器均可通过。 P. S.&#xff1a;测试代码均未展示头文件stdio.h的声明&#xff0c;使用时请自行添加。 目录 1、题目展示2、题目分析3、完整代码演示4、结语 1、题目展示 前面我们了解过如何实现队列…