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

利用队列实现树的高度计算

Browse Source
This commit is contained in:
建峰 2024-06-25 11:45:27 +08:00
parent 229e3f1140
commit d7a05df2f4
1 changed files with 30 additions and 23 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

53
src/tree.c
View File

@ -1521,7 +1521,7 @@ static struct _tree_node* tree_trun_right_then_left(struct _tree* self, struct _
int32_t tree_height(struct _tree* self, struct _tree_node* root) int32_t tree_height(struct _tree* self, struct _tree_node* root)
{ {
#if 1 #if 0
assert(self != NULL); assert(self != NULL);
if(root == NULL) if(root == NULL)
{ {
@ -1536,34 +1536,40 @@ int32_t tree_height(struct _tree* self, struct _tree_node* root)
{ {
return 0; return 0;
} }
struct _tree_node *node = root; int32_t height = 0;
int32_t left_height = 0, right_height = 0; int32_t count_cur_level = 0;
int32_t left_height_max = 0, right_height_max = 0; int32_t count_next_level = 0;
stack_t stack = stack_new(); struct _tree_node* node = root;
stack_init(stack, sizeof(struct _tree_node*)); queue_t queue = queue_new();
stack->push(stack, &root); queue_init(queue, sizeof(struct _tree_node*));
while(!stack->empty(stack))
queue->push(queue, &node);
while(!queue->empty(queue))
{ {
stack->pop(stack, &node); queue->pop(queue, &node);
left_height = stack->size(stack);
if(left_height > left_height_max)
{
left_height_max = left_height;
}
if(node->right != NULL)
{
stack->push(stack, &node->right);
}
if(node->left != NULL) if(node->left != NULL)
{ {
stack->push(stack, &node->left); queue->push(queue, &node->left);
}
if(node->right != NULL)
{
queue->push(queue, &node->right);
}
if(count_cur_level == count_next_level)
{
count_next_level = queue->size(queue);
height++;
count_cur_level = 1;
}
else
{
count_cur_level++;
} }
} }
stack_free(stack); queue_free(queue);
return height;
return left_height_max > right_height_max ? left_height_max : right_height_max;
#endif #endif
} }
@ -2142,6 +2148,7 @@ void tree_avl_inorder(struct _tree* self, struct _tree_node* root)
{ {
stack->pop(stack, &node); stack->pop(stack, &node);
self->print_obj(node->obj); self->print_obj(node->obj);
node = node->left; node = node->left;
} }
} }