src/avl.c

Mon, 14 May 2018 18:12:14 +0200

author
Mike Becker <universe@uap-core.de>
date
Mon, 14 May 2018 18:12:14 +0200
changeset 308
d6f580621512
parent 287
98da78a1e69a
permissions
-rw-r--r--

renames ucx_strcmp() to ucx_cmp_str()

192
1e51558b9d09 updated copyright notice + added files for upcoming AVL tree implementation
Mike Becker <universe@uap-core.de>
parents:
diff changeset
1 /*
1e51558b9d09 updated copyright notice + added files for upcoming AVL tree implementation
Mike Becker <universe@uap-core.de>
parents:
diff changeset
2 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.
1e51558b9d09 updated copyright notice + added files for upcoming AVL tree implementation
Mike Becker <universe@uap-core.de>
parents:
diff changeset
3 *
259
2f5dea574a75 modules documentation
Mike Becker <universe@uap-core.de>
parents: 251
diff changeset
4 * Copyright 2017 Mike Becker, Olaf Wintermann All rights reserved.
192
1e51558b9d09 updated copyright notice + added files for upcoming AVL tree implementation
Mike Becker <universe@uap-core.de>
parents:
diff changeset
5 *
1e51558b9d09 updated copyright notice + added files for upcoming AVL tree implementation
Mike Becker <universe@uap-core.de>
parents:
diff changeset
6 * Redistribution and use in source and binary forms, with or without
1e51558b9d09 updated copyright notice + added files for upcoming AVL tree implementation
Mike Becker <universe@uap-core.de>
parents:
diff changeset
7 * modification, are permitted provided that the following conditions are met:
1e51558b9d09 updated copyright notice + added files for upcoming AVL tree implementation
Mike Becker <universe@uap-core.de>
parents:
diff changeset
8 *
1e51558b9d09 updated copyright notice + added files for upcoming AVL tree implementation
Mike Becker <universe@uap-core.de>
parents:
diff changeset
9 * 1. Redistributions of source code must retain the above copyright
1e51558b9d09 updated copyright notice + added files for upcoming AVL tree implementation
Mike Becker <universe@uap-core.de>
parents:
diff changeset
10 * notice, this list of conditions and the following disclaimer.
1e51558b9d09 updated copyright notice + added files for upcoming AVL tree implementation
Mike Becker <universe@uap-core.de>
parents:
diff changeset
11 *
1e51558b9d09 updated copyright notice + added files for upcoming AVL tree implementation
Mike Becker <universe@uap-core.de>
parents:
diff changeset
12 * 2. Redistributions in binary form must reproduce the above copyright
1e51558b9d09 updated copyright notice + added files for upcoming AVL tree implementation
Mike Becker <universe@uap-core.de>
parents:
diff changeset
13 * notice, this list of conditions and the following disclaimer in the
1e51558b9d09 updated copyright notice + added files for upcoming AVL tree implementation
Mike Becker <universe@uap-core.de>
parents:
diff changeset
14 * documentation and/or other materials provided with the distribution.
1e51558b9d09 updated copyright notice + added files for upcoming AVL tree implementation
Mike Becker <universe@uap-core.de>
parents:
diff changeset
15 *
1e51558b9d09 updated copyright notice + added files for upcoming AVL tree implementation
Mike Becker <universe@uap-core.de>
parents:
diff changeset
16 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
1e51558b9d09 updated copyright notice + added files for upcoming AVL tree implementation
Mike Becker <universe@uap-core.de>
parents:
diff changeset
17 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
1e51558b9d09 updated copyright notice + added files for upcoming AVL tree implementation
Mike Becker <universe@uap-core.de>
parents:
diff changeset
18 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
1e51558b9d09 updated copyright notice + added files for upcoming AVL tree implementation
Mike Becker <universe@uap-core.de>
parents:
diff changeset
19 * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
1e51558b9d09 updated copyright notice + added files for upcoming AVL tree implementation
Mike Becker <universe@uap-core.de>
parents:
diff changeset
20 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
1e51558b9d09 updated copyright notice + added files for upcoming AVL tree implementation
Mike Becker <universe@uap-core.de>
parents:
diff changeset
21 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
1e51558b9d09 updated copyright notice + added files for upcoming AVL tree implementation
Mike Becker <universe@uap-core.de>
parents:
diff changeset
22 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
1e51558b9d09 updated copyright notice + added files for upcoming AVL tree implementation
Mike Becker <universe@uap-core.de>
parents:
diff changeset
23 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
1e51558b9d09 updated copyright notice + added files for upcoming AVL tree implementation
Mike Becker <universe@uap-core.de>
parents:
diff changeset
24 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
1e51558b9d09 updated copyright notice + added files for upcoming AVL tree implementation
Mike Becker <universe@uap-core.de>
parents:
diff changeset
25 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
1e51558b9d09 updated copyright notice + added files for upcoming AVL tree implementation
Mike Becker <universe@uap-core.de>
parents:
diff changeset
26 * POSSIBILITY OF SUCH DAMAGE.
1e51558b9d09 updated copyright notice + added files for upcoming AVL tree implementation
Mike Becker <universe@uap-core.de>
parents:
diff changeset
27 */
1e51558b9d09 updated copyright notice + added files for upcoming AVL tree implementation
Mike Becker <universe@uap-core.de>
parents:
diff changeset
28
251
fae240d633fc changes source directory structure in preperation for autotools rollout
Mike Becker <universe@uap-core.de>
parents: 250
diff changeset
29 #include "ucx/avl.h"
243
2e74828c5e94 adds distance function and ucx_avl_find_node()
Mike Becker <universe@uap-core.de>
parents: 225
diff changeset
30
251
fae240d633fc changes source directory structure in preperation for autotools rollout
Mike Becker <universe@uap-core.de>
parents: 250
diff changeset
31 #include <limits.h>
192
1e51558b9d09 updated copyright notice + added files for upcoming AVL tree implementation
Mike Becker <universe@uap-core.de>
parents:
diff changeset
32
195
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
33 #define ptrcast(ptr) ((void*)(ptr))
216
dee5a88c4db7 added casts for mallocs in AVL implementation (to satisfy c++ compiler)
Mike Becker <universe@uap-core.de>
parents: 205
diff changeset
34 #define alloc_tree(al) (UcxAVLTree*) almalloc((al), sizeof(UcxAVLTree))
dee5a88c4db7 added casts for mallocs in AVL implementation (to satisfy c++ compiler)
Mike Becker <universe@uap-core.de>
parents: 205
diff changeset
35 #define alloc_node(al) (UcxAVLNode*) almalloc((al), sizeof(UcxAVLNode))
195
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
36
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
37 static void ucx_avl_connect(UcxAVLTree *tree,
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
38 UcxAVLNode *node, UcxAVLNode *child, intptr_t nullkey) {
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
39 if (child) {
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
40 child->parent = node;
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
41 }
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
42 // if child is NULL, nullkey decides if left or right pointer is cleared
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
43 if (tree->cmpfunc(
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
44 ptrcast(child ? child->key : nullkey),
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
45 ptrcast(node->key), tree->userdata) > 0) {
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
46 node->right = child;
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
47 } else {
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
48 node->left = child;
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
49 }
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
50 size_t lh = node->left ? node->left->height : 0;
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
51 size_t rh = node->right ? node->right->height : 0;
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
52 node->height = 1 + (lh > rh ? lh : rh);
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
53 }
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
54
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
55 #define avlheight(node) ((node) ? (node)->height : 0)
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
56
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
57 static UcxAVLNode* avl_rotright(UcxAVLTree *tree, UcxAVLNode *l0) {
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
58 UcxAVLNode *p = l0->parent;
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
59 UcxAVLNode *l1 = l0->left;
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
60 if (p) {
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
61 ucx_avl_connect(tree, p, l1, 0);
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
62 } else {
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
63 l1->parent = NULL;
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
64 }
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
65 ucx_avl_connect(tree, l0, l1->right, l1->key);
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
66 ucx_avl_connect(tree, l1, l0, 0);
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
67 return l1;
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
68 }
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
69
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
70 static UcxAVLNode* avl_rotleft(UcxAVLTree *tree, UcxAVLNode *l0) {
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
71 UcxAVLNode *p = l0->parent;
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
72 UcxAVLNode *l1 = l0->right;
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
73 if (p) {
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
74 ucx_avl_connect(tree, p, l1, 0);
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
75 } else {
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
76 l1->parent = NULL;
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
77 }
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
78 ucx_avl_connect(tree, l0, l1->left, l1->key);
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
79 ucx_avl_connect(tree, l1, l0, 0);
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
80 return l1;
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
81 }
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
82
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
83 static void ucx_avl_balance(UcxAVLTree *tree, UcxAVLNode *n) {
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
84 int lh = avlheight(n->left);
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
85 int rh = avlheight(n->right);
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
86 n->height = 1 + (lh > rh ? lh : rh);
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
87
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
88 if (lh - rh == 2) {
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
89 UcxAVLNode *c = n->left;
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
90 if (avlheight(c->right) - avlheight(c->left) == 1) {
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
91 avl_rotleft(tree, c);
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
92 }
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
93 n = avl_rotright(tree, n);
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
94 } else if (rh - lh == 2) {
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
95 UcxAVLNode *c = n->right;
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
96 if (avlheight(c->left) - avlheight(c->right) == 1) {
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
97 avl_rotright(tree, c);
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
98 }
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
99 n = avl_rotleft(tree, n);
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
100 }
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
101
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
102 if (n->parent) {
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
103 ucx_avl_balance(tree, n->parent);
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
104 } else {
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
105 tree->root = n;
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
106 }
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
107 }
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
108
194
0c1b7676e74c finalized AVL tree interface + added implementation skeleton + fixed ucx_ptrcmp()
Mike Becker <universe@uap-core.de>
parents: 192
diff changeset
109 UcxAVLTree *ucx_avl_new(cmp_func cmpfunc) {
0c1b7676e74c finalized AVL tree interface + added implementation skeleton + fixed ucx_ptrcmp()
Mike Becker <universe@uap-core.de>
parents: 192
diff changeset
110 return ucx_avl_new_a(cmpfunc, ucx_default_allocator());
0c1b7676e74c finalized AVL tree interface + added implementation skeleton + fixed ucx_ptrcmp()
Mike Becker <universe@uap-core.de>
parents: 192
diff changeset
111 }
0c1b7676e74c finalized AVL tree interface + added implementation skeleton + fixed ucx_ptrcmp()
Mike Becker <universe@uap-core.de>
parents: 192
diff changeset
112
0c1b7676e74c finalized AVL tree interface + added implementation skeleton + fixed ucx_ptrcmp()
Mike Becker <universe@uap-core.de>
parents: 192
diff changeset
113 UcxAVLTree *ucx_avl_new_a(cmp_func cmpfunc, UcxAllocator *allocator) {
216
dee5a88c4db7 added casts for mallocs in AVL implementation (to satisfy c++ compiler)
Mike Becker <universe@uap-core.de>
parents: 205
diff changeset
114 UcxAVLTree* tree = alloc_tree(allocator);
194
0c1b7676e74c finalized AVL tree interface + added implementation skeleton + fixed ucx_ptrcmp()
Mike Becker <universe@uap-core.de>
parents: 192
diff changeset
115 if (tree) {
0c1b7676e74c finalized AVL tree interface + added implementation skeleton + fixed ucx_ptrcmp()
Mike Becker <universe@uap-core.de>
parents: 192
diff changeset
116 tree->allocator = allocator;
0c1b7676e74c finalized AVL tree interface + added implementation skeleton + fixed ucx_ptrcmp()
Mike Becker <universe@uap-core.de>
parents: 192
diff changeset
117 tree->cmpfunc = cmpfunc;
0c1b7676e74c finalized AVL tree interface + added implementation skeleton + fixed ucx_ptrcmp()
Mike Becker <universe@uap-core.de>
parents: 192
diff changeset
118 tree->root = NULL;
0c1b7676e74c finalized AVL tree interface + added implementation skeleton + fixed ucx_ptrcmp()
Mike Becker <universe@uap-core.de>
parents: 192
diff changeset
119 tree->userdata = NULL;
0c1b7676e74c finalized AVL tree interface + added implementation skeleton + fixed ucx_ptrcmp()
Mike Becker <universe@uap-core.de>
parents: 192
diff changeset
120 }
0c1b7676e74c finalized AVL tree interface + added implementation skeleton + fixed ucx_ptrcmp()
Mike Becker <universe@uap-core.de>
parents: 192
diff changeset
121
0c1b7676e74c finalized AVL tree interface + added implementation skeleton + fixed ucx_ptrcmp()
Mike Becker <universe@uap-core.de>
parents: 192
diff changeset
122 return tree;
0c1b7676e74c finalized AVL tree interface + added implementation skeleton + fixed ucx_ptrcmp()
Mike Becker <universe@uap-core.de>
parents: 192
diff changeset
123 }
0c1b7676e74c finalized AVL tree interface + added implementation skeleton + fixed ucx_ptrcmp()
Mike Becker <universe@uap-core.de>
parents: 192
diff changeset
124
196
1b4cdafef2eb added free() to AVL tree implementation + use UcxAllocator
Mike Becker <universe@uap-core.de>
parents: 195
diff changeset
125 static void ucx_avl_free_node(UcxAllocator *al, UcxAVLNode *node) {
1b4cdafef2eb added free() to AVL tree implementation + use UcxAllocator
Mike Becker <universe@uap-core.de>
parents: 195
diff changeset
126 if (node) {
1b4cdafef2eb added free() to AVL tree implementation + use UcxAllocator
Mike Becker <universe@uap-core.de>
parents: 195
diff changeset
127 ucx_avl_free_node(al, node->left);
1b4cdafef2eb added free() to AVL tree implementation + use UcxAllocator
Mike Becker <universe@uap-core.de>
parents: 195
diff changeset
128 ucx_avl_free_node(al, node->right);
1b4cdafef2eb added free() to AVL tree implementation + use UcxAllocator
Mike Becker <universe@uap-core.de>
parents: 195
diff changeset
129 alfree(al, node);
1b4cdafef2eb added free() to AVL tree implementation + use UcxAllocator
Mike Becker <universe@uap-core.de>
parents: 195
diff changeset
130 }
1b4cdafef2eb added free() to AVL tree implementation + use UcxAllocator
Mike Becker <universe@uap-core.de>
parents: 195
diff changeset
131 }
1b4cdafef2eb added free() to AVL tree implementation + use UcxAllocator
Mike Becker <universe@uap-core.de>
parents: 195
diff changeset
132
1b4cdafef2eb added free() to AVL tree implementation + use UcxAllocator
Mike Becker <universe@uap-core.de>
parents: 195
diff changeset
133 void ucx_avl_free(UcxAVLTree *tree) {
1b4cdafef2eb added free() to AVL tree implementation + use UcxAllocator
Mike Becker <universe@uap-core.de>
parents: 195
diff changeset
134 UcxAllocator *al = tree->allocator;
1b4cdafef2eb added free() to AVL tree implementation + use UcxAllocator
Mike Becker <universe@uap-core.de>
parents: 195
diff changeset
135 ucx_avl_free_node(al, tree->root);
1b4cdafef2eb added free() to AVL tree implementation + use UcxAllocator
Mike Becker <universe@uap-core.de>
parents: 195
diff changeset
136 alfree(al, tree);
1b4cdafef2eb added free() to AVL tree implementation + use UcxAllocator
Mike Becker <universe@uap-core.de>
parents: 195
diff changeset
137 }
1b4cdafef2eb added free() to AVL tree implementation + use UcxAllocator
Mike Becker <universe@uap-core.de>
parents: 195
diff changeset
138
287
98da78a1e69a adds ucx_avl_free_content() function and documentation in modules.md
Mike Becker <universe@uap-core.de>
parents: 259
diff changeset
139 static void ucx_avl_free_content_node(UcxAllocator *al, UcxAVLNode *node,
98da78a1e69a adds ucx_avl_free_content() function and documentation in modules.md
Mike Becker <universe@uap-core.de>
parents: 259
diff changeset
140 ucx_destructor destr) {
98da78a1e69a adds ucx_avl_free_content() function and documentation in modules.md
Mike Becker <universe@uap-core.de>
parents: 259
diff changeset
141 if (node) {
98da78a1e69a adds ucx_avl_free_content() function and documentation in modules.md
Mike Becker <universe@uap-core.de>
parents: 259
diff changeset
142 ucx_avl_free_content_node(al, node->left, destr);
98da78a1e69a adds ucx_avl_free_content() function and documentation in modules.md
Mike Becker <universe@uap-core.de>
parents: 259
diff changeset
143 ucx_avl_free_content_node(al, node->right, destr);
98da78a1e69a adds ucx_avl_free_content() function and documentation in modules.md
Mike Becker <universe@uap-core.de>
parents: 259
diff changeset
144 if (destr) {
98da78a1e69a adds ucx_avl_free_content() function and documentation in modules.md
Mike Becker <universe@uap-core.de>
parents: 259
diff changeset
145 destr(node->value);
98da78a1e69a adds ucx_avl_free_content() function and documentation in modules.md
Mike Becker <universe@uap-core.de>
parents: 259
diff changeset
146 } else {
98da78a1e69a adds ucx_avl_free_content() function and documentation in modules.md
Mike Becker <universe@uap-core.de>
parents: 259
diff changeset
147 alfree(al, node->value);
98da78a1e69a adds ucx_avl_free_content() function and documentation in modules.md
Mike Becker <universe@uap-core.de>
parents: 259
diff changeset
148 }
98da78a1e69a adds ucx_avl_free_content() function and documentation in modules.md
Mike Becker <universe@uap-core.de>
parents: 259
diff changeset
149 }
98da78a1e69a adds ucx_avl_free_content() function and documentation in modules.md
Mike Becker <universe@uap-core.de>
parents: 259
diff changeset
150 }
98da78a1e69a adds ucx_avl_free_content() function and documentation in modules.md
Mike Becker <universe@uap-core.de>
parents: 259
diff changeset
151
98da78a1e69a adds ucx_avl_free_content() function and documentation in modules.md
Mike Becker <universe@uap-core.de>
parents: 259
diff changeset
152 void ucx_avl_free_content(UcxAVLTree *tree, ucx_destructor destr) {
98da78a1e69a adds ucx_avl_free_content() function and documentation in modules.md
Mike Becker <universe@uap-core.de>
parents: 259
diff changeset
153 ucx_avl_free_content_node(tree->allocator, tree->root, destr);
98da78a1e69a adds ucx_avl_free_content() function and documentation in modules.md
Mike Becker <universe@uap-core.de>
parents: 259
diff changeset
154 }
98da78a1e69a adds ucx_avl_free_content() function and documentation in modules.md
Mike Becker <universe@uap-core.de>
parents: 259
diff changeset
155
205
54a7ceb9151f improved avl function names
Mike Becker <universe@uap-core.de>
parents: 204
diff changeset
156 UcxAVLNode *ucx_avl_get_node(UcxAVLTree *tree, intptr_t key) {
195
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
157 UcxAVLNode *n = tree->root;
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
158 int cmpresult;
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
159 while (n && (cmpresult = tree->cmpfunc(
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
160 ptrcast(key), ptrcast(n->key), tree->userdata))) {
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
161 n = cmpresult > 0 ? n->right : n->left;
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
162 }
204
4477987d40cd better and better and better AVL API
Mike Becker <universe@uap-core.de>
parents: 203
diff changeset
163 return n;
4477987d40cd better and better and better AVL API
Mike Becker <universe@uap-core.de>
parents: 203
diff changeset
164 }
4477987d40cd better and better and better AVL API
Mike Becker <universe@uap-core.de>
parents: 203
diff changeset
165
4477987d40cd better and better and better AVL API
Mike Becker <universe@uap-core.de>
parents: 203
diff changeset
166 void *ucx_avl_get(UcxAVLTree *tree, intptr_t key) {
205
54a7ceb9151f improved avl function names
Mike Becker <universe@uap-core.de>
parents: 204
diff changeset
167 UcxAVLNode *n = ucx_avl_get_node(tree, key);
195
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
168 return n ? n->value : NULL;
194
0c1b7676e74c finalized AVL tree interface + added implementation skeleton + fixed ucx_ptrcmp()
Mike Becker <universe@uap-core.de>
parents: 192
diff changeset
169 }
0c1b7676e74c finalized AVL tree interface + added implementation skeleton + fixed ucx_ptrcmp()
Mike Becker <universe@uap-core.de>
parents: 192
diff changeset
170
243
2e74828c5e94 adds distance function and ucx_avl_find_node()
Mike Becker <universe@uap-core.de>
parents: 225
diff changeset
171 UcxAVLNode *ucx_avl_find_node(UcxAVLTree *tree, intptr_t key,
2e74828c5e94 adds distance function and ucx_avl_find_node()
Mike Becker <universe@uap-core.de>
parents: 225
diff changeset
172 distance_func dfnc, int mode) {
2e74828c5e94 adds distance function and ucx_avl_find_node()
Mike Becker <universe@uap-core.de>
parents: 225
diff changeset
173 UcxAVLNode *n = tree->root;
2e74828c5e94 adds distance function and ucx_avl_find_node()
Mike Becker <universe@uap-core.de>
parents: 225
diff changeset
174 UcxAVLNode *closest = NULL;
2e74828c5e94 adds distance function and ucx_avl_find_node()
Mike Becker <universe@uap-core.de>
parents: 225
diff changeset
175
2e74828c5e94 adds distance function and ucx_avl_find_node()
Mike Becker <universe@uap-core.de>
parents: 225
diff changeset
176 intmax_t cmpresult;
2e74828c5e94 adds distance function and ucx_avl_find_node()
Mike Becker <universe@uap-core.de>
parents: 225
diff changeset
177 intmax_t closest_dist;
2e74828c5e94 adds distance function and ucx_avl_find_node()
Mike Becker <universe@uap-core.de>
parents: 225
diff changeset
178 closest_dist = mode == UCX_AVL_FIND_LOWER_BOUNDED ? INTMAX_MIN : INTMAX_MAX;
2e74828c5e94 adds distance function and ucx_avl_find_node()
Mike Becker <universe@uap-core.de>
parents: 225
diff changeset
179
2e74828c5e94 adds distance function and ucx_avl_find_node()
Mike Becker <universe@uap-core.de>
parents: 225
diff changeset
180 while (n && (cmpresult = dfnc(
2e74828c5e94 adds distance function and ucx_avl_find_node()
Mike Becker <universe@uap-core.de>
parents: 225
diff changeset
181 ptrcast(key), ptrcast(n->key), tree->userdata))) {
2e74828c5e94 adds distance function and ucx_avl_find_node()
Mike Becker <universe@uap-core.de>
parents: 225
diff changeset
182 if (mode == UCX_AVL_FIND_CLOSEST) {
2e74828c5e94 adds distance function and ucx_avl_find_node()
Mike Becker <universe@uap-core.de>
parents: 225
diff changeset
183 intmax_t dist = cmpresult;
2e74828c5e94 adds distance function and ucx_avl_find_node()
Mike Becker <universe@uap-core.de>
parents: 225
diff changeset
184 if (dist < 0) dist *= -1;
2e74828c5e94 adds distance function and ucx_avl_find_node()
Mike Becker <universe@uap-core.de>
parents: 225
diff changeset
185 if (dist < closest_dist) {
2e74828c5e94 adds distance function and ucx_avl_find_node()
Mike Becker <universe@uap-core.de>
parents: 225
diff changeset
186 closest_dist = dist;
2e74828c5e94 adds distance function and ucx_avl_find_node()
Mike Becker <universe@uap-core.de>
parents: 225
diff changeset
187 closest = n;
2e74828c5e94 adds distance function and ucx_avl_find_node()
Mike Becker <universe@uap-core.de>
parents: 225
diff changeset
188 }
2e74828c5e94 adds distance function and ucx_avl_find_node()
Mike Becker <universe@uap-core.de>
parents: 225
diff changeset
189 } else if (mode == UCX_AVL_FIND_LOWER_BOUNDED && cmpresult <= 0) {
2e74828c5e94 adds distance function and ucx_avl_find_node()
Mike Becker <universe@uap-core.de>
parents: 225
diff changeset
190 if (cmpresult > closest_dist) {
2e74828c5e94 adds distance function and ucx_avl_find_node()
Mike Becker <universe@uap-core.de>
parents: 225
diff changeset
191 closest_dist = cmpresult;
2e74828c5e94 adds distance function and ucx_avl_find_node()
Mike Becker <universe@uap-core.de>
parents: 225
diff changeset
192 closest = n;
2e74828c5e94 adds distance function and ucx_avl_find_node()
Mike Becker <universe@uap-core.de>
parents: 225
diff changeset
193 }
2e74828c5e94 adds distance function and ucx_avl_find_node()
Mike Becker <universe@uap-core.de>
parents: 225
diff changeset
194 } else if (mode == UCX_AVL_FIND_UPPER_BOUNDED && cmpresult >= 0) {
2e74828c5e94 adds distance function and ucx_avl_find_node()
Mike Becker <universe@uap-core.de>
parents: 225
diff changeset
195 if (cmpresult < closest_dist) {
2e74828c5e94 adds distance function and ucx_avl_find_node()
Mike Becker <universe@uap-core.de>
parents: 225
diff changeset
196 closest_dist = cmpresult;
2e74828c5e94 adds distance function and ucx_avl_find_node()
Mike Becker <universe@uap-core.de>
parents: 225
diff changeset
197 closest = n;
2e74828c5e94 adds distance function and ucx_avl_find_node()
Mike Becker <universe@uap-core.de>
parents: 225
diff changeset
198 }
2e74828c5e94 adds distance function and ucx_avl_find_node()
Mike Becker <universe@uap-core.de>
parents: 225
diff changeset
199 }
2e74828c5e94 adds distance function and ucx_avl_find_node()
Mike Becker <universe@uap-core.de>
parents: 225
diff changeset
200 n = cmpresult > 0 ? n->right : n->left;
2e74828c5e94 adds distance function and ucx_avl_find_node()
Mike Becker <universe@uap-core.de>
parents: 225
diff changeset
201 }
2e74828c5e94 adds distance function and ucx_avl_find_node()
Mike Becker <universe@uap-core.de>
parents: 225
diff changeset
202 return n ? n : closest;
2e74828c5e94 adds distance function and ucx_avl_find_node()
Mike Becker <universe@uap-core.de>
parents: 225
diff changeset
203 }
2e74828c5e94 adds distance function and ucx_avl_find_node()
Mike Becker <universe@uap-core.de>
parents: 225
diff changeset
204
2e74828c5e94 adds distance function and ucx_avl_find_node()
Mike Becker <universe@uap-core.de>
parents: 225
diff changeset
205 void *ucx_avl_find(UcxAVLTree *tree, intptr_t key,
2e74828c5e94 adds distance function and ucx_avl_find_node()
Mike Becker <universe@uap-core.de>
parents: 225
diff changeset
206 distance_func dfnc, int mode) {
2e74828c5e94 adds distance function and ucx_avl_find_node()
Mike Becker <universe@uap-core.de>
parents: 225
diff changeset
207 UcxAVLNode *n = ucx_avl_find_node(tree, key, dfnc, mode);
2e74828c5e94 adds distance function and ucx_avl_find_node()
Mike Becker <universe@uap-core.de>
parents: 225
diff changeset
208 return n ? n->value : NULL;
2e74828c5e94 adds distance function and ucx_avl_find_node()
Mike Becker <universe@uap-core.de>
parents: 225
diff changeset
209 }
2e74828c5e94 adds distance function and ucx_avl_find_node()
Mike Becker <universe@uap-core.de>
parents: 225
diff changeset
210
204
4477987d40cd better and better and better AVL API
Mike Becker <universe@uap-core.de>
parents: 203
diff changeset
211 int ucx_avl_put(UcxAVLTree *tree, intptr_t key, void *value) {
4477987d40cd better and better and better AVL API
Mike Becker <universe@uap-core.de>
parents: 203
diff changeset
212 return ucx_avl_put_s(tree, key, value, NULL);
4477987d40cd better and better and better AVL API
Mike Becker <universe@uap-core.de>
parents: 203
diff changeset
213 }
4477987d40cd better and better and better AVL API
Mike Becker <universe@uap-core.de>
parents: 203
diff changeset
214
4477987d40cd better and better and better AVL API
Mike Becker <universe@uap-core.de>
parents: 203
diff changeset
215 int ucx_avl_put_s(UcxAVLTree *tree, intptr_t key, void *value,
4477987d40cd better and better and better AVL API
Mike Becker <universe@uap-core.de>
parents: 203
diff changeset
216 void **oldvalue) {
195
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
217 if (tree->root) {
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
218 UcxAVLNode *n = tree->root;
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
219 int cmpresult;
197
a82f3456b76d fuck -Wparentheses
Mike Becker <universe@uap-core.de>
parents: 196
diff changeset
220 while ((cmpresult = tree->cmpfunc(
a82f3456b76d fuck -Wparentheses
Mike Becker <universe@uap-core.de>
parents: 196
diff changeset
221 ptrcast(key), ptrcast(n->key), tree->userdata))) {
195
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
222 UcxAVLNode *m = cmpresult > 0 ? n->right : n->left;
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
223 if (m) {
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
224 n = m;
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
225 } else {
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
226 break;
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
227 }
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
228 }
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
229
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
230 if (cmpresult) {
216
dee5a88c4db7 added casts for mallocs in AVL implementation (to satisfy c++ compiler)
Mike Becker <universe@uap-core.de>
parents: 205
diff changeset
231 UcxAVLNode* e = alloc_node(tree->allocator);
204
4477987d40cd better and better and better AVL API
Mike Becker <universe@uap-core.de>
parents: 203
diff changeset
232 if (e) {
4477987d40cd better and better and better AVL API
Mike Becker <universe@uap-core.de>
parents: 203
diff changeset
233 e->key = key; e->value = value; e->height = 1;
4477987d40cd better and better and better AVL API
Mike Becker <universe@uap-core.de>
parents: 203
diff changeset
234 e->parent = e->left = e->right = NULL;
4477987d40cd better and better and better AVL API
Mike Becker <universe@uap-core.de>
parents: 203
diff changeset
235 ucx_avl_connect(tree, n, e, 0);
4477987d40cd better and better and better AVL API
Mike Becker <universe@uap-core.de>
parents: 203
diff changeset
236 ucx_avl_balance(tree, n);
4477987d40cd better and better and better AVL API
Mike Becker <universe@uap-core.de>
parents: 203
diff changeset
237 return 0;
4477987d40cd better and better and better AVL API
Mike Becker <universe@uap-core.de>
parents: 203
diff changeset
238 } else {
4477987d40cd better and better and better AVL API
Mike Becker <universe@uap-core.de>
parents: 203
diff changeset
239 return 1;
4477987d40cd better and better and better AVL API
Mike Becker <universe@uap-core.de>
parents: 203
diff changeset
240 }
195
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
241 } else {
204
4477987d40cd better and better and better AVL API
Mike Becker <universe@uap-core.de>
parents: 203
diff changeset
242 if (oldvalue) {
4477987d40cd better and better and better AVL API
Mike Becker <universe@uap-core.de>
parents: 203
diff changeset
243 *oldvalue = n->value;
4477987d40cd better and better and better AVL API
Mike Becker <universe@uap-core.de>
parents: 203
diff changeset
244 }
195
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
245 n->value = value;
204
4477987d40cd better and better and better AVL API
Mike Becker <universe@uap-core.de>
parents: 203
diff changeset
246 return 0;
195
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
247 }
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
248 } else {
216
dee5a88c4db7 added casts for mallocs in AVL implementation (to satisfy c++ compiler)
Mike Becker <universe@uap-core.de>
parents: 205
diff changeset
249 tree->root = alloc_node(tree->allocator);
204
4477987d40cd better and better and better AVL API
Mike Becker <universe@uap-core.de>
parents: 203
diff changeset
250 if (tree->root) {
4477987d40cd better and better and better AVL API
Mike Becker <universe@uap-core.de>
parents: 203
diff changeset
251 tree->root->key = key; tree->root->value = value;
4477987d40cd better and better and better AVL API
Mike Becker <universe@uap-core.de>
parents: 203
diff changeset
252 tree->root->height = 1;
4477987d40cd better and better and better AVL API
Mike Becker <universe@uap-core.de>
parents: 203
diff changeset
253 tree->root->parent = tree->root->left = tree->root->right = NULL;
4477987d40cd better and better and better AVL API
Mike Becker <universe@uap-core.de>
parents: 203
diff changeset
254
4477987d40cd better and better and better AVL API
Mike Becker <universe@uap-core.de>
parents: 203
diff changeset
255 if (oldvalue) {
4477987d40cd better and better and better AVL API
Mike Becker <universe@uap-core.de>
parents: 203
diff changeset
256 *oldvalue = NULL;
4477987d40cd better and better and better AVL API
Mike Becker <universe@uap-core.de>
parents: 203
diff changeset
257 }
4477987d40cd better and better and better AVL API
Mike Becker <universe@uap-core.de>
parents: 203
diff changeset
258
4477987d40cd better and better and better AVL API
Mike Becker <universe@uap-core.de>
parents: 203
diff changeset
259 return 0;
4477987d40cd better and better and better AVL API
Mike Becker <universe@uap-core.de>
parents: 203
diff changeset
260 } else {
4477987d40cd better and better and better AVL API
Mike Becker <universe@uap-core.de>
parents: 203
diff changeset
261 return 1;
4477987d40cd better and better and better AVL API
Mike Becker <universe@uap-core.de>
parents: 203
diff changeset
262 }
195
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
263 }
194
0c1b7676e74c finalized AVL tree interface + added implementation skeleton + fixed ucx_ptrcmp()
Mike Becker <universe@uap-core.de>
parents: 192
diff changeset
264 }
0c1b7676e74c finalized AVL tree interface + added implementation skeleton + fixed ucx_ptrcmp()
Mike Becker <universe@uap-core.de>
parents: 192
diff changeset
265
204
4477987d40cd better and better and better AVL API
Mike Becker <universe@uap-core.de>
parents: 203
diff changeset
266 int ucx_avl_remove(UcxAVLTree *tree, intptr_t key) {
4477987d40cd better and better and better AVL API
Mike Becker <universe@uap-core.de>
parents: 203
diff changeset
267 return ucx_avl_remove_s(tree, key, NULL, NULL);
4477987d40cd better and better and better AVL API
Mike Becker <universe@uap-core.de>
parents: 203
diff changeset
268 }
4477987d40cd better and better and better AVL API
Mike Becker <universe@uap-core.de>
parents: 203
diff changeset
269
205
54a7ceb9151f improved avl function names
Mike Becker <universe@uap-core.de>
parents: 204
diff changeset
270 int ucx_avl_remove_node(UcxAVLTree *tree, UcxAVLNode *node) {
204
4477987d40cd better and better and better AVL API
Mike Becker <universe@uap-core.de>
parents: 203
diff changeset
271 return ucx_avl_remove_s(tree, node->key, NULL, NULL);
4477987d40cd better and better and better AVL API
Mike Becker <universe@uap-core.de>
parents: 203
diff changeset
272 }
4477987d40cd better and better and better AVL API
Mike Becker <universe@uap-core.de>
parents: 203
diff changeset
273
4477987d40cd better and better and better AVL API
Mike Becker <universe@uap-core.de>
parents: 203
diff changeset
274 int ucx_avl_remove_s(UcxAVLTree *tree, intptr_t key,
4477987d40cd better and better and better AVL API
Mike Becker <universe@uap-core.de>
parents: 203
diff changeset
275 intptr_t *oldkey, void **oldvalue) {
4477987d40cd better and better and better AVL API
Mike Becker <universe@uap-core.de>
parents: 203
diff changeset
276
195
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
277 UcxAVLNode *n = tree->root;
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
278 int cmpresult;
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
279 while (n && (cmpresult = tree->cmpfunc(
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
280 ptrcast(key), ptrcast(n->key), tree->userdata))) {
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
281 n = cmpresult > 0 ? n->right : n->left;
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
282 }
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
283 if (n) {
204
4477987d40cd better and better and better AVL API
Mike Becker <universe@uap-core.de>
parents: 203
diff changeset
284 if (oldkey) {
4477987d40cd better and better and better AVL API
Mike Becker <universe@uap-core.de>
parents: 203
diff changeset
285 *oldkey = n->key;
4477987d40cd better and better and better AVL API
Mike Becker <universe@uap-core.de>
parents: 203
diff changeset
286 }
4477987d40cd better and better and better AVL API
Mike Becker <universe@uap-core.de>
parents: 203
diff changeset
287 if (oldvalue) {
4477987d40cd better and better and better AVL API
Mike Becker <universe@uap-core.de>
parents: 203
diff changeset
288 *oldvalue = n->value;
4477987d40cd better and better and better AVL API
Mike Becker <universe@uap-core.de>
parents: 203
diff changeset
289 }
4477987d40cd better and better and better AVL API
Mike Becker <universe@uap-core.de>
parents: 203
diff changeset
290
195
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
291 UcxAVLNode *p = n->parent;
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
292 if (n->left && n->right) {
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
293 UcxAVLNode *s = n->right;
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
294 while (s->left) {
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
295 s = s->left;
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
296 }
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
297 ucx_avl_connect(tree, s->parent, s->right, s->key);
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
298 n->key = s->key; n->value = s->value;
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
299 p = s->parent;
196
1b4cdafef2eb added free() to AVL tree implementation + use UcxAllocator
Mike Becker <universe@uap-core.de>
parents: 195
diff changeset
300 alfree(tree->allocator, s);
195
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
301 } else {
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
302 if (p) {
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
303 ucx_avl_connect(tree, p, n->right ? n->right:n->left, n->key);
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
304 } else {
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
305 tree->root = n->right ? n->right : n->left;
203
3d999ea3f780 added 1 assert in ucx_avl_remove tests and fixed source code formatting
Olaf Wintermann <olaf.wintermann@gmail.com>
parents: 202
diff changeset
306 if (tree->root) {
202
4c84dd2408d7 fixed bug in ucx_avl_remove
Olaf Wintermann <olaf.wintermann@gmail.com>
parents: 199
diff changeset
307 tree->root->parent = NULL;
4c84dd2408d7 fixed bug in ucx_avl_remove
Olaf Wintermann <olaf.wintermann@gmail.com>
parents: 199
diff changeset
308 }
195
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
309 }
196
1b4cdafef2eb added free() to AVL tree implementation + use UcxAllocator
Mike Becker <universe@uap-core.de>
parents: 195
diff changeset
310 alfree(tree->allocator, n);
195
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
311 }
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
312
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
313 if (p) {
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
314 ucx_avl_balance(tree, p);
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
315 }
204
4477987d40cd better and better and better AVL API
Mike Becker <universe@uap-core.de>
parents: 203
diff changeset
316
4477987d40cd better and better and better AVL API
Mike Becker <universe@uap-core.de>
parents: 203
diff changeset
317 return 0;
195
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
318 } else {
204
4477987d40cd better and better and better AVL API
Mike Becker <universe@uap-core.de>
parents: 203
diff changeset
319 return 1;
195
bec61f067ea0 added AVL tree implementation - TODO: free memory + test cases
Mike Becker <universe@uap-core.de>
parents: 194
diff changeset
320 }
194
0c1b7676e74c finalized AVL tree interface + added implementation skeleton + fixed ucx_ptrcmp()
Mike Becker <universe@uap-core.de>
parents: 192
diff changeset
321 }
0c1b7676e74c finalized AVL tree interface + added implementation skeleton + fixed ucx_ptrcmp()
Mike Becker <universe@uap-core.de>
parents: 192
diff changeset
322
199
e25dc68336ec added ucx_avl_count
Mike Becker <universe@uap-core.de>
parents: 197
diff changeset
323 static size_t ucx_avl_countn(UcxAVLNode *node) {
e25dc68336ec added ucx_avl_count
Mike Becker <universe@uap-core.de>
parents: 197
diff changeset
324 if (node) {
e25dc68336ec added ucx_avl_count
Mike Becker <universe@uap-core.de>
parents: 197
diff changeset
325 return 1 + ucx_avl_countn(node->left) + ucx_avl_countn(node->right);
e25dc68336ec added ucx_avl_count
Mike Becker <universe@uap-core.de>
parents: 197
diff changeset
326 } else {
e25dc68336ec added ucx_avl_count
Mike Becker <universe@uap-core.de>
parents: 197
diff changeset
327 return 0;
e25dc68336ec added ucx_avl_count
Mike Becker <universe@uap-core.de>
parents: 197
diff changeset
328 }
e25dc68336ec added ucx_avl_count
Mike Becker <universe@uap-core.de>
parents: 197
diff changeset
329 }
e25dc68336ec added ucx_avl_count
Mike Becker <universe@uap-core.de>
parents: 197
diff changeset
330
e25dc68336ec added ucx_avl_count
Mike Becker <universe@uap-core.de>
parents: 197
diff changeset
331 size_t ucx_avl_count(UcxAVLTree *tree) {
e25dc68336ec added ucx_avl_count
Mike Becker <universe@uap-core.de>
parents: 197
diff changeset
332 return ucx_avl_countn(tree->root);
e25dc68336ec added ucx_avl_count
Mike Becker <universe@uap-core.de>
parents: 197
diff changeset
333 }
245
db732f8c083a adds AVL predecessor and successor functions
Mike Becker <universe@uap-core.de>
parents: 243
diff changeset
334
db732f8c083a adds AVL predecessor and successor functions
Mike Becker <universe@uap-core.de>
parents: 243
diff changeset
335 UcxAVLNode* ucx_avl_pred(UcxAVLNode* node) {
db732f8c083a adds AVL predecessor and successor functions
Mike Becker <universe@uap-core.de>
parents: 243
diff changeset
336 if (node->left) {
db732f8c083a adds AVL predecessor and successor functions
Mike Becker <universe@uap-core.de>
parents: 243
diff changeset
337 UcxAVLNode* n = node->left;
db732f8c083a adds AVL predecessor and successor functions
Mike Becker <universe@uap-core.de>
parents: 243
diff changeset
338 while (n->right) {
db732f8c083a adds AVL predecessor and successor functions
Mike Becker <universe@uap-core.de>
parents: 243
diff changeset
339 n = n->right;
db732f8c083a adds AVL predecessor and successor functions
Mike Becker <universe@uap-core.de>
parents: 243
diff changeset
340 }
db732f8c083a adds AVL predecessor and successor functions
Mike Becker <universe@uap-core.de>
parents: 243
diff changeset
341 return n;
db732f8c083a adds AVL predecessor and successor functions
Mike Becker <universe@uap-core.de>
parents: 243
diff changeset
342 } else {
db732f8c083a adds AVL predecessor and successor functions
Mike Becker <universe@uap-core.de>
parents: 243
diff changeset
343 UcxAVLNode* n = node;
db732f8c083a adds AVL predecessor and successor functions
Mike Becker <universe@uap-core.de>
parents: 243
diff changeset
344 while (n->parent) {
db732f8c083a adds AVL predecessor and successor functions
Mike Becker <universe@uap-core.de>
parents: 243
diff changeset
345 if (n->parent->right == n) {
db732f8c083a adds AVL predecessor and successor functions
Mike Becker <universe@uap-core.de>
parents: 243
diff changeset
346 return n->parent;
db732f8c083a adds AVL predecessor and successor functions
Mike Becker <universe@uap-core.de>
parents: 243
diff changeset
347 } else {
db732f8c083a adds AVL predecessor and successor functions
Mike Becker <universe@uap-core.de>
parents: 243
diff changeset
348 n = n->parent;
db732f8c083a adds AVL predecessor and successor functions
Mike Becker <universe@uap-core.de>
parents: 243
diff changeset
349 }
db732f8c083a adds AVL predecessor and successor functions
Mike Becker <universe@uap-core.de>
parents: 243
diff changeset
350 }
db732f8c083a adds AVL predecessor and successor functions
Mike Becker <universe@uap-core.de>
parents: 243
diff changeset
351 return NULL;
db732f8c083a adds AVL predecessor and successor functions
Mike Becker <universe@uap-core.de>
parents: 243
diff changeset
352 }
db732f8c083a adds AVL predecessor and successor functions
Mike Becker <universe@uap-core.de>
parents: 243
diff changeset
353 }
db732f8c083a adds AVL predecessor and successor functions
Mike Becker <universe@uap-core.de>
parents: 243
diff changeset
354
db732f8c083a adds AVL predecessor and successor functions
Mike Becker <universe@uap-core.de>
parents: 243
diff changeset
355 UcxAVLNode* ucx_avl_succ(UcxAVLNode* node) {
db732f8c083a adds AVL predecessor and successor functions
Mike Becker <universe@uap-core.de>
parents: 243
diff changeset
356 if (node->right) {
db732f8c083a adds AVL predecessor and successor functions
Mike Becker <universe@uap-core.de>
parents: 243
diff changeset
357 UcxAVLNode* n = node->right;
db732f8c083a adds AVL predecessor and successor functions
Mike Becker <universe@uap-core.de>
parents: 243
diff changeset
358 while (n->left) {
db732f8c083a adds AVL predecessor and successor functions
Mike Becker <universe@uap-core.de>
parents: 243
diff changeset
359 n = n->left;
db732f8c083a adds AVL predecessor and successor functions
Mike Becker <universe@uap-core.de>
parents: 243
diff changeset
360 }
db732f8c083a adds AVL predecessor and successor functions
Mike Becker <universe@uap-core.de>
parents: 243
diff changeset
361 return n;
db732f8c083a adds AVL predecessor and successor functions
Mike Becker <universe@uap-core.de>
parents: 243
diff changeset
362 } else {
db732f8c083a adds AVL predecessor and successor functions
Mike Becker <universe@uap-core.de>
parents: 243
diff changeset
363 UcxAVLNode* n = node;
db732f8c083a adds AVL predecessor and successor functions
Mike Becker <universe@uap-core.de>
parents: 243
diff changeset
364 while (n->parent) {
db732f8c083a adds AVL predecessor and successor functions
Mike Becker <universe@uap-core.de>
parents: 243
diff changeset
365 if (n->parent->left == n) {
db732f8c083a adds AVL predecessor and successor functions
Mike Becker <universe@uap-core.de>
parents: 243
diff changeset
366 return n->parent;
db732f8c083a adds AVL predecessor and successor functions
Mike Becker <universe@uap-core.de>
parents: 243
diff changeset
367 } else {
db732f8c083a adds AVL predecessor and successor functions
Mike Becker <universe@uap-core.de>
parents: 243
diff changeset
368 n = n->parent;
db732f8c083a adds AVL predecessor and successor functions
Mike Becker <universe@uap-core.de>
parents: 243
diff changeset
369 }
db732f8c083a adds AVL predecessor and successor functions
Mike Becker <universe@uap-core.de>
parents: 243
diff changeset
370 }
db732f8c083a adds AVL predecessor and successor functions
Mike Becker <universe@uap-core.de>
parents: 243
diff changeset
371 return NULL;
db732f8c083a adds AVL predecessor and successor functions
Mike Becker <universe@uap-core.de>
parents: 243
diff changeset
372 }
db732f8c083a adds AVL predecessor and successor functions
Mike Becker <universe@uap-core.de>
parents: 243
diff changeset
373 }

mercurial