红黑树（插入）

本文地址：http://blog.csdn.net/spch2008/article/details/9338919

红黑树的插入操作相对删除操作比较简单。

红黑树要满足：任一节点至NULL的任何路径，所含黑节点数必须相同。所以，为了满足此性质，插入节点应该为红色。

如果插入节点的父亲为黑色，则不需要进行调整，若为红色，有三种情况需要讨论。

1. 父节点为红色，叔节点为红色

   对于这种情况，同时改变父、叔节点颜色为黑色，并将爷爷节点置红，这样在这个局部范围内，保持黑色平衡。将x置为爷爷节点，

   如果x此时为根，则置x为黑色，调整结束，否则相当于又插入了一个红色节点，进行新一轮迭代。

  

2. 父节点为红色，叔节点为黑色，左外侧插入

    此时爷爷节点必为黑色。将父节点置为黑色，爷爷节点置为红色，这样，左边多了一个黑色节点，右旋将其降下来。

    新插入节点为x，父节点为p，爷爷节点为g。

   

3. 父节点为红色，叔节点为黑色，左内侧插入

    此时，左旋，将其变成情况2，再由情况2的步骤进行插入。

    

为求简略，只有一侧插入代码，另一侧由于对称性，很好解决。注意：空节点视为黑色。

  x->color = __rb_tree_red;
  while (x != root && x->parent->color == __rb_tree_red) {
    if (x->parent == x->parent->parent->left) {
      __rb_tree_node_base* y = x->parent->parent->right;
      if (y && y->color == __rb_tree_red) {              //情况1
        x->parent->color = __rb_tree_black;
        y->color = __rb_tree_black;
        x->parent->parent->color = __rb_tree_red;
        x = x->parent->parent;
      }
      else {
        if (x == x->parent->right) {                     //情况3，转为情况2
          x = x->parent;
          __rb_tree_rotate_left(x, root);
        }
        x->parent->color = __rb_tree_black;              //情况2
        x->parent->parent->color = __rb_tree_red;
        __rb_tree_rotate_right(x->parent->parent, root);
      }
    }
  }  //end of while
  root->color = __rb_tree_black;