Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(194)

Side by Side Diff: fusl/src/search/tsearch_avl.c

Issue 1573973002: Add a "fork" of musl as //fusl. (Closed) Base URL: https://github.com/domokit/mojo.git@master
Patch Set: Created 4 years, 11 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
« no previous file with comments | « fusl/src/search/tdestroy.c ('k') | fusl/src/select/poll.c » ('j') | no next file with comments »
Toggle Intra-line Diffs ('i') | Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
OLDNEW
(Empty)
1 #include <stdlib.h>
2 #include <search.h>
3
4 /*
5 avl tree implementation using recursive functions
6 the height of an n node tree is less than 1.44*log2(n+2)-1
7 (so the max recursion depth in case of a tree with 2^32 nodes is 45)
8 */
9
10 struct node {
11 const void *key;
12 struct node *left;
13 struct node *right;
14 int height;
15 };
16
17 static int delta(struct node *n) {
18 return (n->left ? n->left->height:0) - (n->right ? n->right->height:0);
19 }
20
21 static void updateheight(struct node *n) {
22 n->height = 0;
23 if (n->left && n->left->height > n->height)
24 n->height = n->left->height;
25 if (n->right && n->right->height > n->height)
26 n->height = n->right->height;
27 n->height++;
28 }
29
30 static struct node *rotl(struct node *n) {
31 struct node *r = n->right;
32 n->right = r->left;
33 r->left = n;
34 updateheight(n);
35 updateheight(r);
36 return r;
37 }
38
39 static struct node *rotr(struct node *n) {
40 struct node *l = n->left;
41 n->left = l->right;
42 l->right = n;
43 updateheight(n);
44 updateheight(l);
45 return l;
46 }
47
48 static struct node *balance(struct node *n) {
49 int d = delta(n);
50
51 if (d < -1) {
52 if (delta(n->right) > 0)
53 n->right = rotr(n->right);
54 return rotl(n);
55 } else if (d > 1) {
56 if (delta(n->left) < 0)
57 n->left = rotl(n->left);
58 return rotr(n);
59 }
60 updateheight(n);
61 return n;
62 }
63
64 static struct node *find(struct node *n, const void *k,
65 int (*cmp)(const void *, const void *))
66 {
67 int c;
68
69 if (!n)
70 return 0;
71 c = cmp(k, n->key);
72 if (c == 0)
73 return n;
74 if (c < 0)
75 return find(n->left, k, cmp);
76 else
77 return find(n->right, k, cmp);
78 }
79
80 static struct node *insert(struct node *n, const void *k,
81 int (*cmp)(const void *, const void *), struct node **found)
82 {
83 struct node *r;
84 int c;
85
86 if (!n) {
87 n = malloc(sizeof *n);
88 if (n) {
89 n->key = k;
90 n->left = n->right = 0;
91 n->height = 1;
92 }
93 *found = n;
94 return n;
95 }
96 c = cmp(k, n->key);
97 if (c == 0) {
98 *found = n;
99 return 0;
100 }
101 r = insert(c < 0 ? n->left : n->right, k, cmp, found);
102 if (r) {
103 if (c < 0)
104 n->left = r;
105 else
106 n->right = r;
107 r = balance(n);
108 }
109 return r;
110 }
111
112 static struct node *remove_rightmost(struct node *n, struct node **rightmost)
113 {
114 if (!n->right) {
115 *rightmost = n;
116 return n->left;
117 }
118 n->right = remove_rightmost(n->right, rightmost);
119 return balance(n);
120 }
121
122 static struct node *remove(struct node **n, const void *k,
123 int (*cmp)(const void *, const void *), struct node *parent)
124 {
125 int c;
126
127 if (!*n)
128 return 0;
129 c = cmp(k, (*n)->key);
130 if (c == 0) {
131 struct node *r = *n;
132 if (r->left) {
133 r->left = remove_rightmost(r->left, n);
134 (*n)->left = r->left;
135 (*n)->right = r->right;
136 *n = balance(*n);
137 } else
138 *n = r->right;
139 free(r);
140 return parent;
141 }
142 if (c < 0)
143 parent = remove(&(*n)->left, k, cmp, *n);
144 else
145 parent = remove(&(*n)->right, k, cmp, *n);
146 if (parent)
147 *n = balance(*n);
148 return parent;
149 }
150
151 void *tdelete(const void *restrict key, void **restrict rootp,
152 int(*compar)(const void *, const void *))
153 {
154 if (!rootp)
155 return 0;
156 struct node *n = *rootp;
157 struct node *ret;
158 /* last argument is arbitrary non-null pointer
159 which is returned when the root node is deleted */
160 ret = remove(&n, key, compar, n);
161 *rootp = n;
162 return ret;
163 }
164
165 void *tfind(const void *key, void *const *rootp,
166 int(*compar)(const void *, const void *))
167 {
168 if (!rootp)
169 return 0;
170 return find(*rootp, key, compar);
171 }
172
173 void *tsearch(const void *key, void **rootp,
174 int (*compar)(const void *, const void *))
175 {
176 struct node *update;
177 struct node *ret;
178 if (!rootp)
179 return 0;
180 update = insert(*rootp, key, compar, &ret);
181 if (update)
182 *rootp = update;
183 return ret;
184 }
185
186 static void walk(const struct node *r, void (*action)(const void *, VISIT, int), int d)
187 {
188 if (r == 0)
189 return;
190 if (r->left == 0 && r->right == 0)
191 action(r, leaf, d);
192 else {
193 action(r, preorder, d);
194 walk(r->left, action, d+1);
195 action(r, postorder, d);
196 walk(r->right, action, d+1);
197 action(r, endorder, d);
198 }
199 }
200
201 void twalk(const void *root, void (*action)(const void *, VISIT, int))
202 {
203 walk(root, action, 0);
204 }
OLDNEW
« no previous file with comments | « fusl/src/search/tdestroy.c ('k') | fusl/src/select/poll.c » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698