C++红黑树

enum Color
{
RED,BLACK
};
template<class K,class V>
struct RBTreeNode
{
typedef RBTreeNode<K, V> Node;
Node* _left;
Node* _right;
Node* _parent;
pair<K, V> _kv;
Color _col;
//构造函数
RBTreeNode(const pair<K,V>& kv)
:_left(nullptr)
,_right(nullptr)
,_parent(nullptr)
,_kv(kv)
,_col(RED)
{}

};

为什么红黑树的节点默认是红色呢？因为如果插入黑色节点，会违反规则4，但是插入红色节点，只有可能影响规则3（不能有连续的红色节点），调整起来比较方便。

四、红黑树的插入

1. 按照二叉搜索的树规则插入新节点

template<class ValueType>
class RBTree
{
  //……
bool Insert(const ValueType& data)
{
PNode& pRoot = GetRoot();
if (nullptr == pRoot)
{
pRoot = new Node(data, BLACK);
// 根的双亲为头节点
pRoot->_pParent = _pHead;
      _pHead->_pParent = pRoot;
}
else
{
// 1. 按照二叉搜索的树方式插入新节点
      // 2. 检测新节点插入后，红黑树的性质是否造到破坏，
//  若满足直接退出，否则对红黑树进行旋转着色处理
}
// 根节点的颜色可能被修改，将其改回黑色
pRoot->_color = BLACK;
_pHead->_pLeft = LeftMost();
_pHead->_pRight = RightMost();
return true;
}
private:
PNode& GetRoot(){ return _pHead->_pParent;}

private:
PNode _pHead;

2. 检测新节点插入后，红黑树的性质是否造到破坏

因为新节点的默认颜色是红色，因此，：如果其父亲节点是黑色，就没有违反红黑树的性质，则不需要调整，可以直接插入。但是当父亲节点为红色的时候，则违反了性质三（不能有连续的红色节点），此时需要对红黑树进行分类讨论。

约定：cur为当前节点，p为父节点，g为祖父节点，uncle为叔叔节点

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

解决方式：将p,u改为黑，g改为红，然后把g当成cur，继续向上调整。

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

但是对于情况二：我们需要分类讨论，因为cur在parent的左边和右边/parent在祖父的左边和右边所产生的旋转是不同的，一共需要分成4类来讨论

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);
return true;
}
else
{
//找到要插入的位置
Node* cur = _root;
Node* parent = cur->_parent;
while (cur!=nullptr)
{
if (kv.first<cur->_kv.first)
{
parent = cur;
cur = cur->_left;
}
else if (kv.first>cur->_kv.first)
{
parent = cur;
cur = cur->_right;
}
else if (kv.first==cur->_kv.first)
{
return false;
}
}
//cur是连接到parent的左边还是右边呢
cur = new Node(kv);
if (cur->_kv.first<parent->_kv.first)
{
parent->_left = cur;
cur->_parent = parent;
}
else if (cur->_kv.first>parent->_kv.first)
{
parent->_right = cur;
cur->_parent = parent;
}
//根据红黑树的规则进行变色+旋转
//只有当父亲是红色且存在，才需要处理，因为父亲是黑色对上面没有影响可以直接插入
while (parent!=nullptr&&parent->_col==RED)
{
Node* grandparent = parent->_parent;
//parent==grandparent->_left
if (parent==grandparent->_left)
{
Node* uncle = grandparent->_right;
//叔叔存在且为红色
if (uncle!=nullptr&&uncle->_col==RED)
{
//变色
parent->_col = uncle->_col = BLACK;
grandparent->_col = RED;
//grandparent给cur继续向上调整
cur = grandparent;
parent = grandparent->_parent;
}
//叔叔不存在或者存在且为黑色
else
{
if (cur==parent->_left)
{
//右单旋
_RotateR(grandparent);
//变色
grandparent->_col = RED;
parent->_col = BLACK;

}
else
{
//双旋
_RotateL(parent);
_RotateR(grandparent);
cur->_col = BLACK;
grandparent->_col = RED;
}
//因为走到else结束后，parent右一种情况是是黑色，有一种情况是红色，但是都以及调整好了，
//不需要再往上调整，所以直接break
break;
}
}
//parent==grandparent->_right
else
{
Node* uncle = grandparent->_left;
if (uncle!=nullptr&&uncle->_col==RED)
{
//变色
uncle->_col = parent->_col = BLACK;
grandparent->_col = RED;
//继续往上处理
cur = grandparent;
parent = grandparent->_parent;
}
else
{
if (cur==parent->_right)
{
//左旋
_RotateL(grandparent);
parent->_col = BLACK;
grandparent->_col = RED;
}
else
{
//双旋
_RotateR(parent);
_RotateL(grandparent);
//变色
cur->_col = BLACK;
grandparent->_col = RED;

}
break;
}
}
}
}
//根必须要是黑色
_root->_col = BLACK;
return true;
}



