一.红黑树之简介
1.红黑树的定义
红黑树,是一种二叉搜索树,但在每个结点上增加一个存储位表示结点的颜色,可以是Red或 Black。 通过对任何一条从根到叶子的路径上各个结点着色方式的限制,红黑树确保没有一条路径会比其他路径长出俩倍(即是最长路径不超过最短路径的两倍)因而是接近平衡的。
2.红黑树的性质
1. 每个结点不是红色就是黑色2. 根节点是黑色的3. 如果一个节点是红色的,则它的两个孩子结点是黑色的4. 对于每个结点,从该结点到其所有后代叶结点的简单路径上,均包含相同数目的黑色结点5. 每个叶子结点都是黑色的(此处的叶子结点指的是空结点)
二.红黑树之结构
1.结构定义
与普通二叉树不同的是,红黑树拥有一个指向父节点的指针,并且多了一个表明颜色的节点数据
enum Color
{
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;
Color col;
RBTreeNode(const pair<K,V>& kv)
: _kv(kv)
, _left(nullptr)
, _right(nullptr)
, _parent(nullptr)
, col(RED) //默认节点是红色
{}
};
template<class K, class V>
class RBTree
{
typedef struct RBTreeNode<K,V> Node;
public:
RBTree()
:_root(nullptr)
{}
private:
Node* _root;
};
三.红黑树之插入
1.构建搜索二叉树
红黑树也是搜索二叉树的一个分支,自然也满足搜索二叉树的规则。这里需要注意的是,如果创建的是根节点,则把节点颜色改为黑色,如果不是,就开始遍历二叉树,要插入的节点小于当前节点, 继续往左搜索,反之则往右搜索。
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->_left;
}
else if (cur->_kv.first < kv.first)
{
parent = cur;
cur = cur->_right;
}
else
{
return false;
}
}
cur = new Node(kv);
cur->col = RED;
if (parent->_kv.first < kv.first)
{
parent->_right = cur;
}
else
{
parent->_left = cur;
}
cur->_parent = parent;
2.修正红黑树【变色+旋转】
插入完成之后,需要修正二叉树使其满足红黑树的规则。
这里要注意几种情况:
①.cur为红,parent为红,grandfather为黑,uncle存在且为红
此时插入cur节点, cur和parent都为红,违背了红黑树的性质3,则需要进行修正。
这种情况首先找到cur的叔叔节点,如果叔叔节点存在,就将叔叔节点和父亲节点都变成黑色,grandfather变红。
之后再继续往上调整。
②.cur为红,parent为红,grandfather为黑,uncle不存在/uncle存在且为黑
*而当uncle不存在时,cur一定是新插入节点,如果不是,不满足性质4
*而当uncle存在时,那uncle也一定是黑色的,那么cur也必须是黑色的,但现在变成了红色是因为之前在下面调整的时候,cur变成了红色
此时这种情况,cur是parent的左孩子,将grandfather右旋,把parent的颜色改成黑色,grandfather的颜色改成红色
但如果是这种情况,cur是parent的右孩子,此时右旋已经解决不了问题,需要用到左右旋,之后再将cur的颜色改为黑色,grandfather的颜色改为红色
3.红黑树的旋转
①.右旋
先来看一个简易版的:
进阶版:
总结一下:
右旋即是parent节点的左连接cur的右孩子,parent再成为cur的右孩子
void RotateR(Node* parent)
{
Node* cur = parent->_left;
Node* ppnode = parent->_parent; //parent可能有父亲,保存以备后续连接
Node* curleft = cur->_left;
if (curleft)
parent->_left = curleft;
cur->_right = parent;
parent->_parent = cur;
if (_root == parent) //parent就是根,不用连接
{
cur = _root;
}
else
{
if (ppnode->_left == parent) //parent不是根,需要连接
{
ppnode->_left = cur;
}
else
{
ppnode->_right = cur;
}
}
cur->_parent = ppnode;
}
②.左旋
左旋类似于右旋。
即是:parent节点的右连接cur的左孩子,parent再成为cur的左孩子
void RotateL(Node* parent)
{
Node* cur = parent->_right;
Node* curleft = cur->_left;
Node* ppnode = parent->_parent;
parent->_right = curleft;
if (curleft)//如果curleft存在再变,空指针问题
{
curleft->_parent = parent;
}
cur->_left = parent; //cur的左连接parent
parent->_parent = cur; //更新parent的父亲
if (parent == _root) //更新根节点
{
_root = cur;
}
else
{
if (ppnode->_left == parent)
{
ppnode->_left = cur;
}
else
{
ppnode->_right = cur;
}
}
cur->_parent = ppnode;
}
4.红黑树的插入代码
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);
cur->_col = RED;
if (parent->_kv.first < kv.first)
{
parent->_right = cur;
}
else
{
parent->_left = cur;
}
cur->_parent = parent;
while (parent && parent->_col == RED)
{
Node* grandfather = parent->_parent;
if (parent == grandfather->_left)
{
Node* uncle = grandfather->_right;
// u存在且为红
if (uncle && uncle->_col == RED)
{
// 变色
parent->_col = uncle->_col = BLACK;
grandfather->_col = RED;
// 继续向上处理
cur = grandfather;
parent = cur->_parent;
}
else // u不存在 或 存在且为黑
{
if (cur == parent->_left)
{
RotateR(grandfather);
parent->_col = BLACK;
grandfather->_col = RED;
}
else
{
RotateL(parent);
RotateR(grandfather);
cur->_col = BLACK;
grandfather->_col = RED;
}
break;
}
}
else // parent == grandfather->_right
{
Node* uncle = grandfather->_left;
// u存在且为红
if (uncle && uncle->_col == RED)
{
// 变色
parent->_col = uncle->_col = BLACK;
grandfather->_col = RED;
// 继续向上处理
cur = grandfather;
parent = cur->_parent;
}
else
{
if (cur == parent->_right)
{
RotateL(grandfather);
grandfather->_col = RED;
parent->_col = BLACK;
}
else
{
RotateR(parent);
RotateL(grandfather);
cur->_col = BLACK;
grandfather->_col = RED;
}
break;
}
}
}
_root->_col = BLACK;
return true;
}
void RotateL(Node* parent)
{
Node* cur = parent->_right;
Node* curleft = cur->_left;
Node* ppnode = parent->_parent;
parent->_right = curleft;
if (curleft)//如果curleft存在再变,空指针问题
{
curleft->_parent = parent;
}
cur->_left = parent; //cur的左连接parent
parent->_parent = cur; //更新parent的父亲
if (parent == _root) //更新根节点
{
_root = cur;
}
else
{
if (ppnode->_left == parent)
{
ppnode->_left = cur;
}
else
{
ppnode->_right = cur;
}
}
cur->_parent = ppnode;
}
void RotateR(Node* parent)
{
Node* cur = parent->_left;
Node* ppnode = parent->_parent;
Node* curleft = cur->_left;
if (curleft)
parent->_left = curleft;
cur->_right = parent;
parent->_parent = cur;
if (_root == parent)
{
cur = _root;
}
else
{
if (ppnode->_left == parent)
{
ppnode->_left = cur;
}
else
{
ppnode->_right = cur;
}
}
cur->_parent = ppnode;
}
四.红黑树的验证
根据红黑树的五条性质一一检查即可
bool CheckColour(Node* root, int blacknum, int benchmark)
{
if (root == nullptr)
{
if (blacknum != benchmark)
return false;
return true;
}
if (root->_col == BLACK)
{
++blacknum;
}
if (root->_col == RED && root->_parent && root->_parent->_col == RED)
{
cout << root->_kv.first << "出现连续红色节点" << endl;
return false;
}
return CheckColour(root->_left, blacknum, benchmark)
&& CheckColour(root->_right, blacknum, benchmark);
}
bool IsBalance()
{
return IsBalance(_root);
}
bool IsBalance(Node* root)
{
if (root == nullptr)
return true;
if (root->_col != BLACK)
{
return false;
}
// 基准值
int benchmark = 0;
Node* cur = _root;
while (cur)
{
if (cur->_col == BLACK)
++benchmark;
cur = cur->_left;
}
return CheckColour(root, 0, benchmark);
}
五.红黑树的高度
与二叉树一样的思路,分治思想进行计数
int Height()
{
return Height(_root);
}
int Height(Node* root)
{
if (root == nullptr)
return 0;
int leftHeight = Height(root->_left);
int rightHeight = Height(root->_right);
return leftHeight > rightHeight ? leftHeight + 1 : rightHeight + 1;
}