Linux数据结构之rbtree

引言

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);