红黑树之删除操作

转载请标明出处，原文地址：http://blog.csdn.net/hackbuteer1/article/details/7760584
      算法导论书上给出的红黑树的性质如下，跟STL源码剖析书上面的4条性质大同小异。
      1、每个结点或是红色的，或是黑色的
      2、根节点是黑色的
      3、每个叶结点（NIL）是黑色的
      4、如果一个节点是红色的，则它的两个儿子都是黑色的。
      5、对于每个结点，从该结点到其子孙结点的所有路径上包含相同数目的黑色结点。
      从红黑树上删除一个节点，可以先用普通二叉搜索树的方法，将节点从红黑树上删除掉，然后再将被破坏的红黑性质进行恢复。
      我们回忆一下普通二叉树的节点删除方法：Z指向需要删除的节点，Y指向实质结构上被删除的结点，如果Z节点只有一个子节点或没有子节点，那么Y就是指向Z指向的节点。如果Z节点有两个子节点，那么Y指向Z节点的后继节点（其实前趋也是一样的），而Z的后继节点绝对不可能有左子树。因此， 仅从结构来看，二叉树上实质被删除的节点最多只可能有一个子树 。
现在我们来看红黑性质的恢复过程：
      如果Y指向的节点是个红色节点，那么直接删除掉Y以后，红黑性质不会被破坏。操作结束。
      如果Y指向的节点是个黑色节点，那么就有几条红黑性质可能受到破坏了。首先是包含Y节点的所有路径，黑高度都减少了一（第５条被破坏）。其次，如果Y的有红色子节点，Y又有红色的父节点，那么Y被删除后，就出现了两个相邻的红色节点（第４条被破坏）。最后，如果Y指向的是根节点，而Y的子节点又是红色的，那么Y被删除后，根节点就变成红色的了（第２条被破坏）。
      其中，第５条被破坏是让我们比较难受的。因为这影响到了全局。这样动作就太大太复杂了。而且在这个条件下，进行其它红黑性质的恢复也很困难。所以我们首先解决这个问题：如果不改变含Y路径的黑高度，那么树的其它部分的黑高度就必须做出相应的变化来适应它。所以，我们想办法恢复原来含Y节点的路径的黑高度。做法就是：无条件的把Y节点的黑色，推到它的子节点X上去。（X可能是NIL节点）。这样，X就可能具有双重黑色，或同时具有红黑两色，也就是第１条性质被破坏了。
      但第１条性质是比较容易恢复的：一、如果X是同时具有红黑两色，那么好办，直接把X涂成黑色，就行了。而且这样把所有问题都解决了。因为将X变为黑色，２、４两条如果有问题的话也会得到恢复，算法结束。二、如果X是双黑色，那么我们希望把这种情况向上推一直推到根节点（调整树结构和颜色，X的指向新的双黑色节点，X不断向上移动），让根节点具双黑色，这时，直接把X的一层黑色去掉就行了（因为根节点被包含在所有的路径上，所以这样做所有路径同时黑高减少一，不会破坏红黑特征）。
      下面就具体地分析如何恢复1、２、４三个可能被破坏的红黑特性：我们知道，如果X指向的节点是有红黑两色，或是X是根节点时，只需要简单的对X进行一些改变就行了。要对除X节点外的其它节点进行操作时，必定是这样的情况：X节点是双层黑色，且X有父节点P。由知可知，X必然有兄弟节点W，而且这个W节点必定有两个子节点。（因为这是原树满足红黑条件要求而自然具备的。X为双黑色，那么P的另一个子节点以下一定要有至少两层的节点，否则黑色高度不可能和X路径一致）。所以我们就分析这些节点之间如何变形，把问题限制在比较小的范围内解决。另一个前提是：X在一开始，肯定是树底的叶节点或是NIL节点，所以在递归向上的过程中，每一步都保证下一步进行时，至少 X的子树是满足红黑特性的。因此子树的情况就可以认为是已经正确的了，这样，分析就只限制在X节点，X的父节点P和X的兄弟节点W，以及W的两个子节点中。
      下面仅仅考虑X原本是黑色的情况即可。
      在这种情况下，X此时应该具有双重黑色，算法的过程就是将这多出的一重黑色向上移动，直到遇到红节点或者根节点。
      接着往下分析， 会遇到4种情况，实际上是8种， 因为其中4种是相互对称的，这可以通过判断X是其父节点的右孩子还是左孩子来区分。下面我们以X是其父节点的左孩子的情况来分析这4种情况，实际上接下来的调整过程，就是要想方设法将经过X的所有路径上的黑色节点个数增加1。
      具体分为以下四种情况：（下面针对x是左儿子的情况讨论，右儿子对称）
      Case1：X的兄弟W是红色（想办法将其变为黑色）
       由于W是红色的，因此其儿子节点和父节点必为黑色，只要将W和其父节点的颜色对换，在对父节点进行一次左旋转，便将W的左子节点放到了X的兄弟节点上，X的兄弟节点变成了黑色，且红黑性质不变。但还不算完，只是暂时将情况1转变成了下面的情况2或3或4。

    Case2：X的兄弟节点W是黑色的，而且W的两个子节点都是黑色的。此时可以将X的一重黑色和W的黑色同时去掉，而转加给他们的父节点上，这是X就指向它的父节点了，因此此时父节点具有双重颜色了。这一重黑色节点上移。

      如果父节点原来是红色的，现在又加一层黑色，那么X现在指向的这个节点就是红黑两色的，直接把X（也就是父节点）着为黑色。问题就已经完整解决了。
     如果父节点现在是双层黑色，那就以父节点为新的X进行向上的下一次的递归。
    Case3：X的兄弟节点W是黑色的，而且W的左子节点是红色的，右子节点是黑色的。此时通过交换W和其左子节点的颜色并进行一次向右旋转就可转换成下面的第四种情况。注意，原来L是红色的，所以L的子节点一定是黑色的，所以旋转中L节点的一个子树挂到之后着为红色的W节点上不会破坏红黑性质。变形后黑色高度不变。

    Case4： X的兄弟节点W是黑色的，而且W的右子节点是红色的。这种情况下，做一次左旋，W就处于根的位置，将W保持为原来的根的位置的颜色，同时将W的两个新的儿子节点的颜色变为黑色，去掉X的一重黑色。这样 整个问题也就得到了解决。递归结束。（在代码上，为了标识递归结束，我们把X指向根节点）

      因此，只要按上面四种情况一直递归处理下去，X最终总会指向根结点或一个红色结点，这时我们就可以结束递归并把问题解决了。
      以上就是红黑树的节点删除全过程。
      总结：
      如果我们通过上面的情况画出所有的分支图，我们可以得出如下结论
      插入操作：解决的是 红-红 问题
      删除操作：解决的是 黑-黑 问题
      即你可以从分支图中看出，需要往上遍历的情况为红红(插入)，或者为黑黑黑（删除）的情况，如果你认真分析并总结所有的情况后，并坚持下来，红黑树也就没有想象中的那么恐怖了，并且很美妙；
      详细的红黑树删除节点的代码如下：

