diff options
Diffstat (limited to 'src/03/rb_tree.c')
| -rw-r--r-- | src/03/rb_tree.c | 15 |
1 files changed, 15 insertions, 0 deletions
diff --git a/src/03/rb_tree.c b/src/03/rb_tree.c index 811192d..4185da9 100644 --- a/src/03/rb_tree.c +++ b/src/03/rb_tree.c @@ -1,5 +1,6 @@ #include "rb_tree.h" #include <stdlib.h> +#include <stdio.h> RBTree *rb_tree_initialize(int value) { RBTree *tree = malloc(sizeof(RBTree)); @@ -9,3 +10,17 @@ RBTree *rb_tree_initialize(int value) { tree->value = value; return tree; } + +RBTree *rb_tree_insert(RBTree *tree, int value) { + if (tree == NULL) + return rb_tree_initialize(value); + + if (value < tree->value) { + tree->left = rb_tree_insert(tree->left, value); + } else if (value > tree->value) { + tree->right = rb_tree_insert(tree->right, value); + } else { + printf("KABOOM"); + } + return tree; +} |
