引言
rbtree是linux内核红黑树实现。
1 使用介绍
1.1 定义结构
一般在struct中嵌套struct rb_node作为树节点:
struct xxx_st {
# 略略略... ...
struct rb_node node;
# 略略略... ...
};
1.2 初始化根
将struct rb_root赋值为RB_ROOT:
struct rb_root xxx_root = RB_ROOT;
1.3 插入节点
先通过rb_link_node插入节点,再通过rb_insert_color重新平衡红黑树:
// 将节点插入到parent的左节点。注:插入到xxx_root,则第三个参数为NULL。
rb_link_node(&xxx_st_ptr->node, parent, &parent->rb_left);
// 重新平衡红黑树
rb_insert_color(&xxx_st_ptr->node, &xxx_root);
1.4 移除节点
通过rb_erase移除节点:
rb_erase(&xxx_st_ptr->node, &xxx_root);
注:该API会重新平衡红黑树
1.5 节点转换
通过rb_entry可将节点转为具体类型:
struct xxx_st_ptr *xxx_st_ptr = rb_entry(parent, struct xxx_st, node);