feat(tree): started standard binary tree

This commit is contained in:
Jef Roosens 2023-01-19 22:23:19 +01:00
parent df50f96983
commit 98f158e1f5
Signed by: Jef Roosens
GPG key ID: B75D4F293C7052DB
6 changed files with 182 additions and 7 deletions

61
src/tree/node.c Normal file
View 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;
}

View file

@ -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

View file

@ -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;
}