算法导论笔记 红黑树(3)

//---------------------------15/03/20----------------------------

    

    //删除

    

    //相当于copy一个u给v

    RB_TRANSPLANT(T,u,v)

    {

        if(u,p == T.nil)

        {

            T.root=v;

        }

        else if(u == u.p.left)

        {

            u.p.left=v;

        }

        else

        {

            u.p.right=v;

        }

        v.p=u.p;

    }

    

    //先做普通的树的删除操作，多加了记录颜色，和判断是否修复的操作

    RB_DELETE(T,z)

    {

       

        y=z;

        y_original_color= y.color;  //记录y原来的颜色

        

        //如果z的儿子小于一个

        if(z.left == T.nil) //没有左儿子，把右儿子赋值给x，让z.right成为z。

        {

            x=z.right;

            RB_TRANSPLANT(T,z,z.right);

        }

        else if(z.right == T.nil)   //没有右儿子，做相反的操作

        {

            x = z.left;

            RB_TRANSPLANT(T,z,z.left);

        }

        else                        //有两个儿子

        {

            y=TREE_MINIMUN(z.right);    //找到要移动到z位置的y(min函数可以找到z.right所有子节点中最小的节点

            y_original_color=y.color;

            x＝y.right;

            if(y.p == z)                //y就是z的右儿子

            {

                x.p=y;                  //x的父亲是y

            }

            else                    //y是z的右儿子的最左下角的节点(最小的节点)

            {

                RB_TRANSPLANT(T,y,y.right); //让y.right成为y

                y.right=z.right;            //y.right被赋值为z.right

                y.right.p=y;                //y.right.p(y.right已经是z.right了)被赋值为y

                                            //这样z和z.right的连线就替换成y和y.right(之前的z.right)

            }

            RB_TRANSPLANT(T,z,y);           //这里替换z和z.p的连线

            y.left=z.left;

            y.left.p=y;                     //替换和左儿子的连线

            y.color=z.color;                //改变颜色   到此y成功替换掉了z(包括 父亲，儿子，颜色)

        }

        

        if(y_original_color == BLACK)       //如果y原来是黑色的，就要进行修补操作，因为如果y原来是黑色的

                                            //当y被提升到z后，原来y的所有子节点的黑高都会少一

        {

            RB_DELETE_FIXUP(T,x);

        }

            

    }

    

    

    RB_DELETE_FIXUP(T,x);

    {

        /*

            首先给x添加一个黑色属性(并不拿来判断，判断还是用原来的属性)

            当x是红色时 使x代表红黑(红色才是color属性)  x是黑色时 代表 黑黑

            这么假设是为了让y的黑色属性加到x身上(为了使x底下的节点的黑高都加回一)

            这样当x是红色时(红黑) 只要改变x成黑色就可以使

            x和x的所有子节点黑高和原来的一样了.

            如果x已经修补完毕(修补完毕后会有一个操作把x设置为T.root) 或者 x是红色的(只需改变成黑色)

         

        */

        while(x != T.root && x.color == BLACK)

        {

            if(x == x.p.left)

            {

                //w表示x的兄弟节点

                w=x.p.right;

                

                //case1: w是红色的(说明w的父节点和子节点都是黑色)，可以把他转化成case 2 3 4

                //左转x.p 并调整颜色保持平衡,这时x的父节点就变成红色了。

                if(w.color == RED)

                {

                    w.color==BLACK;

                    x.p.color=RED;

                    LEFT_ROTATE(T,x.p);

                    w=x.p.right;

                }

                //底下的三种情况w是黑色的

                

                /*

                    case2:w的儿子都是黑色

                    把x和w的黑属性上移(把 黑高加1的这个属性赋给x.p，x还是黑色，w变成红色)

                    这样可以保证x.p这条路径的黑高不变

                    这时x变成了原先的x的父节点,如果x.p是红色那就完成了，否则就开始循环

                */

                if(w.left.color == BLACK && w.right.color == BLACK)

                {

                    w.color= RED;

                    x = x.p;

                }

                else

                {

                    /*

                        case3:w的右儿子是黑色的(这不是我们想要的)，那就把他变成红色

                        右转w并重新填颜色(w变成红色，换上来的w的左儿子变成黑色(原本是红色，

                        因为进入case3就说明不可能左右儿子都是黑色，而w的右儿子已经是黑色了)

                        这时候w要再赋值一次，因为w右转下去了，新的w是原来w的左儿子

                    */

                    if(w.right.color == BLACK)

                    {

                        w.left.color = BLACK;

                        w.color=RED;

                        RIGHT_ROTATE(T,w);

                        w= x.p.right;

                    }

                    /*

                        case4:w的右儿子是红色的，此时左转x.p 并改变颜色就完成了(所以要把根节点赋值给x用来判断完成)

                        先左转x.p；

                        把w的颜色变成原来x.p的颜色，这样右边的路径就相当于去掉了w节点，w本来是黑色的，

                        再把w.right变成黑色(原来是红色),此时右边的路径黑高又加了一。右边已经平衡

                        使x.p的颜色变成黑色(不管原来是什么颜色，这个操作相当于添加一个黑色节点)，这样x这边的黑高都加一

                        要使这条路径的黑高不变只需要去掉x的黑高加一属性就好(从黑黑变成黑)

                        x的兄弟节点(原来是w的左儿子)他的上面是一个黑色节点(x.p，他原来上面是w，也是黑色的)再上面是

                        w节点(颜色是原来w.p的颜色)所以这部分黑高不变.所以左边也平衡完成。

                    */

                    w.color=x.p.color;

                    x.p.color=BLACK;

                    w.right.color=BLACK;

                    LEFT_ROTATE(T,x.p);

                    x=T.root;

                }

            }

            else

            {

                

            }

        }

        x.color=BLACK;

    }

    

    

    

    

}