Question

我有来自Thomas H Cormen的书籍“算法导论”中的伪代码以及其他红色和黑色树的伪代码。

插入和插入修复的伪代码位于here

我在插入修复函数中遇到了一个seg错误：while(n->parent->color == 'r')在尝试插入“8”时利用下面的测试数据：

测试数据：

i 5
i 7
i 1
i 8
i 3

我认为这是因为n的父母可能不存在？但我不确定如何在不完全搞砸伪代码的情况下适当更改代码。：

这是我的插页：

void insert_fix(node * n)
{
    node * y;
    if(n->parent)
    {
        while(n->parent->color == 'r')
        {
            if(n->parent == n->parent->parent->left)
            {
                y = n->parent->parent->right;
                if(y->color == 'r')
                {
                    n->parent->color = 'b';
                    y->color = 'b';
                    n->parent->parent->color = 'r';
                    n = n->parent->parent;
                }
                else
                {
                    if(n == n->parent->right)
                    {
                        n = n->parent;
                        rotate_left(n);
                    }
                    n->parent->color = 'b';
                    n->parent->parent->color = 'r';
                }
            }
            else
            {
                y = n->parent->parent->left;
                if(y->color == 'r')
                {
                    n->parent->color = 'b';
                    y->color = 'b';
                    n->parent->parent->color = 'r';
                    n = n->parent->parent;
                }
                else
                {
                    if(n == n->parent->left)
                    {
                        n = n->parent;
                        rotate_right(n);
                    }
                    n->parent->color = 'b';
                    n->parent->parent->color = 'r';
                }
            }
        }
    }
    root->color = 'b';
}

这是我的插入修复：

void insert_fix(node * n)
{
    node * y;
    if(n->parent)
    {
        while(n->parent->color == 'r')
        {
            if(n->parent == n->parent->parent->left)
            {
                y = n->parent->parent->right;
                if(y->color == 'r')
                {
                    n->parent->color = 'b';
                    y->color = 'b';
                    n->parent->parent->color = 'r';
                    n = n->parent->parent;
                }
                else
                {
                    if(n == n->parent->right)
                    {
                        n = n->parent;
                        rotate_left(n);
                    }
                    n->parent->color = 'b';
                    n->parent->parent->color = 'r';
                }
            }
            else
            {
                y = n->parent->parent->left;
                if(y->color == 'r')
                {
                    n->parent->color = 'b';
                    y->color = 'b';
                    n->parent->parent->color = 'r';
                    n = n->parent->parent;
                }
                else
                {
                    if(n == n->parent->left)
                    {
                        n = n->parent;
                        rotate_right(n);
                    }
                    n->parent->color = 'b';
                    n->parent->parent->color = 'r';
                }
            }
        }
    }
    root->color = 'b';
}

*为了清楚起见，我在上面的while循环中添加了额外的if语句，这样如果节点是root就不会发生任何事情，但它并没有像我希望的那样解决我的问题。

为了更好的衡量，我根据我在互联网上找到的信息编写了左右旋转功能，我认为它们编写得非常好：

void rotate_right(node *n)
{
    node* left = n->left;
    swap_nodes(n, left);
    n->left = left->right;
    if(left->right != NULL)
        left->right->parent = n;
    left->right = n;
    n->parent = left;
}

void rotate_left(node *n)
{
    node* right = n->right;
    swap_nodes(n, right);
    n->right = right->left;
    if(right->left != NULL)
        right->left->parent = n;
    right->left = n;
    n->parent = right;
}

void swap_nodes(node* oldNode, node* newNode)
{
    if(oldNode->parent == NULL)
        root = newNode;
    else
    {
        if(oldNode == oldNode->parent->left)
            oldNode->parent->left = newNode;
        else
            oldNode->parent->right = newNode;
    }

    if(newNode != NULL)
        newNode->parent = oldNode->parent;
}

同样，我觉得我的真实代码准确地遵循伪代码，但我无法弄清楚我缺少的部分。

让我知道！谢谢！

Answer 1

我认为CLRS中的RBT插入算法存在一个错误。当父母和叔叔的颜色都是红色时，算法首先将父母和叔叔重新涂成黑色，然后向上移动，如代码n = n->parent->parent。但是，这不是案件的结束。相反，算法应该在向上移动后递归调用自身。也就是说，应该以您的代码为例，声明如下：

insert_fix(n);

在n = n->parent->parent.

之后

如果我错了，请纠正我。在这种情况下，BTW，the wikipedia entry "red-black tree"可能会有所帮助。

红色黑树插入和修复从有问题的伪代码

1 个答案: