0

我一直在浏览 Robert Sedgewick 在算法中描述的红黑树。这是插入红黑树的代码。

public void put(Key key, Value val) {
    root = put(root, key, val);
    root.color = BLACK;
    assert check();
}

// insert the key-value pair in the subtree rooted at h
private Node put(Node h, Key key, Value val) { 
    if (h == null) return new Node(key, val, RED, 1);

    int cmp = key.compareTo(h.key);
    if      (cmp < 0) h.left  = put(h.left,  key, val); 
    else if (cmp > 0) h.right = put(h.right, key, val); 
    else              h.val   = val;

    // fix-up any right-leaning links
    if (isRed(h.right) && !isRed(h.left))      h = rotateLeft(h);
    if (isRed(h.left)  &&  isRed(h.left.left)) h = rotateRight(h);
    if (isRed(h.left)  &&  isRed(h.right))     flipColors(h);
    h.N = size(h.left) + size(h.right) + 1;

    return h;
}

这是一张可视化红黑修复的图像。 红黑可视化 考虑这种情况，当要插入的项目位于顶部 3-node 的中间时。我们必须执行三个 if 语句中给出的三个操作，即h=rotateLeft(h),h=rotateRight(h)和flipcolors(h)。问题是当我们分配h = rotateLeft(h). 返回的节点是指向具有两个连续左红色链接的三个节点中的中间节点的指针。但该算法假设返回的节点是 3 个节点中的顶部节点，具有 2 个连续的左红色链接。因此，当我们再次rotateRight(h)以与开始时相同的位置结束时。可能是我还没有理解算法。

这是代码rotateLeft(h)

private Node rotateLeft(Node h) {
    assert (h != null) && isRed(h.right);
    Node x = h.right;
    h.right = x.left;
    x.left = h;
    x.color = x.left.color;
    x.left.color = RED;
    x.N = h.N;
    h.N = size(h.left) + size(h.right) + 1;
    return x;
}

请帮助我理解如何h=rotateLeft(h)在三个节点中给出顶部节点而不是中间节点，并带有两个连续的红色左链接。

4

1 回答 1

1

我终于明白了算法是如何工作的。之后h=rotateLeft(h)，第二个和第三个if statements评估为false。并被h退回。递归的上一层，我们得到等式左边的h.left =h位置h，是三个节点中的顶级节点，具有两个连续的红色左链接。然后第一个if语句评估为假，第二个if语句评估为真，我们进行右旋转，然后我们进行翻转颜色。

如果我错了，请纠正我。

于 2013-10-31T09:52:58.763 回答