数据结构/C++:红黑树
- 概念
- 实现
- 基本结构
- 插入
- uncle为红色节点
- uncle为黑色节点
- 总代码展示
概念
红黑树是一种二叉搜索树,一般的二叉搜索会发送不平衡现象,导致搜索效率下降,于是学者们开始探索如何让二叉搜索树保持平衡,这种树叫做自平衡二叉搜索树。起初学者发明了AVL树,其通过一定算法保持了二叉搜索树的严格平衡,不久后Rudolf Bayer发明了红黑树,红黑树的平衡是较为宽泛的,为了保持平衡,红黑树付出的代价比AVL树更小。因此红黑树被更为广泛的使用,比如Java,C++,python中,使用的自平衡二叉搜索树都是红黑树,而不是AVL树。
如果想了解AVL树,可以看这篇博客:[数据结构/C++:AVL树]
红黑树的要求如下:
红黑树中,最长路径的长度不会超过最短路径的两倍
先解释一下路径
的概念:从根走到nullptr
。
有不少人认为路径是从根走到叶子节点,这是不正确的。
红黑树用了五条规则来限制一棵树,从而达到以上要求:
- 每个节点不是红色就是黑色
- 根节点一定是黑色
- 不可以出现连续的红色节点(黑色可以连续出现)
- 每一条路径都包含相同数目的黑色节点
nullptr
视为黑色节点
只要满足以上五个条件,那么这棵树就是一颗红黑树,而且满足最长路径的长度不会超过最短路径的两倍。为什么呢?
五条规则中,我标红了3,4两条规则:
- 不可以出现连续的红色节点(黑色可以连续出现)
- 每一条路径都包含相同数目的黑色节点
由于每一条路径都必须包含相同数目的黑色节点,现在我们假设一棵红黑树,所有路径的黑色节点数目都是x
,那么最短的路径长度就是全为黑色节点,长度为x
。
如果想让一条路径变长,那么就只能插入更多的红色节点(因为黑色节点数目相同),但是红色节点又不能连续出现,所以只能是黑红黑红黑红黑红黑红......
这样排列,一个黑节点匹配一个红节点,因此最长路径的长度就是黑色节点的两倍2x
。
可以发现,红黑树通过这两条核心规则,保证了二叉搜索树的平衡。
比如以下就是一颗红黑树:
其最短路径为最左侧的路径,长度为2,即两个黑节点。
其最长路径为最右侧的路径,长度为4,即一红一黑排列。
要注意的是:不是所有的红黑树都会出现以上的全黑路径,或者一红一黑路径的,这只是极端情况。
接下来我们通过实现红黑树,来了解红黑树是如何自平衡的:
实现
基本结构
首先我们要在节点中加入一个成员来表示节点的颜色,颜色有红黑和黑色两种状态,这里我使用枚举来区分两者:
enum Colour
{
RED,
BLACK
};
在某些红黑树的实现中,使用bool
值来表示红黑颜色,这也是可以的,但是本博客以枚举来表示颜色。
节点类:
template<class K, class V>
struct RBTreeNode
{
RBTreeNode* _left;
RBTreeNode* _right;
RBTreeNode* _parent;
pair<K, V> _kv;
Colour _col;
};
_left
:左子树
_right
:右子树
_parent
:父节点
_kv
:节点存储的值
_col
:该节点的颜色
节点类还需要一个构造函数进行初始化,现在的问题就是:新的节点要初始化为什么颜色?
先来考虑一下:插入红色节点和插入黑色节点,谁对红黑树影响大?
对于一棵红黑树,其所有路径的黑色节点数目都相同,如果我们在某一条路径末尾插入了黑色节点,那么整棵树的所有其它路径都会少一个黑节点。而插入红色节点只影响当前路径,所以新节点应该是红色节点。
构造函数:
RBTreeNode(const pair<K, V>& kv)
: _left(nullptr)
, _right(nullptr)
, _parent(nullptr)
, _kv(kv)
, _col(RED)//初始化为红节点
{}
接着就是红黑树本体,类中只存储一个根节点_root
:
template<class K, class V>
class RBTree
{
typedef RBTreeNode<K, V> Node;
private:
Node* _root = nullptr;
}
现在我们有了红黑树的基本结构,接下来就实现它的插入操作:
插入
那么我们先写出当基本的二叉搜索树的插入代码逻辑,既然要插入,那么就要先找到合适的位置插入,代码如下:
bool Insert(const pair<K, V>& kv)
{
if (_root == nullptr)
{
_root = new Node(kv);
_root->_col = BLACK;//保持根为黑节点
}
Node* cur = _root;
Node* parent = nullptr;
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->_left = cur;
else
parent->_right = cur;
cur->_parent = parent;
//调整红黑树
//......
//......
//......
return true;
}
接下来,我先解析以上代码的逻辑:
if (_root == nullptr) { _root = new Node(kv); _root->_col = BLACK;//保持根为黑节点 }
如果我们插入节点时,根节点
_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; } }
以上代码,是在找到合适的插入位置,当
key
大于当前节点cur->_kv.first < kv.first
,那么cur
就向左寻找,反之向右寻找。如果当前节点值等于key
,那么说明该节点已经存在,返回false
代表插入失败。当我们的cur
为空指针,说明已经找到了插入的节点,此时跳出循环进行插入。
cur = new Node(kv); if (parent->_kv.first > kv.first) parent->_left = cur; else parent->_right = cur; cur->_parent = parent;
到达此处,说明前面已经找到插入的位置了,而
parent
节点就是插入位置的父亲节点。根据key
的大小,来判断插入到左边还是右边,插入完成后,再让新节点的_parent
指向parent
。
至此我们就完成了插入操作,接下来就要根据不同情况对红黑树进行调整。
对于红黑树的插入,我们需要关注新节点的父亲parent
,祖父grandfather
,叔叔uncle
三个节点:
- 先根据父亲节点的颜色,来判断是否需要调整
父亲节点为黑色:
新插入的节点默认为红色,所以新插入节点不会影响路径上黑色节点的数目,而parent
是黑节点,我们也没有出现连续的红色节点,所以这种情况无需任何调整,直接插入就可以。
父亲节点为红色:
如果父亲节点为红色,我们就会出现连续的红色节点,这时我们就需要进行调整了
以上两种情况总结为:
当
parent
为黑色,直接插入
当parent
为红色,插入后需要进行调整
当前的代码为:
在这里插入代码片
当parent
为红色,我们就需要再根据uncle
的颜色,将插入分类两类:uncle为红色
以及uncle为黑色
。
值得注意的是:由于parent
是红色节点,此时的grandfather
一定是黑色节点,因为不能出现连续红色节点
这两种情况的操作不同,我们先看到uncle为红色
的情况:
uncle为红色节点
当
uncle
节点为红色,此时需要进行变色
变色如下:
由于新插入了红色的cur
节点,此时parent
与cur
出现了连续的红色节点,于是我们将parent
改为黑色。但是此时以parent
为根的所有路径就会多出一个黑节点,于是把grandfather
变为红色,来抵消这个新增的黑节点。但是此时以uncle
为根的路径又会少一个黑节点,于是把uncle
变黑。
但是我们将grandfather
变为了红色,这有可能会影响到上一层节点,比如这样:
我们把grandfather
变红之后,又出现了两个红色节点相连的情况,所以我们要写一个while循环,来反复向上检查。
当前代码如下:
while (parent && parent->_col == RED)//只有parent为红,才更新 (parent可能不存在)
{
Node* grandfather = parent->_parent;
if (parent == grandfather->_left)
{
Node* uncle = grandfather->_right;
//uncle存在且为红节点
if (uncle && uncle->_col == RED)
{
parent->_col = uncle->_col = BLACK;
grandfather->_col = RED;
cur = grandfather;
parent = cur->_parent;
}
else//uncle为黑节点
{
//其它处理
}
}
else
{
Node* uncle = grandfather->_left;
//uncle存在且为红节点
if (uncle && uncle->_col == RED)
{
parent->_col = uncle->_col = BLACK;
grandfather->_col = RED;
cur = grandfather;
parent = cur->_parent;
}
else//uncle为黑节点
{
//其它处理
}
}
}
_root->_col = BLACK;//在循环内部不判断root情况,统一处理
代码解析:
while (parent && parent->_col == RED)
该代码用于检测
cur
的parent
的颜色,通过我们前面的推导,如果parent
为红色才需要调整,因此进入循环的条件之一是parent
为红色。另外的parent
有可能为nullptr
,此时我们要避免访问空指针,所以空指针也不能进循环
if (parent == grandfather->_left) { } else { }
这一段代码是在检测
parent
节点是grandfather
的左子树还是右子树,这将涉及到我们如何找uncle
以及下一种情况的调整,此时我们要分类讨论。当parent == grandfather->_left
成立,那么uncle
就是grandfather
的右子树:Node* uncle = grandfather->_right;
,反之就是左子树
if (uncle && uncle->_col == RED) { parent->_col = uncle->_col = BLACK; grandfather->_col = RED; cur = grandfather; parent = cur->_parent; }
我们找到
uncle
后,如果uncle
是红色,那么直接进行变色操作,把parent
和uncle
的颜色变为黑色,grandfather
变为红色。
随后由于我们的变色操作可能会影响上一层,此时调整节点,进入下一次while
循环
在整个
while
循环外侧,还有一句代码:_root->_col = BLACK;
这是因为我们在先前的while循环中,有可能出现对
_root
节点的操作,导致_root
的颜色改变,而_root
需要保持黑色。如果我们在循环内部,每一次都检测_root
有点麻烦了,于是我们直接在每一次调整完节点后,把_root
强行矫正为黑色
至此我们就讨论完了uncle
为红色节点的情况,接下来我们就讨论uncle
为黑色节点:
uncle为黑色节点
由于红黑树中,nullptr
也算作黑色节点,所以uncle
为黑色分为以下两种情况:
uncle
为空指针uncle
不为空指针
图示如下:
如果
uncle
为空指针,那么cur
一定是新插入的节点。
因为如果cur
不是新插入的节点,那么cur
和parent
一定有一个原先是黑色节点,不然会出现连续的红色节点。但是如果cur
和parent
有一个是黑色节点,那么grandfather
的左子树就比右子树多出一个黑节点,这就违背了红黑树规则。无论怎样,原先的树都不可能符合规则,所以cur
一定是新插入的节点,破坏了规则。
如果
uncle
不为空指针,那么cur
一定是从黑色节点变成的红色节点(不是新插入的)。
因为如果uncle
存在,那么grandfather
的右子树就存在一个黑节点,而parent
是红节点,所以cur
和parent
的右子树中都至少有一个黑节点,才能保证每一条路径黑节点数目相同。因此cur
原先一定是黑节点,是因为cur
下层插入了新节点,然后通过while
循环向上走,影响到了当前层。
对于这种uncle
为黑色的情况,我们需要通过旋转+变色来维持红黑树。
旋转又分为单旋和双旋:
当
cur
与parent
的关系和parent
与grandfather
的关系一致时,需要进行单旋
比如我们刚刚的情况:
cur
是parent
的左子树,parent
是grandfather
的左子树,关系一致。
我们需要对其进行右单旋+变色:
这个旋转的算法在此我就不过多讲解了,可以去AVL树的博客中了解。我重点讲解一下变色和旋转的合理性:
一次插入过程中,走到这一步,说明前面一定经过了
uncle
为红色的情况,而对uncle
为红色的情况进行变色并不会对任何路径的黑色节点数目造成影响,因此目前还是符合黑色节点数目相同规则的。
同为parent
的子树,以cur
和C
为根的路径,黑节点数目相同
同为grandfather
的子树,以parent
和uncle
为根的路径黑节点数目相同
而parent
是红色节点,所以cur
,C
以及uncle
为根的路径,黑节点数目都相同
进行单旋,会把
c
树交给grandfather
做子树,而c
与uncle
为根的路径黑节点数目相同,不违背规则(旋转的合理性)
旋转后,
parent
作新根,grandfather
与cur
作为左右子树grandfather
为根的路径,整体上就会比以cur
为根的路径多出一个黑节点(即grandfather
本身)
因此,将grandfather
改为红节点,来平衡parent
左右子树的黑节点
而红色节点不能连续出现,再把parent
改为黑节点
当
cur
与parent
的关系和parent
与grandfather
的关系不一致时,需要进行双旋
以上结构中,cur
是parent
的左子树,parent
是grandfather
的右子树,关系不一致,要进行双旋。
同样的,讲解一下变色和旋转的合理性:
一次插入过程中,走到这一步,说明前面一定经过了
uncle
为红色的情况,而对uncle
为红色的情况进行变色并不会对任何路径的黑色节点数目造成影响,因此目前还是符合黑色节点数目相同规则的。
同为parent
的子树,以cur
和A
为根的路径,黑节点数目相同
同为cur
的子树,以B
和C
为根的路径,黑节点数目相同
由于cur
是红节点,所以以A
,B
,C
为根的路径,黑节点数目相同
相同的手段,由于parent
是红节点,所以A
与uncle
为根的路径的黑节点数目相同
因此A
,B
,C
,uncle
为根的路径,黑节点数目都相同
进行双旋,会把
C
子树交给grandfather
做子树,而C
与uncle
黑节点数目相同,不违背规则也会把B交给parent
做子树
A
与B
黑节点数目相同,不违背规则
旋转后,cur
作新根,grandfather
与parent
作为左右子树grandfather
为根的路径,整体上就会比以parent
为根的路径多出一个黑节点(grandfather
本身)
因此,将grandfather
改为红节点,来平衡cur
左右子树的黑节点而红色节点不能连续出现,再把cur
改为黑节点
以上单旋和双旋的变色,看似复杂,其实最后都是把新根的颜色变为黑色,新根的左右子树变为红色。由于我们旋转后,新根都是黑节点,所以不会影响上层,可以直接跳出循环。
代码如下:
当parent == grandfather->_left
:
else//uncle为黑节点 (旋转)
{
if (cur == parent->_left)
{
RotateR(grandfather);//右单旋
parent->_col = BLACK;//变色
grandfather->_col = RED;//变色
}
else
{
RotateL(parent);//左右双旋 - 左单旋
RotateR(grandfather);//左右双旋 - 右单旋
cur->_col = BLACK;//变色
grandfather->_col = RED;//变色
}
break;//旋转后一定平衡
}
当parent == grandfather->_right
:
else//uncle为黑节点 (旋转)
{
if (cur == parent->_right)
{
RotateL(grandfather);//左单旋
parent->_col = BLACK;//变色
grandfather->_col = RED;//变色
}
else
{
RotateR(parent);//右左双旋 - 右单旋
RotateL(grandfather);//右左双旋 - 左单旋
cur->_col = BLACK;//变色
grandfather->_col = RED;//变色
}
break;//旋转后一定平衡
}
insert
总代码:
bool Insert(const pair<K, V>& kv)
{
if (_root == nullptr)
{
_root = new Node(kv);
_root->_col = BLACK;//保持根为黑节点
}
Node* cur = _root;
Node* parent = nullptr;
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->_left = cur;
else
parent->_right = cur;
cur->_parent = parent;
while (parent && parent->_col == RED)//只有parent为红,才更新 (parent可能不存在)
{
Node* grandfather = parent->_parent;
if (parent == grandfather->_left)
{
Node* uncle = grandfather->_right;
//uncle存在且为红节点
if (uncle && uncle->_col == RED)
{
parent->_col = uncle->_col = BLACK;
grandfather->_col = RED;
cur = grandfather;
parent = cur->_parent;
}
else//uncle不存在或为黑节点 (旋转)
{
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
{
Node* uncle = grandfather->_left;
//uncle存在且为红节点
if (uncle && uncle->_col == RED)
{
parent->_col = uncle->_col = BLACK;
grandfather->_col = RED;
cur = grandfather;
parent = cur->_parent;
}
else//uncle不存在或为黑节点 (旋转)
{
if (cur == parent->_right)
{
RotateL(grandfather);
parent->_col = BLACK;
grandfather->_col = RED;
}
else
{
RotateR(parent);
RotateL(grandfather);
cur->_col = BLACK;
grandfather->_col = RED;
}
break;//旋转后一定平衡
}
}
}
_root->_col = BLACK;//在循环内部不判断root情况,统一处理
return true;
}
总代码展示
红黑树总代码:
RBTree.h
:
#pragma once
#include <iostream>
#include <assert.h>
using namespace std;
enum Colour
{
RED,
BLACK
};
template<class K, class V>
struct RBTreeNode
{
RBTreeNode* _left;
RBTreeNode* _right;
RBTreeNode* _parent;
pair<K, V> _kv;
Colour _col;
RBTreeNode(const pair<K, V>& kv)
: _left(nullptr)
, _right(nullptr)
, _parent(nullptr)
, _kv(kv)
, _col(RED)
{}
};
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->_col = BLACK;//保持根为黑节点
}
Node* cur = _root;
Node* parent = nullptr;
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->_left = cur;
else
parent->_right = cur;
cur->_parent = parent;
while (parent && parent->_col == RED)//只有parent为红,才更新 (parent可能不存在)
{
Node* grandfather = parent->_parent;
if (parent == grandfather->_left)
{
Node* uncle = grandfather->_right;
//uncle存在且为红节点
if (uncle && uncle->_col == RED)
{
parent->_col = uncle->_col = BLACK;
grandfather->_col = RED;
cur = grandfather;
parent = cur->_parent;
}
else//uncle不存在或为黑节点 (旋转)
{
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
{
Node* uncle = grandfather->_left;
//uncle存在且为红节点
if (uncle && uncle->_col == RED)
{
parent->_col = uncle->_col = BLACK;
grandfather->_col = RED;
cur = grandfather;
parent = cur->_parent;
}
else//uncle不存在或为黑节点 (旋转)
{
if (cur == parent->_right)
{
RotateL(grandfather);
parent->_col = BLACK;
grandfather->_col = RED;
}
else
{
RotateR(parent);
RotateL(grandfather);
cur->_col = BLACK;
grandfather->_col = RED;
}
break;//旋转后一定平衡
}
}
}
_root->_col = BLACK;//在循环内部不判断root情况,统一处理
return true;
}
//左单旋
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;
subR->_parent = nullptr;
}
else
{
if (ppNode->_left == parent)
ppNode->_left = subR;
else
ppNode->_right = subR;
subR->_parent = ppNode;
}
}
//右单旋
void RotateR(Node* parent)
{
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;
}
}
size_t Size()
{
return _Size(_root);
}
size_t _Size(Node* root)
{
if (root == nullptr)
return 0;;
return _Size(root->_left) + _Size(root->_right) + 1;
}
Node* Find(const K& key)
{
Node* cur = _root;
while (cur)
{
if (cur->_kv.first < key)
{
cur = cur->_right;
}
else if (cur->_kv.first > key)
{
cur = cur->_left;
}
else
{
return cur;
}
}
return nullptr;
}
//中序
void InOrder()
{
_InOrder(_root);
cout << "end" << endl;
}
int Height()
{
return _Height(_root);
}
private:
//中序
void _InOrder(Node* root)
{
if (root == nullptr)
return;
_InOrder(root->_left);
cout << root->_kv.first << " - ";
_InOrder(root->_right);
}
//求高度
int _Height(Node* root)
{
if (root == nullptr)
return 0;
return max(Height(root->_left), Height(root->_right)) + 1;
}
Node* _root = nullptr;
};