apaki/unicstl
1
0
Fork 0
mirror of https://gitee.com/apaki/unicstl.git synced 2025-05-18 03:51:35 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

将avl_rebalance函数也修改成非递归的实现方式

Browse Source
This commit is contained in:
建峰 2024-06-25 14:12:21 +08:00
parent f744410a4c
commit 19ac6dbf49
2 changed files with 56 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
.vscode/settings.json vendored
View File

@ -15,6 +15,7 @@
"deque.h": "c",
"config.h": "c",
"limits": "c",
"*.tcc": "c"
"*.tcc": "c",
"cstdlib": "c"
}
}

56
src/tree.c
View File

@ -200,14 +200,15 @@ int32_t tree_height(struct _tree* self, struct _tree_node* root)
*/
static bool tree_avl_rebalance(struct _tree* self, struct _tree_node* root)
{
#if 0
assert(self != NULL);
if(root == NULL)
{
return true;
return false;
}
if(root->left == NULL && root->right == NULL)
{
return true;
return false;
}
// self->print_obj(root->obj);
@ -245,6 +246,56 @@ static bool tree_avl_rebalance(struct _tree* self, struct _tree_node* root)
self->_root = root;
}
return true;
#else
assert(self != NULL);
if(root == NULL)
{
return false;
}
if(root->left == NULL && root->right == NULL)
{
return false;
}
int balance = 0;
do
{
tree_set_balance(self, root);
balance = root->balance;
if(balance == 2)
{
if(root->right->balance >= 0)
{
root = tree_turn_left(self, root);
}
else
{
root = tree_trun_right_then_left(self, root);
}
}
else if(balance == -2)
{
if(root->left->balance <= 0)
{
root = tree_turn_right(self, root);
}
else
{
root = tree_trun_left_then_right(self, root);
}
}
// if node become the new root
if(root->parent == NULL)
{
break;
}
root = root->parent;
}while(root != NULL);
self->_root = root;
return true;
#endif
}
static struct _tree_node * tree_node_new(struct _tree* self, void* obj)
@ -993,6 +1044,7 @@ bool tree_avl_init(struct _tree *self, uint32_t obj_size)
self->min = tree_min;
self->_root = NULL;
return true;
}