[cpp]  view plain copy

1. #include<iostream>  
2. using namespace std;  
3.   
4. // 定义节点颜色    
5. enum COLOR  
6. {    
7.     BLACK = 0,    
8.     RED    
9. };    
10.     
11. // 红黑树节点    
12. typedef struct RB_Tree_Node  
13. {    
14.     int key;    
15.     struct RB_Tree_Node *left;    
16.     struct RB_Tree_Node *right;    
17.     struct RB_Tree_Node *parent;    
18.     unsigned char RB_COLOR;    
19. }RB_Node;  
20.   
21. // 红黑树，包含一个指向根节点的指针    
22. typedef struct RBTree  
23. {    
24.     RB_Node* root;  
25. }*RB_Tree;  
26.   
27. // 红黑树的NIL节点    
28. static RB_Tree_Node NIL = {0, 0, 0, 0, BLACK};   
29.   
30. #define PNIL (&NIL)   // NIL节点地址   
31.   
32. void Init_RBTree(RB_Tree pTree) // 初始化一棵红黑树    
33. {    
34.     pTree->root = PNIL;    
35. }     
36.   
37. // 查找最小键值节点    
38. RB_Node* RBTREE_MIN(RB_Node* pRoot)    
39. {    
40.     while (PNIL != pRoot->left)  
41.     {  
42.         pRoot = pRoot->left;  
43.     }    
44.     return pRoot;  
45. }  
46.   
47.   
48. /* 
49.                           15 
50.                         /    \ 
51.                        /      \ 
52.                       /        \ 
53.                      6          18 
54.                     /  \       /  \ 
55.                    /    \     /    \ 
56.                   3      7   17    20 
57.                  /  \     \ 
58.                 /    \     \ 
59.                2      4     13 
60.                             / 
61.                            / 
62.                           9 
63. */  
64. // 查找指定节点的后继节点    
65. RB_Node* RBTREE_SUCCESSOR(RB_Node*  pRoot)    
66. {    
67.     if (PNIL != pRoot->right)    // 查找图中6的后继节点时就调用RBTREE_MIN函数  
68.     {    
69.         return RBTREE_MIN(pRoot->right);    
70.     }  
71.     // 节点没有右子树的时候，进入下面的while循环（如查找图中13的后继节点时，它的后继节点是15）  
72.     RB_Node* pParent = pRoot->parent;    
73.     while((PNIL != pParent) && (pRoot == pParent->right))  
74.     {    
75.         pRoot = pParent;  
76.         pParent = pRoot->parent;        
77.     }  
78.     return pParent;  
79. }  
80.   
81. // 红黑树的节点删除  
82. RB_Node* Delete(RB_Tree pTree , RB_Node* pDel)    
83. {    
84.     RB_Node* rel_delete_point;  
85.     if(pDel->left == PNIL || pDel->right == PNIL)  
86.         rel_delete_point = pDel;  
87.     else  
88.         rel_delete_point = RBTREE_SUCCESSOR(pDel);     // 查找后继节点  
89.   
90.     RB_Node* delete_point_child;    
91.     if(rel_delete_point->right != PNIL)    
92.     {    
93.         delete_point_child = rel_delete_point->right;    
94.     }    
95.     else if(rel_delete_point->left != PNIL)    
96.     {    
97.         delete_point_child = rel_delete_point->left;    
98.     }    
99.     else    
100.     {    
101.         delete_point_child = PNIL;    
102.     }    
103.     delete_point_child->parent = rel_delete_point->parent;    
104.     if(rel_delete_point->parent == PNIL)    // 删除的节点是根节点  
105.     {    
106.         pTree->root = delete_point_child;  
107.     }    
108.     else if(rel_delete_point == rel_delete_point->parent->right)  
109.     {    
110.         rel_delete_point->parent->right = delete_point_child;    
111.     }    
112.     else    
113.     {    
114.         rel_delete_point->parent->left = delete_point_child;    
115.     }  
116.     if(pDel != rel_delete_point)  
117.     {  
118.         pDel->key = rel_delete_point->key;  
119.     }  
120.     if(rel_delete_point->RB_COLOR == BLACK)    
121.     {    
122.         DeleteFixUp(pTree , delete_point_child);    
123.     }  
124.     return rel_delete_point;    
125. }    
126.                
127.   
128. /* 
129. 算法导论上的描述如下： 
130. RB-DELETE-FIXUP(T, x)  
131. 1 while x ≠ root[T] and color[x] = BLACK  
132. 2     do if x = left[p[x]]  
133. 3           then w ← right[p[x]]  
134. 4                if color[w] = RED  
135. 5                   then color[w] ← BLACK                           Case 1  
136. 6                        color[p[x]] ← RED                          Case 1  
137. 7                        LEFT-ROTATE(T, p[x])                       Case 1  
138. 8                        w ← right[p[x]]                            Case 1  
139. 9                if color[left[w]] = BLACK and color[right[w]] = BLACK  
140. 10                   then color[w] ← RED                            Case 2  
141. 11                        x p[x]                                    Case 2  
142. 12                   else if color[right[w]] = BLACK  
143. 13                           then color[left[w]] ← BLACK            Case 3  
144. 14                                color[w] ← RED                    Case 3  
145. 15                                RIGHT-ROTATE(T, w)                Case 3  
146. 16                                w ← right[p[x]]                   Case 3  
147. 17                         color[w] ← color[p[x]]                   Case 4  
148. 18                         color[p[x]] ← BLACK                      Case 4  
149. 19                         color[right[w]] ← BLACK                  Case 4  
150. 20                         LEFT-ROTATE(T, p[x])                     Case 4  
151. 21                         x ← root[T]                              Case 4  
152. 22        else (same as then clause with "right" and "left" exchanged)  
153. 23 color[x] ← BLACK   
154. */    
155. //接下来的工作，很简单，即把上述伪代码改写成c++代码即可    
156. void DeleteFixUp(RB_Tree pTree , RB_Node* node)    
157. {    
158.     while(node != pTree->root && node->RB_COLOR == BLACK)    
159.     {    
160.         if(node == node->parent->left)    
161.         {    
162.             RB_Node* brother = node->parent->right;    
163.             if(brother->RB_COLOR==RED)   //情况1：x的兄弟w是红色的。    
164.             {    
165.                 brother->RB_COLOR = BLACK;    
166.                 node->parent->RB_COLOR = RED;    
167.                 RotateLeft(node->parent);    
168.             }    
169.             else     //情况2：x的兄弟w是黑色的，    
170.             {    
171.                 if(brother->left->RB_COLOR == BLACK && brother->right->RB_COLOR == BLACK)  //w的两个孩子都是黑色的    
172.                 {    
173.                     brother->RB_COLOR = RED;    
174.                     node = node->parent;    
175.                 }    
176.                 else  
177.                 {  
178.                     if(brother->right->RB_COLOR == BLACK)   //情况3：x的兄弟w是黑色的，w的右孩子是黑色（w的左孩子是红色）。    
179.                     {  
180.                         brother->RB_COLOR = RED;  
181.                         brother->left->RB_COLOR = BLACK;  
182.                         RotateRight(brother);  
183.                         brother = node->parent->right;      //情况3转换为情况4  
184.                     }  
185.                     //情况4：x的兄弟w是黑色的，且w的右孩子时红色的  
186.                     brother->RB_COLOR = node->parent->RB_COLOR;    
187.                     node->parent->RB_COLOR = BLACK;    
188.                     brother->right->RB_COLOR = BLACK;    
189.                     RotateLeft(node->parent);    
190.                     node = pTree->root;  
191.                 }//else  
192.             }//else  
193.         }    
194.         else   //同上，原理一致，只是遇到左旋改为右旋，遇到右旋改为左旋即可。其它代码不变。    
195.         {    
196.             RB_Node* brother = node->parent->left;    
197.             if(brother->RB_COLOR == RED)    
198.             {    
199.                 brother->RB_COLOR = BLACK;    
200.                 node->parent->RB_COLOR = RED;    
201.                 RotateRight(node->parent);    
202.             }    
203.             else    
204.             {    
205.                 if(brother->left->RB_COLOR==BLACK && brother->right->RB_COLOR == BLACK)    
206.                 {    
207.                     brother->RB_COLOR = RED;    
208.                     node = node->parent;    
209.                 }    
210.                 else  
211.                 {  
212.                     if(brother->left->RB_COLOR==BLACK)    
213.                     {    
214.                         brother->RB_COLOR = RED;    
215.                         brother->right->RB_COLOR = BLACK;    
216.                         RotateLeft(brother);  
217.                         brother = node->parent->left;      //情况3转换为情况4  
218.                     }  
219.                     brother->RB_COLOR = node->parent->RB_COLOR;    
220.                     node->parent->RB_COLOR = BLACK;    
221.                     brother->left->RB_COLOR = BLACK;    
222.                     RotateRight(node->parent);    
223.                     node = pTree->root;    
224.                 }    
225.             }    
226.         }    
227.     }//while   
228.     node->RB_COLOR = BLACK;    //如果X节点原来为红色，那么直接改为黑色    
229. }