红黑树（二）插入

最新推荐文章于 2024-11-02 20:41:42 发布

! 不再维护

最新推荐文章于 2024-11-02 20:41:42 发布

阅读量300

点赞数

文章标签：结构二叉树红黑树数据结构搜索

　接下来介绍红黑树的插入操作，介绍插入之前，我们先来了解一下红黑树的性质。

　　1、每个节点不是红色就是黑色

　　2、跟节点为黑色。

　　3、如果节点为红，子节点必须为黑。

　　4、任意节点至树尾端的任何路径，黑节点必须相同。

规则4主要是保证树的平衡性，不过它的要求不是很严。主要是为了减少调整操作。根据规则4，我们可以判断出新节点都是红节点。（如果新节点是黑节点，那么每次插入都要进行调整）

由于要经常使用某个节点的父亲。所以这里添加了一个指向父亲的指针。所以我们先看一下红黑树的结构组成。

typedef int value_type;
typedef bool color_type;
const color_type red = true;
const color_type black = false;

typedef struct node
{
    value_type value;
    color_type color;

    struct node * parent;
    struct node * left;
    struct node * right;
} Node;

typedef struct Tree
{
    Node * root;
} Tree;

Node节点，value，关键字信息。color，颜色。parent，left，right，父亲，左孩子，右孩子。

Tree只有一个根节点。

插入操作，与二叉树的插入是一样的。

首先是找到插入位置。

//根节点返回NULL,找到返回当前节点,否则返回x的父亲(x是节点所在的位置)
static Node * search_node(const int key, Tree * tree)
{
    Node * x, * y;

    x = tree->root;
    y = nil;

    while ( x != nil )
    {
        y = x;
        if ( key < x->value )
            x = x->left;
        else if ( key >  x->value )
            x = x->right;
        else
            break;
    }

    return y;
}

nil是一个空节点，所有应该为空的指针都指向它。这个节点是黑色的，parent，left，right都为NULL，没有初始化value。（不是一定要有nil节点）

其次将元素插入

void rb_insert(const value_type value, Tree * tree)
{
    Node * new_node;//新节点
    Node * x = search_node(value, tree);//插入点
    
    if ( x == nil )
    {
        tree->root = make_new(value);
        tree->root->color = black;//根节点为黑色
    }
    else
    {
        if ( x->value == value )//关键字不可重复
            return;

        new_node = make_new(value);
        if ( value < x->value )//将新节点插入二叉树中
            x->left = new_node;
        else
            x->right = new_node;
        new_node->parent = x;//设置新节点的父亲

        if ( x->color == red )//如果新节点的父亲为红色,调整
            insert_fixup(new_node, tree);
    }
}

新节点一定是红色的，所以不会违反规则4，但有可能违反规则3。也就是说如果插入节点的父亲是红色的，那么违反了规则3。

我们需要进行调整。

调整时会遇到三种情况，根据不同的情况进行不同的调整。

调整是一个循环过程，当x为根节点或者x是黑色时结束。（这个操作次数并不会很多）

三种情况在代码中看注释。

static void insert_fixup(Node * x, Tree * tree)
{
    Node * y;//y是x的伯父节点
    while ( x != tree->root && x->parent->color == red )
    {
        if ( x->parent == x->parent->parent->left )//父亲是祖父左孩子
        {
            y = x->parent->parent->right;

            //case 1:    伯父是红色
            //处理办法
            //        1.将父亲,伯父变成黑色
            //        2.将祖父变成红色
            //        3.将祖父设为x，向上检查
            if ( y->color == red )
            {
                x->parent->color = black;
                y->color = black;
                x->parent->parent->color = red;
                x = x->parent->parent;
            }
            else
            {
                //case 2:    x是父亲的右孩子
                //处理办法:
                //        1.将x设为x的父亲
                //        2.以x为旋转点,左旋
                //经过上述操作,转换成case 3
                if ( x == x->parent->right )
                {
                    x = x->parent;
                    left_rotate(x, tree);
                }
                
                //case 3:    x是父亲的左孩子
                //处理办法:
                //        1.将x的父亲设为黑色
                //        2.将x的祖父设为红色
                //        3.以x的祖父为旋转点,右旋
                x->parent->color = black;
                x->parent->parent->color = red;
                right_rotate(x->parent->parent, tree);
            }
        }
        else//与上述处理相同,只是将left与right互换
        {
            y = x->parent->parent->left;

            if ( y->color == red )
            {
                x->parent->color = black;
                y->color = black;
                x->parent->parent->color = red;
                x = x->parent->parent;
            }
            else
            {
                if ( x == x->parent->left )
                {
                    x = x->parent;
                    right_rotate(x, tree);
                }

                x->parent->color = black;
                x->parent->parent->color = red;
                left_rotate(x->parent->parent, tree);
            }
        }
    }//while end

    tree->root->color = black;
}