红黑树的概念
红黑树,是一种二叉搜索树,但在每个结点上增加一个存储位表示结点的颜色,可以是Red或
Black。 通过对任何一条从根到叶子的路径上各个结点着色方式的限制,红黑树确保没有一条路
径会比其他路径长出俩倍,因而是接近平衡的。
红黑树的性质
- 每个结点不是红色就是黑色
- 根节点是黑色的
- 如果一个节点是红色的,则它的两个孩子结点是黑色的
- 对于每个结点,从该结点到其所有后代叶结点的简单路径上,均 包含相同数目的黑色结点
- 每个叶子结点都是黑色的(此处的叶子结点指的是空结点)
红黑树节点的定义
enum color
{
RED,
BLACK,
};
template<class Key, class Val>
struct RBTreeNode
{
RBTreeNode<Key, Val>* _left;
RBTreeNode<Key, Val>* _right;
RBTreeNode<Key, Val>* _parent;
pair<Key, Val> _kv;
color _clo;
//颜色
RBTreeNode(const pair<Key, Val>& kv)
:_left(nullptr)
, _right(nullptr)
, _parent(nullptr)
, _kv(kv)
, _clo(RED)
{}
};
红黑树的插入(难点)
-
按照二叉搜索的树规则插入新节点
搜索二叉树 -
检测新节点插入后,红黑树的性质是否造到破坏
因为新节点的默认颜色是红色,因此:如果其双亲节点的颜色是黑色,没有违反红黑树任何
性质,则不需要调整;但当新插入节点的双亲节点颜色为红色时,就违反了性质三不能有连
在一起的红色节点,此时需要对红黑树分情况来讨论:
约定:cur为当前节点, p为父节点, g为祖父节点, u为叔叔节点
情况1:cur为红,p为红,g为黑,u存在且为红
将p,u,g的颜色进行调整
调整后
如果g的父亲节点为红则需继续向上调整
uncle->_clo = parent->_clo = BLACK;
grandparent->_clo = RED;
cur = grandparent;
parent = cur->_parent;
情况2:cur为红, p为红, g为黑, u不存在/u存在且为黑
cur为p的左孩子,p为g的左孩子
左右单旋看AVL 树的插入:左右单旋
先将g进行右单旋
再将p改为黑,g改为红
//grandparent->_left == parent
if (uncle == nullptr || uncle->_clo == BLACK)
{
if (cur == parent->_left)
{
RotateR(grandparent);
grandparent->_clo = RED;
parent->_clo = BLACK;
if (_root == grandparent)
{
_root = parent;
}
}
如果cur在右边,则进行左单旋
//grandparent->_right == parent
if (uncle == nullptr || uncle->_clo == BLACK)
{
if (cur == parent->_right)
{
RotateL(grandparent);
grandparent->_clo = RED;
parent->_clo = BLACK;
if (_root == grandparent)
{
_root = parent;
}
}
}
情况3: cur为红, p为红, g为黑, u不存在/u存在且为黑
p为g的左孩子,cur为p的右孩子
先将p进行左单旋,得到情况2,再跟情况2处理的方式一样
//cur==parent->_right
//grandparent->_left == parent
else
{
RotateL(parent);
RotateR(grandparent);
grandparent->_clo = RED;
cur->_clo = BLACK;
if (_root == grandparent)
{
_root = cur;
}
}
p为g的右孩子
//cur == parent->_left
//grandparent->_right == parent
else
{
RotateR(parent);
RotateL(grandparent);
grandparent->_clo = RED;
cur->_clo = BLACK;
if (_root == grandparent)
{
_root = cur;
}
}