Showing
2 changed files
with
4 additions
and
142 deletions
... | ... | @@ -104,97 +104,9 @@ TR_treeDelete(TR_Tree * this, const void * search, TR_TreeComp comp) |
104 | 104 | * rebalancing process...(this does not make much sense, but |
105 | 105 | * to be honest I don't know now.) |
106 | 106 | */ |
107 | - while(1) { | |
108 | - TR_Tree sibling; | |
109 | - | |
110 | - // case 1 | |
111 | - if (NULL == TR_TREE_PARENT(node)) { | |
112 | - break; | |
113 | - } | |
114 | - | |
115 | - sibling = TR_TREE_SIBLING(node); | |
116 | - if (NULL != sibling && rbRed == sibling->color) { | |
117 | - // case 2 | |
118 | - node->parent->color = rbRed; | |
119 | - sibling->color = rbBlack; | |
120 | - | |
121 | - if (NULL != node->parent->right && node != node->parent->right) { | |
122 | - TR_TREE_ROTATE(left, this, node->parent); | |
123 | - } else { | |
124 | - TR_TREE_ROTATE(right, this, node->parent); | |
125 | - } | |
126 | - sibling = TR_TREE_SIBLING(node); | |
127 | - } | |
128 | - | |
129 | - if (NULL == sibling | |
130 | - || (rbBlack == sibling->color | |
131 | - && TR_TREE_NODE_BLACK(sibling->left) | |
132 | - && TR_TREE_NODE_BLACK(sibling->right))) { | |
133 | - // case 3/4 | |
134 | - if (NULL != sibling) { | |
135 | - sibling->color = rbRed; | |
136 | - } | |
137 | - | |
138 | - if (rbBlack == node->parent->color) { | |
139 | - // case 3 | |
140 | - node = node->parent; | |
141 | - continue; | |
142 | - } else { | |
143 | - // case 4 | |
144 | - node->parent->color = rbBlack; | |
145 | - break; | |
146 | - } | |
147 | - } | |
148 | - | |
149 | - // case 5 | |
150 | - if (NULL != sibling && rbBlack == sibling->color) { | |
151 | - | |
152 | - if (node == node->parent->left | |
153 | - && TR_TREE_NODE_BLACK(sibling->right) | |
154 | - && TR_TREE_NODE_STRICT_RED(sibling->left)) { | |
155 | - | |
156 | - sibling->color = rbRed; | |
157 | - sibling->left->color = rbBlack; | |
158 | - | |
159 | - TR_TREE_ROTATE(right, this, sibling); | |
160 | - | |
161 | - } else if (node == node->parent->right | |
162 | - && TR_TREE_NODE_BLACK(sibling->left) | |
163 | - && TR_TREE_NODE_STRICT_RED(sibling->right)) { | |
164 | - | |
165 | - sibling->color = rbRed; | |
166 | - sibling->right->color = rbBlack; | |
167 | - | |
168 | - TR_TREE_ROTATE(left, this, sibling); | |
169 | - } | |
170 | - sibling = TR_TREE_SIBLING(node); | |
171 | - } | |
172 | - | |
173 | - // case 6 | |
174 | - // do we need to reinitialize sibling here? | |
175 | - if (NULL != sibling) { | |
176 | - sibling->color = node->parent->color; | |
177 | - } | |
178 | - | |
179 | - if (NULL != node && NULL != node->parent) { | |
180 | - node->parent->color = rbBlack; | |
181 | - | |
182 | - if (NULL != node->parent->right | |
183 | - && node != node->parent->right) { | |
184 | - | |
185 | - if (NULL != sibling->right) { | |
186 | - sibling->right->color = rbBlack; | |
187 | - } | |
188 | - TR_TREE_ROTATE(left, this, node->parent); | |
189 | - } else { | |
190 | - if (NULL != sibling->left) { | |
191 | - sibling->left->color = rbBlack; | |
192 | - } | |
193 | - TR_TREE_ROTATE(right, this, node->parent); | |
194 | - } | |
195 | - } | |
196 | - | |
197 | - break; | |
107 | + { | |
108 | + TR_Tree sibling = TR_TREE_SIBLING(node); | |
109 | + TR_TREE_BALANCE_DELETE(this, node, sibling); | |
198 | 110 | } |
199 | 111 | |
200 | 112 | TR_delete(del_node); | ... | ... |
... | ... | @@ -69,57 +69,7 @@ TR_treeInsert(TR_Tree * this, const void * search, TR_TreeComp comp) |
69 | 69 | * we expect node not to be NULL and pointing to our |
70 | 70 | * new node at this point...now rabalance the tree |
71 | 71 | */ |
72 | - while (1) { | |
73 | - // case 1 | |
74 | - if (NULL == TR_TREE_PARENT(node)) { | |
75 | - node->color = rbBlack; | |
76 | - break; | |
77 | - } | |
78 | - | |
79 | - // case 2 | |
80 | - if (rbBlack == node->parent->color) { | |
81 | - break; | |
82 | - } | |
83 | - | |
84 | - // case 3 | |
85 | - TR_Tree uncle = TR_TREE_UNCLE(node); | |
86 | - | |
87 | - if (NULL != uncle && rbRed == uncle->color) { | |
88 | - node->parent->color = rbBlack; | |
89 | - uncle->color = rbBlack; | |
90 | - node->parent->parent->color = rbRed; | |
91 | - | |
92 | - node = node->parent->parent; | |
93 | - continue; | |
94 | - } | |
95 | - | |
96 | - // case 4 | |
97 | - if (node == node->parent->right | |
98 | - && node->parent == node->parent->parent->left) { | |
99 | - | |
100 | - TR_TREE_ROTATE(left, this, node->parent); | |
101 | - node = node->left; | |
102 | - | |
103 | - } else if (node == node->parent->left | |
104 | - && node->parent == node->parent->parent->right) { | |
105 | - | |
106 | - TR_TREE_ROTATE(right, this, node->parent); | |
107 | - node = node->right; | |
108 | - | |
109 | - } | |
110 | - | |
111 | - // case 5 | |
112 | - node->parent->color = rbBlack; | |
113 | - node->parent->parent->color = rbRed; | |
114 | - | |
115 | - if (node == node->parent->left) { | |
116 | - TR_TREE_ROTATE(right, this, node->parent->parent); | |
117 | - } else { | |
118 | - TR_TREE_ROTATE(left, this, node->parent->parent); | |
119 | - } | |
120 | - | |
121 | - break; | |
122 | - } | |
72 | + TR_TREE_BALANCE_INSERT(this, node); | |
123 | 73 | |
124 | 74 | return new_node->data; |
125 | 75 | } | ... | ... |
Please
register
or
login
to post a comment