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

tree修改node的new和free接口函数

Browse Source
This commit is contained in:
建峰 2025-04-27 23:36:42 +08:00
parent eb918955f5
commit 4530508a1b
1 changed files with 47 additions and 53 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

100
src/tree.c
View File

@ -12,6 +12,48 @@
#include "queue.h" #include "queue.h"
#include "stack.h" #include "stack.h"
static struct _tree_node* tree_node_new(struct _tree* self, void* obj)
{
assert(self != NULL);
void* obj_new = malloc(self->_obj_size);
if (obj_new == NULL)
{
return NULL;
}
memmove(obj_new, obj, self->_obj_size);
struct _tree_node* node_new = (struct _tree_node*)malloc(sizeof(struct _tree_node));
if (node_new == NULL)
{
free(obj_new);
return NULL;
}
node_new->obj = obj_new;
node_new->parent = NULL;
node_new->left = NULL;
node_new->right = NULL;
node_new->balance = 0;
return node_new;
}
static void tree_node_free(struct _tree_node** node)
{
if (node != NULL && (*node) != NULL)
{
if ((*node)->obj != NULL)
{
free((*node)->obj);
}
free(*node);
*node = NULL;
}
else
{
printf("+++\n");
}
}
static uint32_t tree_height_node(struct _tree* self, struct _tree_node* root) static uint32_t tree_height_node(struct _tree* self, struct _tree_node* root)
{ {
@ -378,54 +420,6 @@ static struct _tree_node* tree_find_max(struct _tree* self, struct _tree_node* r
#endif #endif
} }
static struct _tree_node* tree_node_new(struct _tree* self, void* obj)
{
assert(self != NULL);
void* obj_new = malloc(self->_obj_size);
if (obj_new == NULL)
{
goto done;
}
memmove(obj_new, obj, self->_obj_size);
struct _tree_node* node_new = (struct _tree_node*)malloc(sizeof(struct _tree_node));
if (node_new == NULL)
{
goto done;
}
node_new->obj = obj_new;
node_new->parent = NULL;
node_new->left = NULL;
node_new->right = NULL;
node_new->balance = 0;
return node_new;
done:
if (obj_new != NULL)
{
free(obj_new);
}
if (node_new != NULL)
{
free(node_new);
}
return NULL;
}
static bool tree_node_free(struct _tree_node* node)
{
if (node != NULL)
{
if (node->obj != NULL)
{
free(node->obj);
}
free(node);
}
return true;
}
/** /**
* @brief find the position to insert or find object * @brief find the position to insert or find object
* *
@ -499,7 +493,7 @@ static bool tree_avl_insert(struct _tree* self, void* obj)
else else
{ {
// if obj exist, just return false // if obj exist, just return false
tree_node_free(node); tree_node_free(&node);
return false; return false;
} }
@ -570,7 +564,7 @@ static bool tree_avl_delete_single_child(struct _tree* self, struct _tree_node*
self->_rebalance(self, node->parent); self->_rebalance(self, node->parent);
} }
tree_node_free(node); tree_node_free(&node);
return true; return true;
} }
@ -643,7 +637,7 @@ static bool tree_clear(struct _tree* self)
{ {
queue->push(queue, &node->right); queue->push(queue, &node->right);
} }
tree_node_free(node); tree_node_free(&node);
} }
queue_free(&queue); queue_free(&queue);
self->_root = NULL; self->_root = NULL;
@ -828,7 +822,7 @@ static bool tree_rb_insert(struct _tree* self, void* obj)
else else
{ {
// if obj exist, just return false // if obj exist, just return false
tree_node_free(node); tree_node_free(&node);
return false; return false;
} }
@ -1146,7 +1140,7 @@ static bool tree_rb_delete(struct _tree* self, void* obj)
self->_root = NULL; self->_root = NULL;
} }
tree_node_free(tmp); tree_node_free(&tmp);
self->_size--; self->_size--;
return true; return true;
} }