apaki/unicstl
1
0
Fork 0
mirror of https://gitee.com/apaki/unicstl.git synced 2025-07-03 23:56:54 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

修复一个bug，红黑树的左右旋直接调用了AVL树的左右旋转，里面有平衡参数的设置，导致冲突。

Browse Source
This commit is contained in:
建峰 2024-07-01 11:09:38 +08:00
parent bbc21c85a3
commit e3f59e2b90
1 changed files with 87 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

96
src/tree.c
View File

@ -1062,6 +1062,80 @@ bool tree_set_color(struct _tree_node* node, rbt_color color)
} }
static struct _tree_node* tree_rb_turn_left(struct _tree* self, struct _tree_node* root)
{
assert(self != NULL);
assert(root != NULL);
struct _tree_node* node = root->right;
if(node == NULL)
{
return root;
}
if(root->parent == NULL)
{
self->_root = node; // step1
}
else
{
if(root->parent->left == root)
{
root->parent->left = node; // step1
}
else if(root->parent->right == root)
{
root->parent->right = node; // setp1
}
}
node->parent = root->parent; // step2
root->parent = node; // step3
root->right = node->left; // step4
if(node->left != NULL)
{
node->left->parent = root; // step5
}
node->left = root; // step6
return node;
}
static struct _tree_node* tree_rb_turn_right(struct _tree* self, struct _tree_node* root)
{
assert(self != NULL);
assert(root != NULL);
struct _tree_node* node = root->left;
if(node == NULL)
{
return root;
}
if(root->parent == NULL)
{
self->_root = node; // step1
}
else
{
if(root->parent->left == root)
{
root->parent->left = node; // step1
}
else if(root->parent->right == root)
{
root->parent->right = node; // setp1
}
}
node->parent = root->parent; // step2
root->parent = node; // step3
root->left = node->right; // step4
if(node->right != NULL)
{
node->right->parent = root; // step5
}
node->right = root; // step6
return node;
}
bool tree_rb_insert(struct _tree* self, void* obj) bool tree_rb_insert(struct _tree* self, void* obj)
{ {
assert(self != NULL); assert(self != NULL);
@ -1167,11 +1241,11 @@ static bool tree_rb_rebalance(struct _tree* self, struct _tree_node* node)
{ {
if(node == father->right) if(node == father->right)
{ {
node = tree_turn_left(self, father); node = tree_rb_turn_left(self, father);
} }
father->color = RBT_BLACK; father->color = RBT_BLACK;
grandfather->color = RBT_RED; grandfather->color = RBT_RED;
node = tree_turn_right(self, grandfather); node = tree_rb_turn_right(self, grandfather);
break; break;
} }
} }
@ -1189,11 +1263,11 @@ static bool tree_rb_rebalance(struct _tree* self, struct _tree_node* node)
{ {
if(node == father->left) if(node == father->left)
{ {
node = tree_turn_right(self, father); node = tree_rb_turn_right(self, father);
} }
father->color = RBT_BLACK; father->color = RBT_BLACK;
grandfather->color = RBT_RED; grandfather->color = RBT_RED;
node = tree_turn_left(self, grandfather); node = tree_rb_turn_left(self, grandfather);
break; break;
} }
} }
@ -1233,7 +1307,7 @@ bool tree_rb_delete_fix(struct _tree* self, struct _tree_node* node)
// so ... // so ...
brother->color = RBT_BLACK; brother->color = RBT_BLACK;
father->color = RBT_RED; father->color = RBT_RED;
tmp = tree_turn_left(self, father); tmp = tree_rb_turn_left(self, father);
// After deleting the node, it became unbalanced // After deleting the node, it became unbalanced
// so convert to case5 // so convert to case5
} }
@ -1243,7 +1317,7 @@ bool tree_rb_delete_fix(struct _tree* self, struct _tree_node* node)
brother->color = father->color; brother->color = father->color;
father->color = RBT_BLACK; father->color = RBT_BLACK;
brother->right->color = RBT_BLACK; brother->right->color = RBT_BLACK;
node = tree_turn_left(self, father); node = tree_rb_turn_left(self, father);
// After deleting the node, it remains balanced // After deleting the node, it remains balanced
break; break;
} }
@ -1252,7 +1326,7 @@ bool tree_rb_delete_fix(struct _tree* self, struct _tree_node* node)
// case 3 // case 3
brother->color = RBT_RED; brother->color = RBT_RED;
brother->left->color = RBT_BLACK; brother->left->color = RBT_BLACK;
tmp = tree_turn_right(self, brother); tmp = tree_rb_turn_right(self, brother);
// Convert to case2 // Convert to case2
} }
else else
@ -1268,6 +1342,8 @@ bool tree_rb_delete_fix(struct _tree* self, struct _tree_node* node)
else else
{ {
// case 5 // case 5
// father is red, brother has no children
// if delete the node, it remains balanced
brother->color = RBT_RED; brother->color = RBT_RED;
father->color = RBT_BLACK; father->color = RBT_BLACK;
break; break;
@ -1283,7 +1359,7 @@ bool tree_rb_delete_fix(struct _tree* self, struct _tree_node* node)
// case1 // case1
brother->color = RBT_BLACK; brother->color = RBT_BLACK;
father->color = RBT_RED; father->color = RBT_RED;
tmp = tree_turn_right(self, father); tmp = tree_rb_turn_right(self, father);
} }
else if(brother->left != NULL && brother->left->color == RBT_RED) else if(brother->left != NULL && brother->left->color == RBT_RED)
{ {
@ -1291,7 +1367,7 @@ bool tree_rb_delete_fix(struct _tree* self, struct _tree_node* node)
brother->color = father->color; brother->color = father->color;
father->color = RBT_BLACK; father->color = RBT_BLACK;
brother->left->color = RBT_BLACK; brother->left->color = RBT_BLACK;
node = tree_turn_right(self, father); node = tree_rb_turn_right(self, father);
break; break;
} }
else if(brother->right != NULL && brother->right->color == RBT_RED) else if(brother->right != NULL && brother->right->color == RBT_RED)
@ -1299,7 +1375,7 @@ bool tree_rb_delete_fix(struct _tree* self, struct _tree_node* node)
// case3 // case3
brother->color = RBT_RED; brother->color = RBT_RED;
brother->right->color = RBT_BLACK; brother->right->color = RBT_BLACK;
tmp = tree_turn_left(self, brother); tmp = tree_rb_turn_left(self, brother);
// convert to case2 // convert to case2
} }
else else