//右单旋
void _RotateR(Node* parent)
{
//记录两个重要节点
Node* subL = parent->_left;
Node* subLR = subL->_right;

//连接
parent->_left = subLR;
//如果subL的右边为空，空不能解引用，需要注意一下
if (subLR->_right != nullptr)
{
subLR->_parent = parent;
}
subL->_right = parent;
//处理根的问题
Node* grandparent = parent->_parent;

parent->_parent = subL;
subL->_parent = grandparent;
//如果原本parent是根，则让_root=subL
if (parent == _root)
{
_root = subL;
subL->_parent = nullpptr;
}
//如果parent是子树，则连接subL和grandparent
else
{
if (grandparent->_right == parent)
{
grandparent->_right = subL;
}
else if (grandparent->_left == parent)
{
grandparent->_left = subL;
}
}

}
//左单旋
void _RotateL(Node* parent)
{
Node* subR = parent->_right;
Node* subRL = subR->_left;

parent->_right = subRL;
subR->_left = parent;

Node* grandparent = parent->_parent;
parent->_parent = subR;
if (subRL != nullptr)
{
subRL->_parent = parent;
}
//处理根的问题
if (parent == _root)
{
_root = subR;
subR->_parent = nullptr;
}
else
{
if (parent == grandparent->_right)
{
grandparent->_right = subR;
}
else if (parent == grandparent->_left)
{
grandparent->_left = subR;
}
}

}
private:
Node* _root = nullptr;
};
}

五、红黑树的验证

红黑树的验证检测分为两步：

（1）检测其是非满足二叉搜索树的规则（中序是否为有序）

（2）检测其是否满足红黑树的性质

bool isBalance()
{
if (_root==nullptr)
{
return true;
}
if (_root->_col==RED)
{
return false;
}

//求参考值
int refval = 0;
Node* cur = _root;
while (cur!=nullptr)
{
if (cur->_col==BLACK)
{
refval++;
}
cur = cur->_left;
}
return Check(_root, 0, refval);
}
//检查颜色是否符合规则和黑色节点的数量是否相同
//blacknum是根节点到当前节点路径上的黑色节点数量
bool Check(Node* root,int blacknum,const int refVal)
{
            //走到空的时候说明该路径已经走完了，比对balcknum的值
if (root==nullptr)
{
if (blacknum!=refVal)
{
cout << "存在不同路径的黑色节点数量不相等" << endl;
return false;
}
return true;
}
//因为检测孩子不好检测，有多种情况，所以我们反过来从孩子检测父亲的颜色
if (root->_col==RED&& root->_parent->_col == RED)
{
cout << "有连续的红色节点" << endl;
return false;
}
if (root->_col==BLACK)
{
blacknum++;
}
return Check(root->_left,blacknum,refVal)
&& Check(root->_right,blacknum,refVal);
}

六、红黑树和AVL树的比较

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

红黑树常常因为其高效的性能，而被用到map/set等数据结构中。

原文地址：https://blog.csdn.net/2303_79336820/article/details/142344424

免责声明：本站文章内容转载自网络资源，如本站内容侵犯了原著者的合法权益，可联系本站删除。更多内容请关注自学内容网（zxcms.com）！

上一篇：虚幻引擎解决构建问题
下一篇：项目第十一弹：客户端模块设计与实现

设计模式介绍
观察者模式（Observer）：定义对象间的一种一对多的依赖关系，以便当一个对象的状态发生改变时，所有依赖于它的对象都得到通知并被自动更新。外观模式（Facade）：为子系统中的一组接口提供一个一致的
阅读更多2024-09-24
《Effective Debugging：软件和系统调试的66个有效方法》读书笔记-Part2
编程技术：代码评审、审读代码、调试机制、日志、单元测试、断言、改动受测程序验证推想、缩小正确范例与错误代码之间的差距、简化可疑代码、将可疑代码改用另外一种编程语言来写、改善可疑代码可读性、清除bug根
阅读更多2024-09-24
单片机学到什么程度才可以去工作？
如果独立写，对你来说，估计很吃力，这个过程比较考验你的综合能力，比如对单片机外设的熟练度，实现功能的思路，实现思路的代码能力，最后是整合代码，怎么避免一些功能冲突的BUG。刚开始，不一定非要具备独立写
阅读更多2024-09-24
VirtualBox 克隆已有的虚拟机
VirtualBox 克隆已有的虚拟机 CentOS
阅读更多2024-09-24
鲲鹏计算这五年：硬生态基本盘稳住，才能放手进击软生态
鲲鹏产业生态5年，神州鲲泰一路生花的进击
阅读更多2024-09-24
第四章：存储系统：第一节——存储器概述
以下图片，我们在第一章介绍过。■ 现代计算机以存储器为中心，它是计算机中存放指令和数据的主要部件。存储器的容量越大，能存储的信息越多。提高存储系统的访问速度，是提高计算机处理信息速度的重要措施。■因此
阅读更多2024-09-24
网络通信——NAT（网络地址转换）
一.NAT是什么1.首先运营商维护的网络时公网的，使用的是公网的IP地址，而私有的IP地址的不能的在公网上路由，为了保证网络互通，所以有了NAT技术。2.NAT一般部署在公网和私网上的网关设备上面。二
阅读更多2024-09-24
rsync+inotify
rsync是linux系统下的数据镜像备份工具。使用快速增量备份工具Remote Sync可以远程同步。它使用一种称为“Rsync演算法”的技术，只传输文件的不同部分，而不是每次都整份传送，这使得它在
阅读更多2024-09-24
得物App荣获新奖项，科技创新助力高质量发展
在此次服贸会上，得物App作为科技创新与品质消费的杰出代表，受邀参会并荣获“科技创新服务示范案例”奖项，这是国家层面对得物App以科技创新保障品质消费、提升消费体验成效的肯定。在产业转型和消费升级的大
阅读更多2024-09-24
使用Python实现图形学曲线和曲面的B样条曲线算法
B样条曲线是由控制点和节点向量（Knot vector）定义的。给定 n + 1 个控制点P0P1PnP0P1...Pn和一个节点向量tttCt∑i0nNiptPiCti0∑nNiptPi
阅读更多2024-09-24

C++红黑树

一、红黑树的概念

二、红黑树的性质

三、红黑树节点的定义