diff options
author | Alex Auvolat <alex@adnab.me> | 2016-07-15 23:12:14 +0200 |
---|---|---|
committer | Alex Auvolat <alex@adnab.me> | 2016-07-15 23:12:14 +0200 |
commit | 32407e728971006ed3d0885e01c22fb66c8adc57 (patch) | |
tree | 89483d39e8e2638383f815d4e73b647334fe2fe9 /src/common/libalgo/btree.c | |
parent | ba4e59a1d687173ac5cfa74d26d71d6059dc6bc6 (diff) | |
download | kogata-32407e728971006ed3d0885e01c22fb66c8adc57.tar.gz kogata-32407e728971006ed3d0885e01c22fb66c8adc57.zip |
Move stuff around, again
Diffstat (limited to 'src/common/libalgo/btree.c')
-rw-r--r-- | src/common/libalgo/btree.c | 343 |
1 files changed, 0 insertions, 343 deletions
diff --git a/src/common/libalgo/btree.c b/src/common/libalgo/btree.c deleted file mode 100644 index d1bfd18..0000000 --- a/src/common/libalgo/btree.c +++ /dev/null @@ -1,343 +0,0 @@ -#include <malloc.h> -#include <debug.h> - -#include <btree.h> - -typedef struct btree_item { - void *key, *val; - - int height; - struct btree_item *left, *right; -} btree_item_t; - -struct btree { - key_cmp_fun_t cf; - kv_iter_fun_t releasef; - - btree_item_t *root; - int nitems; -}; - -btree_t* create_btree(key_cmp_fun_t cf, kv_iter_fun_t relf) { - btree_t* t = (btree_t*)malloc(sizeof(btree_t)); - - if (t == 0) return 0; - - t->cf = cf; - t->releasef = relf; - - t->root = 0; - t->nitems = 0; - - return t; -} - -void _btree_delete_item_rec(btree_item_t *i,kv_iter_fun_t relf) { - if (i == 0) return; - - _btree_delete_item_rec(i->left, relf); - _btree_delete_item_rec(i->right, relf); - - if (relf) relf(i->key, i->val); - free(i); -} -void delete_btree(btree_t *t) { - _btree_delete_item_rec(t->root, t->releasef); - - free(t); -} - -size_t btree_count(btree_t *i) { - return i->nitems; -} - -// ============== // -// TREE ROTATIONS // -// ============== // - -static inline int height(btree_item_t *i) { - if (i == 0) return 0; - return i->height; -} - -void btree_recalc_height(btree_item_t *i) { - if (i == 0) return; - - i->height = MAX(height(i->left), height(i->right)) + 1; -} - -btree_item_t* btree_rotate_left(btree_item_t *i) { - // I(X(a, b), Y) -> a X b I Y -> X(a, I(b, Y)) - - btree_item_t *x = i->left; - if (x == 0) return i; - btree_item_t *b = x->right; - - x->right = i; - i->left = b; - - btree_recalc_height(i); - btree_recalc_height(x); - - return x; -} - -btree_item_t* btree_rotate_right(btree_item_t *i) { - // I(X, Y(a,b)) -> X I a Y b -> Y(I(X, a), b) - - btree_item_t *y = i->right; - if (y == 0) return i; - btree_item_t *a = y->left; - - y->left = i; - i->right = a; - - btree_recalc_height(i); - btree_recalc_height(y); - - return y; -} - -btree_item_t* btree_equilibrate(btree_item_t *i) { - if (i == 0) return 0; - - while (height(i->left) - height(i->right) >= 2) - i = btree_rotate_left(i); - - while (height(i->right) - height(i->left) >= 2) - i = btree_rotate_right(i); - - return i; -} - -// =================== // -// ADDING AND DELETING // -// =================== // - -btree_item_t* _btree_insert(btree_t *t, btree_item_t *root, btree_item_t *i) { - if (root == 0) return i; - - if (t->cf(i->key, root->key) <= 0) { - root->left = _btree_insert(t, root->left, i); - } else { - root->right = _btree_insert(t, root->right, i); - } - btree_recalc_height(root); - - return btree_equilibrate(root); -} -bool btree_add(btree_t *t, void* key, void* val) { - btree_item_t *x = (btree_item_t*)malloc(sizeof(btree_item_t)); - if (x == 0) return false; - - x->key = key; - x->val = val; - x->left = x->right = 0; - btree_recalc_height(x); - - t->root = _btree_insert(t, t->root, x); - t->nitems++; - - return true; -} - -btree_item_t *_btree_extract_smallest(btree_item_t *i, btree_item_t **out_smallest) { - ASSERT(i != 0); - if (i->left == 0) { - *out_smallest = i; - return i->right; - } else { - i->left = _btree_extract_smallest(i->left, out_smallest); - btree_recalc_height(i); - return btree_equilibrate(i); - } -} -btree_item_t *_btree_remove_aux(btree_t *t, btree_item_t *i, const void* key) { - if (i == 0) return 0; - - int c = t->cf(key, i->key); - if (c < 0) { - i->left = _btree_remove_aux(t, i->left, key); - return btree_equilibrate(i); - } else if (c > 0) { - i->right = _btree_remove_aux(t, i->right, key); - return btree_equilibrate(i); - } else { - // remove item i - - btree_item_t *new_i; - if (i->right == 0 || i->left == 0) { - new_i = (i->right == 0 ? i->left : i->right); - } else { - btree_item_t *new_i_right = _btree_extract_smallest(i->right, &new_i); - - new_i->left = i->left; - new_i->right = new_i_right; - - btree_recalc_height(new_i); - new_i = btree_equilibrate(new_i); - } - - if (t->releasef) t->releasef(i->key, i->val); - free(i); - t->nitems--; - - return _btree_remove_aux(t, new_i, key); // loop because several elements may correspond - } -} -void btree_remove(btree_t *t, const void* key) { - t->root = _btree_remove_aux(t, t->root, key); -} - -btree_item_t *_btree_extract_smallest_v(btree_item_t *i, btree_item_t **out_smallest) { - ASSERT(i != 0); - if (i->left == 0) { - *out_smallest = i; - return i->right; - } else { - i->left = _btree_extract_smallest_v(i->left, out_smallest); - btree_recalc_height(i); - return btree_equilibrate(i); - } -} -btree_item_t *_btree_remove_aux_v(btree_t *t, btree_item_t *i, const void* key, const void* val) { - if (i == 0) return 0; - - int c = t->cf(key, i->key); - if (c < 0) { - i->left = _btree_remove_aux_v(t, i->left, key, val); - return btree_equilibrate(i); - } else if (c > 0) { - i->right = _btree_remove_aux_v(t, i->right, key, val); - return btree_equilibrate(i); - } else if (i->val == val) { - // remove item i - - btree_item_t *new_i; - if (i->right == 0 || i->left == 0) { - new_i = (i->right == 0 ? i->left : i->right); - } else { - btree_item_t *new_i_right = _btree_extract_smallest_v(i->right, &new_i); - - new_i->left = i->left; - new_i->right = new_i_right; - - btree_recalc_height(new_i); - new_i = btree_equilibrate(new_i); - } - - if (t->releasef) t->releasef(i->key, i->val); - free(i); - t->nitems--; - - return _btree_remove_aux_v(t, new_i, key, val); // loop because several elements may correspond - } else { - i->left = _btree_remove_aux_v(t, i->left, key, val); - i->right = _btree_remove_aux_v(t, i->right, key, val); - btree_recalc_height(i); - return btree_equilibrate(i); - } -} -void btree_remove_v(btree_t *t, const void* key, const void* val) { - t->root = _btree_remove_aux_v(t, t->root, key, val); -} - -// ======================== // -// LOOKING UP AND ITERATING // -// ======================== // - -btree_item_t *_btree_find_aux(btree_t *t, btree_item_t *i, const void* key) { - if (i == 0) return 0; - - int c = t->cf(key, i->key); - if (c == 0) { - return i; - } else if (c < 0) { - return _btree_find_aux(t, i->left, key); - } else { - return _btree_find_aux(t, i->right, key); - } -} -void* btree_find(btree_t *t, const void* key) { - - btree_item_t *i = _btree_find_aux(t, t->root, key); - - if (i == 0) return 0; - return i->val; -} - -btree_item_t *_btree_lower_aux(btree_t *t, btree_item_t *i, const void* key) { - if (i == 0) return 0; - - int c = t->cf(key, i->key); - if (c == 0) { - return i; - } else if (c < 0) { - return _btree_lower_aux(t, i->left, key); - } else { - btree_item_t *r = _btree_lower_aux(t, i->right, key); - if (r == 0) r = i; - return r; - } -} -void* btree_lower(btree_t *t, const void* key, void** actual_key) { - btree_item_t *i = _btree_lower_aux(t, t->root, key); - - if (i == 0) return 0; - if (actual_key != 0) *actual_key = i->key; - return i->val; -} - -btree_item_t *_btree_upper_aux(btree_t *t, btree_item_t *i, const void* key) { - if (i == 0) return 0; - - int c = t->cf(key, i->key); - if (c == 0) { - return i; - } else if (c < 0) { - btree_item_t *r = _btree_upper_aux(t, i->left, key); - if (r == 0) r = i; - return r; - } else { - return _btree_upper_aux(t, i->right, key); - } -} -void* btree_upper(btree_t *t, const void* key, void** actual_key) { - btree_item_t *i = _btree_upper_aux(t, t->root, key); - - if (i == 0) return 0; - if (actual_key != 0) *actual_key = i->key; - return i->val; -} - - -void _btree_iter_aux(btree_item_t *i, kv_iter_fun_t f) { - if (i == 0) return; - - _btree_iter_aux(i->left, f); - f(i->key, i->val); - _btree_iter_aux(i->right, f); -} -void btree_iter(btree_t *t, kv_iter_fun_t f) { - _btree_iter_aux(t->root, f); -} - -void _btree_iter_on_aux(btree_t *t, btree_item_t *i, const void* key, kv_iter_fun_t f) { - if (i == 0) return; - - int c = t->cf(key, i->key); - if (c == 0) { - _btree_iter_on_aux(t, i->left, key, f); - f(i->key, i->val); - _btree_iter_on_aux(t, i->right, key, f); - } else if (c < 0) { - _btree_iter_on_aux(t, i->left, key, f); - } else { - _btree_iter_on_aux(t, i->right, key, f); - } -} -void btree_iter_on(btree_t *t, const void* key, kv_iter_fun_t f) { - _btree_iter_on_aux(t, t->root, key, f); -} - -/* vim: set ts=4 sw=4 tw=0 noet :*/ |