红黑树 是一颗二叉搜索树,它在每个节点上增加了一个存储位来表示节点的颜色,可以是red或black。通过对任何一条从根节点到叶子节点的简单路径上的颜色来约束,红黑树保证了最长路径不超过最短路经的两倍,因此近似于平衡。
红黑树的规则:
1、每个节点不是红色就是黑色的。
2、根结点是黑色的。
3、如果一个节点是红色的,则它的两个子结点是黑色的。即每条路径上不能存在两个连续的红节点。
4、对每个节点,从该节点到其他节点的简单路径上,均包含相同数目的黑色节点。
红黑树节点RBTreeNode的实现,利用三叉链(left、right、parent)、key、value及颜色col。
enum colour
{
RED,
BLACK,
};
template<class K, class V>
struct RBTreeNode
{
RBTreeNode<K, V>* _left;
RBTreeNode<K, V>* _right;
RBTreeNode<K, V>* _parent;
K _key;
V _value;
colour _col;
RBTreeNode(const K& key = K(), const V& value = V())
:_left(NULL)
, _right(NULL)
, _parent(NULL)
, _key(key)
, _value(value)
, _col(RED)//初始化插入节点的颜色为红色,不影响黑节点个数
{}
};
一、红黑树的插入
红黑树的插入类似于二叉搜索树,但是每次插入后都到要注意是否满足红黑树的规则,特别是规则3和规则4。如果不满足就需要调整树的结构,下面对插入节点时分成以下几种情况:
注:cur(插入节点)parent(cur的父亲节点)grandfather(parent的父亲节点)uncle(父亲的兄弟节点)
1、根节点root为空,直接插入新节点并给root,设置根节点的颜色为BLACK。
2、根节点root不为空,找到插入节点的位置并插入节点cur。cur节点是红色,若parent是红节点,则需要进行调整。