forked from vieter-v/libvieter
feat(tree): started standard binary tree
This commit is contained in:
parent
df50f96983
commit
98f158e1f5
6 changed files with 182 additions and 7 deletions
61
src/tree/node.c
Normal file
61
src/tree/node.c
Normal file
|
|
@ -0,0 +1,61 @@
|
|||
#include "node.h"
|
||||
|
||||
vieter_tree_node *vieter_tree_node_init() {
|
||||
return calloc(1, sizeof(vieter_tree_node));
|
||||
}
|
||||
|
||||
vieter_tree_error vieter_tree_node_insert(vieter_tree_node *node, uint64_t key,
|
||||
void *data) {
|
||||
vieter_tree_node *parent = node;
|
||||
|
||||
while (node != NULL) {
|
||||
if (node->key == key) {
|
||||
return vieter_tree_already_present;
|
||||
}
|
||||
|
||||
parent = node;
|
||||
|
||||
if (key < node->key) {
|
||||
node = parent->left;
|
||||
} else {
|
||||
node = parent->right;
|
||||
}
|
||||
}
|
||||
|
||||
vieter_tree_node *new_node = vieter_tree_node_init();
|
||||
new_node->key = key;
|
||||
new_node->data = data;
|
||||
new_node->parent = parent;
|
||||
|
||||
if (key < node->key) {
|
||||
parent->left = new_node;
|
||||
} else {
|
||||
parent->right = new_node;
|
||||
}
|
||||
|
||||
return vieter_tree_ok;
|
||||
}
|
||||
|
||||
vieter_tree_error vieter_tree_node_search(void **out, vieter_tree_node *node,
|
||||
uint64_t key) {
|
||||
while (node != NULL) {
|
||||
if (node->key == key) {
|
||||
*out = node->data;
|
||||
|
||||
return vieter_tree_ok;
|
||||
}
|
||||
|
||||
if (key < node->key) {
|
||||
node = node->left;
|
||||
} else {
|
||||
node = node->right;
|
||||
}
|
||||
}
|
||||
|
||||
return vieter_tree_not_present;
|
||||
}
|
||||
|
||||
vieter_tree_error vieter_tree_node_remove(void **out, vieter_tree_node *node,
|
||||
uint64_t key) {
|
||||
return vieter_tree_not_present;
|
||||
}
|
||||
|
|
@ -1,13 +1,23 @@
|
|||
#ifndef VIETER_BALANCED_TREE_NODE
|
||||
#define VIETER_BALANCED_TREE_NODE
|
||||
#ifndef VIETER_TREE_NODE
|
||||
#define VIETER_TREE_NODE
|
||||
|
||||
#include "vieter_tree.h"
|
||||
#include <stdint.h>
|
||||
|
||||
typedef struct vieter_tree_node {
|
||||
uint64_t key;
|
||||
void *data;
|
||||
struct vieter_tree_node *parent;
|
||||
struct vieter_tree_node *left;
|
||||
struct vieter_tree_node *right;
|
||||
} vieter_tree_node;
|
||||
|
||||
vieter_tree_node *vieter_tree_node_init();
|
||||
|
||||
vieter_tree_error vieter_tree_node_insert(vieter_tree_node *node, uint64_t key, void *data);
|
||||
|
||||
vieter_tree_error vieter_tree_node_search(void **out, vieter_tree_node *node, uint64_t key);
|
||||
|
||||
vieter_tree_error vieter_tree_node_remove(void **out, vieter_tree_node *node, uint64_t key);
|
||||
|
||||
#endif
|
||||
|
|
|
|||
|
|
@ -1,6 +1,59 @@
|
|||
#include "vieter_tree.h"
|
||||
#include "node.h"
|
||||
#include "vieter_tree.h"
|
||||
|
||||
struct vieter_tree {
|
||||
uint64_t size;
|
||||
uint64_t size;
|
||||
vieter_tree_node *root;
|
||||
};
|
||||
|
||||
vieter_tree *vieter_tree_init() { return calloc(1, sizeof(vieter_tree)); }
|
||||
|
||||
uint64_t vieter_tree_size(vieter_tree *tree) { return tree->size; }
|
||||
|
||||
vieter_tree_error vieter_tree_insert(vieter_tree *tree, uint64_t key,
|
||||
void *data) {
|
||||
if (tree->size == 0) {
|
||||
tree->root = vieter_tree_node_init();
|
||||
tree->root->key = key;
|
||||
tree->root->data = data;
|
||||
tree->size = 1;
|
||||
|
||||
return vieter_tree_ok;
|
||||
}
|
||||
|
||||
vieter_tree_error res = vieter_tree_node_insert(tree->root, key, data);
|
||||
|
||||
if (res != vieter_tree_ok) {
|
||||
return res;
|
||||
}
|
||||
|
||||
tree->size++;
|
||||
|
||||
return vieter_tree_ok;
|
||||
}
|
||||
|
||||
vieter_tree_error vieter_tree_search(void **out, vieter_tree *tree,
|
||||
uint64_t key) {
|
||||
if (tree->size == 0) {
|
||||
return vieter_tree_not_present;
|
||||
}
|
||||
|
||||
return vieter_tree_node_search(out, tree->root, key);
|
||||
}
|
||||
|
||||
vieter_tree_error vieter_tree_remove(void **out, vieter_tree *tree,
|
||||
uint64_t key) {
|
||||
if (tree->size == 0) {
|
||||
return vieter_tree_not_present;
|
||||
}
|
||||
|
||||
vieter_tree_error res = vieter_tree_node_remove(out, tree->root, key);
|
||||
|
||||
if (res != vieter_tree_ok) {
|
||||
return res;
|
||||
}
|
||||
|
||||
tree->size--;
|
||||
|
||||
return vieter_tree_ok;
|
||||
}
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue