红黑树

《算法导论》中关于红黑树的定义：

1. 每个节点或者是红色的，或者是黑色的；
2. 根节点是黑色的；
3. 每一个叶子节点（最后的空节点）是黑色的；
4. 如果一个节点是红色的，那么他的孩子节点都是黑色的；
5. 从任意一个节点到叶子节点，经过的黑色节点是一样的。

在算法4中，讲述了红黑树与2-3树的等价性，我们先理解2-3树和红黑树之间的关系。

1 2-3树（绝对平衡的树）

满足二分搜索树的基本性质，节点可以存放一个元素或两个元素。每个节点有2个或3个孩子。

2 红黑树和2-3树的等价性

这样的定义，就是我们认为所有的红色节点都是左倾斜的。

例如：

3 红黑树（保持“黑平衡”的二叉树）

红黑树，严格意义上，不是平衡二叉树。
最大高度：2logn
复杂度级别：O(logn)

3.1 节点定义

template <class K,class V>
class RBTNode{
public:
    static const bool RED = true;
    static const bool BLACK = false;

    K key;
    V value;
    RBTNode<K,V> *left,*right;
    bool color;//颜色属性

    RBTNode(K key,V value):key(key),value(value),left(nullptr),
                           right(nullptr){
        this->color=RED;//默认颜色是红色
    }
    RBTNode(RBTNode<K,V> *node){
        this->key=node->key;
        this->value=node->value;
        this->left=node->left;
        this->right=node->right;
        this->color=node->color;
    }
};

3.2 左旋转

向树中添加一个新元素时，若所添加的元素比已知节点大，那么就会添加到右子树中，这样就不是左倾斜的，需要做一个左旋转。

在左旋转中并不去维持红黑树性质。

//左旋转
RBTNode<K,V> *leftNode(RBTNode<K,V> *node){
    RBTNode<K,V> *x=node->right;
    node->right=x->left;
    x->left=node;

    x->color=node->color;
    node->color=RED;//颜色设置为红色

    return x;
}

3.3 右旋转

//右旋转
RBTNode<K,V> *rightNode(RBTNode<K,V> *node){
	RBTNode<K,V> *x=node->left;
    node->left=x->right;
    x->right=node;

    x->color=node->color;
    node->color=RED;

    return x;
}

3.4 颜色翻转

//颜色翻转
void flipColors(RBTNode<K,V> *node){
	node->color=RED;
    node->left->color=BLACK;
    node->right->color=BLACK;
}

3.5 添加元素

2-3树中添加一个新元素，或者添加进2-节点，形成一个3-节点；或者添加进3-节点，暂时形成一个4-节点。

当红黑树为空时，添加一个元素，因为设置的树初始颜色为红色，但是红黑树的性质：根节点为黑色，所以需要翻转颜色。

// 向以node为根的红黑树中,插入节点(key, value)
// 返回插入新节点后的红黑树的根
RBTNode<K,V> *add(RBTNode<K,V> *node, K key, V value) {
    if (node == nullptr) {
        size++;
        return new RBTNode<K,V>(key, value);//默认插入红色节点
    }
    if (key == node->key) {
        node->value = value;
    } else if (key < node->key) {
        node->left = add(node->left, key, value);
    } else {
        node->right = add(node->right, key, value);
    }
    //注意第一个if判断条件：
    //此时节点需要判断是否需要左旋转？
    //但是如果节点的左孩子是红色，直接翻转就可以了
    if(isRed(node->right)&&!isRed(node->left))
        node=leftNode(node);
    if(isRed(node->left)&&isRed(node->left->left))
        node=rightNode(node);
    if(isRed(node->left)&&isRed(node->right))
        flipColors(node);

    return node;
}