【Java】红黑树的删除操作概述和代码实现

删除操作概述：
红黑树的删除操作相比于它的插入操作还是比较复杂的，一共涉及到了四种调整的情况，其核心就是因为删除了节点（黑），要保持每条路径上的黑色节点个数相等。
当删除节点为红色时，不用做任何调整；当删除节点时黑色，但它的孩子为红色时，直接把它的孩子变成黑色补上去。
删除情况调整（看兄弟）：
1、删除的时黑色节点，它的孩子也是黑色（或null），它的兄弟节点也是黑色：把兄弟节点变成黑色，指针指向父节点，向上回溯，如果父节点为红色，直接将他变成黑色；
2、删的是黑色节点，该节点的孩子皆为黑色，兄弟也为黑色，但兄弟的右孩子未红色：兄弟的颜色和父节点的颜色交换后，将兄弟红孩的颜色变黑然后以父节点做左旋操作；
3、删的是黑色节点，该节点的孩子皆为黑色，兄弟也为黑色，但兄弟的左孩子为红色：将兄弟的红孩颜色于兄弟的颜色交换，然后以兄弟做右旋操作，然后再如同情况2的方式处理；
4、删除节点时黑色，兄弟是红色：将兄弟和父节点的颜色交换，然后以父节点做左旋操作；

代码实现：
删除操作

private void remove(T data) {
    if (this.root == null)
        return;
    RBNode<T> cur = this.root;
    while (cur != null) {
        if (cur.getData().compareTo(data) > 0) {
            cur = cur.getLeft();
        } else if (cur.getData().compareTo(data) < 0) {
            cur = cur.getRight();
        } else {
            break;
        }
    }
    if (cur == null) {
        return;
    }
    if (cur.getLeft() != null && cur.getRight() != null) {
        RBNode<T> old = cur;
        cur = cur.getLeft();
        while (cur.getRight() != null) {
            cur = cur.getRight();
        }
        old.setData(cur.getData());
    }
    // 删除cur指向的节点
    RBNode<T> child = cur.getLeft();
    if (child == null) {
        child = cur.getRight();
    }
    if (child != null) {
        child.setParent(cur.getParent());
        if (cur.getParent() == null) {
            this.root = child;
        } else {
            if (cur.getParent().getLeft() == cur) {
                cur.getParent().setLeft(child);
            } else {
                cur.getParent().setRight(child);
            }
        }


        if (color(cur) == Color.BLACK) {
            fixAfterRemove(child);
        } else

        {
            if (cur.getParent() == null) {
                this.root = null;
            } else {
                if (color(cur) == Color.BLACK) {
                    fixAfterRemove(cur);
                }

                if (cur.getParent().getLeft() == cur) {
                    cur.getParent().setLeft(null);
                } else {
                    cur.getParent().setRight(null);
                }
            }
        }
    }
}

删除调整：

private void fixAfterRemove(RBNode<T> node) {
    while(color(node) == Color.BLACK){//当删除的节点为黑色时  为红色时直接删除 没有影响
        if(node == left(parent(node)))   { // 左子树删除黑色节点
            RBNode<T> brother = right(parent(node));//当前节点在左，那他的兄弟则在右
            if(color(brother) == Color.RED){ // 情况4 兄弟时红色 兄弟的孩子时黑色
                setColor(brother, Color.BLACK);//交换父兄颜色
                setColor(parent(node), Color.RED);
                leftRotate(parent(node));//以父亲为根节点左旋
                brother = right(parent(node));//更新兄弟节点的指针
            }
            if(color(left(brother)) == Color.BLACK // 情况1 兄弟的两个孩子是黑色  兄弟黑色
                    && color(right(brother)) == Color.BLACK){
                setColor(brother, Color.RED);//把兄弟弄成红色  继续向上回溯
                node = parent(node);//当前指针更新到父亲
            } else {
                // 情况3的前半段
                if(color(right(brother)) != Color.RED){
                    setColor(brother, Color.RED);
                    setColor(left(brother), Color.BLACK);
                    rightRotate(brother);
                    brother = right(parent(node));
                }

                // 统一处理情况2
                setColor(brother, color(parent(node)));//父兄颜色交换
                setColor(parent(node), Color.BLACK);
                setColor(right(brother), Color.BLACK);
                leftRotate(parent(node));
                break;
            }
        } else {
            // 右子树删除黑色节点
            RBNode<T> brother = left(parent(node));
            if(color(brother) == Color.RED){ // 情况4
                setColor(brother, Color.BLACK);
                setColor(parent(node), Color.RED);
                rightRotate(parent(node));
                brother = left(parent(node));
            }

            if(color(left(brother)) == Color.BLACK // 情况1
                    && color(right(brother)) == Color.BLACK){
                setColor(brother, Color.RED);
                node = parent(node);
            } else {
                // 情况3的前半段
                if(color(left(brother)) != Color.RED){
                    setColor(brother, Color.RED);
                    setColor(right(brother), Color.BLACK);
                    leftRotate(brother);
                    brother = left(parent(node));
                }

                // 统一处理情况2
                setColor(brother, color(parent(node)));
                setColor(parent(node), Color.BLACK);
                setColor(left(brother), Color.BLACK);
                rightRotate(parent(node));
                break;
            }
        }
        if(node == right(parent(node))){ // 右子树删除黑色节点
            RBNode<T> brother =left(parent(node));
            if(color(brother) == Color.RED){ // 情况4 兄弟时红色 兄弟的孩子时黑色
                setColor(brother, Color.BLACK);//交换父兄颜色
                setColor(parent(node), Color.RED);
                rightRotate(parent(node));//以父亲为根节点左旋
                brother = left(parent(node));//更新兄弟节点的指针
            }
            if(color(left(brother)) == Color.BLACK // 情况1 兄弟的两个孩子是黑色  兄弟黑色
                    && color(right(brother)) == Color.BLACK){
                setColor(brother, Color.RED);//把兄弟弄成红色  继续向上回溯
                node = parent(node);//当前指针更新到父亲
            } else {
                // 情况3的前半段
                if(color(left(brother)) != Color.RED){
                    setColor(brother, Color.RED);
                    setColor(left(brother), Color.BLACK);
                   leftRotate(brother);
                    brother =left(parent(node));
                }

                // 统一处理情况2
                setColor(brother, color(parent(node)));//父兄颜色交换
                setColor(parent(node), Color.BLACK);
                setColor(left(brother), Color.BLACK);
                rightRotate(parent(node));
                break;
            }
        } else {
            // 右子树删除黑色节点
            RBNode<T> brother = right(parent(node));
            if(color(brother) == Color.RED){ // 情况4
                setColor(brother, Color.BLACK);
                setColor(parent(node), Color.RED);
               leftRotate(parent(node));
                brother = right(parent(node));
            }

            if(color(left(brother)) == Color.BLACK // 情况1
                    && color(right(brother)) == Color.BLACK){
                setColor(brother, Color.RED);
                node = parent(node);
            } else {
                // 情况3的前半段
                if(color(right(brother)) != Color.RED){
                    setColor(brother, Color.RED);
                    setColor(right(brother), Color.BLACK);
                    rightRotate(brother);
                    brother = right(parent(node));
                }

                // 统一处理情况2
                setColor(brother, color(parent(node)));
                setColor(parent(node), Color.BLACK);
                setColor(right(brother), Color.BLACK);
                leftRotate(parent(node));
                break;
            }
        }
    }
    // 在当前这一路遇见红色节点，直接补一个黑色节点
    setColor(node, Color.BLACK);
}