Skip to content

AVL tree

include/structures/avl.h View source View on GitHub
struct avl_tree_node {
    int                   height;
    struct avl_tree_node  *left;
    struct avl_tree_node  *right;
    struct avl_tree_node  *parent;
};
struct avl_tree_node_ops {
    int  (*cmp)(const struct avl_tree_node *a, const struct avl_tree_node *b);
    int  (*cmp_key)(const struct avl_tree_node *node, const void *key);
};
struct avl_tree {
    struct avl_tree_node            *root;
    const struct avl_tree_node_ops  *ops;
};
void avl_tree_init(struct avl_tree *tree, const struct avl_tree_node_ops *ops);
void avl_tree_insert(struct avl_tree *tree, struct avl_tree_node *node);
void avl_tree_remove(struct avl_tree *tree, struct avl_tree_node *node);
struct avl_tree_node * avl_tree_find(const struct avl_tree *tree, const void *key);
struct avl_tree_node * avl_tree_first(const struct avl_tree *tree);
struct avl_tree_node * avl_tree_last(const struct avl_tree *tree);
struct avl_tree_node * avl_tree_next(const struct avl_tree_node *node);
struct avl_tree_node * avl_tree_prev(const struct avl_tree_node *node);
int avl_tree_empty(const struct avl_tree *tree);
#define avl_entry(ptr, type, member) container_of(ptr, type, member)
#define avl_tree_for_each(pos, tree) \
    for ((pos) = avl_tree_first(tree); (pos); (pos) = avl_tree_next(pos))
#define avl_tree_for_each_safe(pos, n, tree) \
    for ((pos) = avl_tree_first(tree), \
        (n) = (pos) ? avl_tree_next(pos) : NULL; \
         (pos); (pos) = (n), (n) = (pos) ? avl_tree_next(pos) : NULL)