blob: adf2b8b462b3a02fe463eb7f81223cb610d26b2f (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
|
/*
* rbtree.c
*
* Simple implementation of a left-leaning red-black tree with 64-bit
* integer keys. The search operation will return the highest node <=
* the key; only search, insert, and full-tree deletion is supported,
* but additional standard llrbtree operations can be coded up at will.
*
* See http://www.cs.princeton.edu/~rs/talks/LLRB/RedBlack.pdf for
* information about left-leaning red-black trees.
*/
#include "rbtree.h"
#include "nasmlib.h"
const struct rbtree *rb_search(const struct rbtree *tree, uint64_t key)
{
const struct rbtree *best = NULL;
while (tree) {
if (tree->key == key)
return tree;
else if (tree->key > key)
tree = tree->left;
else {
best = tree;
tree = tree->right;
}
}
return best;
}
static bool is_red(struct rbtree *h)
{
return h && h->red;
}
static struct rbtree *rotate_left(struct rbtree *h)
{
struct rbtree *x = h->right;
h->right = x->left;
x->left = h;
x->red = x->left->red;
x->left->red = true;
return x;
}
static struct rbtree *rotate_right(struct rbtree *h)
{
struct rbtree *x = h->left;
h->left = x->right;
x->right = h;
x->red = x->right->red;
x->right->red = true;
return x;
}
static void color_flip(struct rbtree *h)
{
h->red = !h->red;
h->left->red = !h->left->red;
h->right->red = !h->right->red;
}
struct rbtree *rb_insert(struct rbtree *tree, uint64_t key, void *data)
{
if (!tree) {
struct rbtree *node = nasm_malloc(sizeof *node);
node->key = key;
node->data = data;
node->red = true;
return node;
}
if (is_red(tree->left) && is_red(tree->right))
color_flip(tree);
if (key < tree->key)
tree->left = rb_insert(tree->left, key, data);
else
tree->right = rb_insert(tree->right, key, data);
if (is_red(tree->right))
tree = rotate_left(tree);
if (is_red(tree->left) && is_red(tree->left->left))
tree = rotate_right(tree);
return tree;
}
void rb_free(struct rbtree *tree)
{
if (tree) {
rb_free(tree->left);
rb_free(tree->right);
nasm_free(tree);
}
}
|
