详解Linux内核红黑树算法的实现 http://blog.csdn.net/npy_lp/article/details/7420689
详解Linux内核红黑树算法的实现
版权声明:本文为博主原创文章,未经博主允许不得转载。
开发平台:Ubuntu11.04
内核源码:linux-2.6.38.8.tar.bz2
关于二叉查找树的概念请参考博文《详解二叉查找树算法的实现》。
平衡二叉树(BalancedBinary Tree或Height-Balanced Tree)又称AVL树。它或者是一棵空树,或者是具有下列性质的二叉树:它的左子树和右子树都是平衡二叉树,且左子树和右子树的深度之差的绝对值不超过1。若将二叉树上结点的平衡因子BF(BalanceFactor)定义为该结点的左子树的深度减去它的右子树的深度,则平衡二叉树上所有结点的平衡因子只可能是-1、0和1。(此段定义来自严蔚敏的《数据结构(C语言版)》)
红黑树是一种在插入或删除结点时都需要维持平衡的二叉查找树,并且每个结点都具有颜色属性:
(1)、一个结点要么是红色的,要么是黑色的。
(2)、根结点是黑色的。
(3)、如果一个结点是红色的,那么它的子结点必须是黑色的,也就是说在沿着从根结点出发的任何路径上都不会出现两个连续的红色结点。
(4)、从一个结点到一个NULL指针的每条路径上必须包含相同数目的黑色结点。
(此图片来自维基百科)
Linux内核红黑树的算法都定义在linux-2.6.38.8/include/linux/rbtree.h和linux-2.6.38.8/lib/rbtree.c两个文件中。
1、结构体
- struct rb_node
- {
- unsigned long rb_parent_color;
- #define RB_RED 0
- #define RB_BLACK 1
- struct rb_node *rb_right;
- struct rb_node *rb_left;
- } __attribute__((aligned(sizeof(long))));
这里的巧妙之处是使用成员rb_parent_color同时存储两种数据,一是其双亲结点的地址,另一是此结点的着色。__attribute__((aligned(sizeof(long))))属性保证了红黑树中的每个结点的首地址都是32位对齐的(在32位机上),也就是说每个结点首地址的bit[1]和bit[0]都是0,因此就可以使用bit[0]来存储结点的颜色属性而不干扰到其双亲结点首地址的存储。
操作rb_parent_color的函数:
- #define rb_parent(r) ((struct rb_node *)((r)->rb_parent_color & ~3)) //获得其双亲结点的首地址
- #define rb_color(r) ((r)->rb_parent_color & 1) //获得颜色属性
- #define rb_is_red(r) (!rb_color(r)) //判断颜色属性是否为红
- #define rb_is_black(r) rb_color(r) //判断颜色属性是否为黑
- #define rb_set_red(r) do { (r)->rb_parent_color &= ~1; } while (0) //设置红色属性
- #define rb_set_black(r) do { (r)->rb_parent_color |= 1; } while (0) //设置黑色属性
- static inline void rb_set_parent(struct rb_node *rb, struct rb_node *p) //设置其双亲结点首地址的函数
- {
- rb->rb_parent_color = (rb->rb_parent_color & 3) | (unsigned long)p;
- }
- static inline void rb_set_color(struct rb_node *rb, int color) //设置结点颜色属性的函数
- {
- rb->rb_parent_color = (rb->rb_parent_color & ~1) | color;
- }
初始化新结点:
- static inline void rb_link_node(struct rb_node * node, struct rb_node * parent,
- struct rb_node ** rb_link)
- {
- node->rb_parent_color = (unsigned long )parent; //设置其双亲结点的首地址(根结点的双亲结点为NULL),且颜色属性设为黑色
- node->rb_left = node->rb_right = NULL; //初始化新结点的左右子树
- *rb_link = node; //指向新结点
- }
指向红黑树根结点的指针:
- struct rb_root
- {
- struct rb_node *rb_node;
- };
- #define RB_ROOT (struct rb_root) { NULL, } //初始化指向红黑树根结点的指针
- #define rb_entry(ptr, type, member) container_of(ptr, type, member) //用来获得包含struct rb_node的结构体的首地址
- #define RB_EMPTY_ROOT(root) ((root)->rb_node == NULL) //判断树是否为空
- #define RB_EMPTY_NODE(node) (rb_parent(node) == node) //判断node的双亲结点是否为自身
- #define RB_CLEAR_NODE(node) (rb_set_parent(node, node)) //设置双亲结点为自身
2、插入
首先像二叉查找树一样插入一个新结点,然后根据情况作出相应的调整,以使其满足红黑树的颜色属性(其实质是维持红黑树的平衡)。
函数rb_insert_color使用while循环不断地判断双亲结点是否存在,且颜色属性为红色。
若判断条件为真,则分成两部分执行后续的操作:
(1)、当双亲结点是祖父结点左子树的根时,则:
a、存在叔父结点,且颜色属性为红色。
b、当node是其双亲结点右子树的根时,则左旋,然后执行第c步。
c、当node是其双亲结点左子树的根时。
(2)、当双亲结点是祖父结点右子树的根时的操作与第(1)步大致相同,这里略过不谈。
若为假,则始终设置根结点的颜色属性为黑色。
- void rb_insert_color(struct rb_node *node, struct rb_root *root)
- {
- struct rb_node *parent, *gparent;
- while ((parent = rb_parent(node)) && rb_is_red(parent)) //双亲结点不为NULL,且颜色属性为红色
- {
- gparent = rb_parent(parent); //获得祖父结点
- if (parent == gparent->rb_left) //双亲结点是祖父结点左子树的根
- {
- {
- register struct rb_node *uncle = gparent->rb_right; //获得叔父结点
- if (uncle && rb_is_red(uncle)) //叔父结点存在,且颜色属性为红色
- {
- rb_set_black(uncle); //设置叔父结点为黑色
- rb_set_black(parent); //设置双亲结点为黑色
- rb_set_red(gparent); //设置祖父结点为红色
- node = gparent; //node指向祖父结点
- continue; //继续下一个while循环
- }
- }
- if (parent->rb_right == node) //当node是其双亲结点右子树的根时
- {
- register struct rb_node *tmp;
- __rb_rotate_left(parent, root); //左旋
- tmp = parent; //调整parent和node指针的指向
- parent = node;
- node = tmp;
- }
- rb_set_black(parent); //设置双亲结点为黑色
- rb_set_red(gparent); //设置祖父结点为红色
- __rb_rotate_right(gparent, root); //右旋
- } else { // !(parent == gparent->rb_left)
- {
- register struct rb_node *uncle = gparent->rb_left;
- if (uncle && rb_is_red(uncle))
- {
- rb_set_black(uncle);
- rb_set_black(parent);
- rb_set_red(gparent);
- node = gparent;
- continue;
- }
- }
- if (parent->rb_left == node)
- {
- register struct rb_node *tmp;
- __rb_rotate_right(parent, root);
- tmp = parent;
- parent = node;
- node = tmp;
- }
- rb_set_black(parent);
- rb_set_red(gparent);
- __rb_rotate_left(gparent, root);
- } //end if (parent == gparent->rb_left)
- } //end while ((parent = rb_parent(node)) && rb_is_red(parent))
- rb_set_black(root->rb_node);
- }
3、删除
像二叉查找树的删除操作一样,首先需要找到所需删除的结点,然后根据该结点左右子树的有无分为三种情形:
若node结点的颜色属性为黑色,则需要调用__rb_erase_color函数来进行调整。
- void rb_erase(struct rb_node *node, struct rb_root *root)
- {
- struct rb_node *child, *parent;
- int color;
- if (!node->rb_left) //删除结点无左子树
- child = node->rb_right;
- else if (!node->rb_right) //删除结点无右子树
- child = node->rb_left;
- else //左右子树都有
- {
- struct rb_node *old = node, *left;
- node = node->rb_right;
- while ((left = node->rb_left) != NULL)
- node = left;
- if (rb_parent(old)) {
- if (rb_parent(old)->rb_left == old)
- rb_parent(old)->rb_left = node;
- else
- rb_parent(old)->rb_right = node;
- } else
- root->rb_node = node;
- child = node->rb_right;
- parent = rb_parent(node);
- color = rb_color(node);
- if (parent == old) {
- parent = node;
- } else {
- if (child)
- rb_set_parent(child, parent);
- parent->rb_left = child;
- node->rb_right = old->rb_right;
- rb_set_parent(old->rb_right, node);
- }
- node->rb_parent_color = old->rb_parent_color;
- node->rb_left = old->rb_left;
- rb_set_parent(old->rb_left, node);
- goto color;
- } //end else
- parent = rb_parent(node); //获得删除结点的双亲结点
- color = rb_color(node); //获取删除结点的颜色属性
- if (child)
- rb_set_parent(child, parent);
- if (parent)
- {
- if (parent->rb_left == node)
- parent->rb_left = child;
- else
- parent->rb_right = child;
- }
- else
- root->rb_node = child;
- color:
- if (color == RB_BLACK) //如果删除结点的颜色属性为黑色,则需调用__rb_erase_color函数来进行调整
- __rb_erase_color(child, parent, root);
- }
4、遍历
rb_first和rb_next函数可组成中序遍历,即以升序遍历红黑树中的所有结点。
- struct rb_node *rb_first(const struct rb_root *root)
- {
- struct rb_node *n;
- n = root->rb_node;
- if (!n)
- return NULL;
- while (n->rb_left)
- n = n->rb_left;
- return n;
- }
- struct rb_node *rb_next(const struct rb_node *node)
- {
- struct rb_node *parent;
- if (rb_parent(node) == node)
- return NULL;
- /* If we have a right-hand child, go down and then left as far
- as we can. */
- if (node->rb_right) {
- node = node->rb_right;
- while (node->rb_left)
- node=node->rb_left;
- return (struct rb_node *)node;
- }
- /* No right-hand children. Everything down and left is
- smaller than us, so any 'next' node must be in the general
- direction of our parent. Go up the tree; any time the
- ancestor is a right-hand child of its parent, keep going
- up. First time it's a left-hand child of its parent, said
- parent is our 'next' node. */
- while ((parent = rb_parent(node)) && node == parent->rb_right)
- node = parent;
- return parent;
- }
5、在应用程序中使用
Linux内核中红黑树算法的实现非常通用、巧妙,而且免费又开源,因此完全可以把它运用到自己的应用程序中。
(1)、从内核中拷贝源文件:
- $ mkdir redblack
- $ cd redblack/
- $ cp ../linux-2.6.38.8/lib/rbtree.c .
- $ cp ../linux-2.6.38.8/include/linux/rbtree.h .
(2)、修改源文件:
a、C文件rbtree.c
修改包含头文件的代码
- //删除以下两行代码
- #include <linux/rbtree.h>
- #include <linux/module.h>
- //新增以下代码,即包含当前目录中的头文件rbtree.h
- #include "rbtree.h"
删除所有的EXPORT_SYMBOL宏
- EXPORT_SYMBOL(rb_insert_color);
- EXPORT_SYMBOL(rb_erase);
- EXPORT_SYMBOL(rb_augment_insert);
- EXPORT_SYMBOL(rb_augment_erase_begin);
- EXPORT_SYMBOL(rb_augment_erase_end);
- EXPORT_SYMBOL(rb_first);
- EXPORT_SYMBOL(rb_last);
- EXPORT_SYMBOL(rb_next);
- EXPORT_SYMBOL(rb_prev);
- EXPORT_SYMBOL(rb_replace_node);
b、头文件rbtree.h
删除包含头文件的代码,并添加三个宏定义
- //删除以下两行代码
- #include <linux/kernel.h>
- #include <linux/stddef.h>
- /* linux-2.6.38.8/include/linux/stddef.h */
- #undef NULL
- #if defined(__cplusplus)
- #define NULL 0
- #else
- #define NULL ((void *)0)
- #endif
- /* linux-2.6.38.8/include/linux/stddef.h */
- #define offsetof(TYPE, MEMBER) ((size_t) &((TYPE *)0)->MEMBER)
- /* linux-2.6.38.8/include/linux/kernel.h */
- #define container_of(ptr, type, member) ({ \
- const typeof( ((type *)0)->member ) *__mptr = (ptr); \
- (type *)( (char *)__mptr - offsetof(type,member) );})
(3)、示例代码
Linux内核红黑树的使用方法请参考linux-2.6.38.8/Documentation/rbtree.txt文件。
- /* test.c */
- #include <stdio.h>
- #include <stdlib.h>
- #include "rbtree.h"
- struct mytype {
- struct rb_node my_node;
- int num;
- };
- struct mytype *my_search(struct rb_root *root, int num)
- {
- struct rb_node *node = root->rb_node;
- while (node) {
- struct mytype *data = container_of(node, struct mytype, my_node);
- if (num < data->num)
- node = node->rb_left;
- else if (num > data->num)
- node = node->rb_right;
- else
- return data;
- }
- return NULL;
- }
- int my_insert(struct rb_root *root, struct mytype *data)
- {
- struct rb_node **tmp = &(root->rb_node), *parent = NULL;
- /* Figure out where to put new node */
- while (*tmp) {
- struct mytype *this = container_of(*tmp, struct mytype, my_node);
- parent = *tmp;
- if (data->num < this->num)
- tmp = &((*tmp)->rb_left);
- else if (data->num > this->num)
- tmp = &((*tmp)->rb_right);
- else
- return -1;
- }
- /* Add new node and rebalance tree. */
- rb_link_node(&data->my_node, parent, tmp);
- rb_insert_color(&data->my_node, root);
- return 0;
- }
- void my_delete(struct rb_root *root, int num)
- {
- struct mytype *data = my_search(root, num);
- if (!data) {
- fprintf(stderr, "Not found %d.\n", num);
- return;
- }
- rb_erase(&data->my_node, root);
- free(data);
- }
- void print_rbtree(struct rb_root *tree)
- {
- struct rb_node *node;
- for (node = rb_first(tree); node; node = rb_next(node))
- printf("%d ", rb_entry(node, struct mytype, my_node)->num);
- printf("\n");
- }
- int main(int argc, char *argv[])
- {
- struct rb_root mytree = RB_ROOT;
- int i, ret, num;
- struct mytype *tmp;
- if (argc < 2) {
- fprintf(stderr, "Usage: %s num\n", argv[0]);
- exit(-1);
- }
- num = atoi(argv[1]);
- printf("Please enter %d integers:\n", num);
- for (i = 0; i < num; i++) {
- tmp = malloc(sizeof(struct mytype));
- if (!tmp)
- perror("Allocate dynamic memory");
- scanf("%d", &tmp->num);
- ret = my_insert(&mytree, tmp);
- if (ret < 0) {
- fprintf(stderr, "The %d already exists.\n", tmp->num);
- free(tmp);
- }
- }
- printf("\nthe first test\n");
- print_rbtree(&mytree);
- my_delete(&mytree, 21);
- printf("\nthe second test\n");
- print_rbtree(&mytree);
- return 0;
- }
编译并执行:
- $ gcc rbtree.c test.c -o test
- richard@tanglinux:~/algorithm/redblack$ ./test 10
- Please enter 10 integers:
- 23
- 4
- 56
- 32
- 89
- 122
- 12
- 21
- 45
- 23
- The 23 already exists.
- the first test
- 4 12 21 23 32 45 56 89 122
- the second test
- 4 12 23 32 45 56 89 122
详解Linux内核红黑树算法的实现 http://blog.csdn.net/npy_lp/article/details/7420689相关推荐
- 详解Linux内核红黑树算法的实现
复制太难看了,以后试试 博文地址:http://blog.csdn.net/npy_lp/article/details/7420689
- 从最大似然到EM算法浅解 http://blog.csdn.net/zouxy09/article/details/8537620
1. EM blog的举例就是group 然后就是每个group的function很有效地串联所学的知识,看到的论文,所有的思考,都是有一定的逻辑关系,如何逐渐develop你的想法,都是有一定的源头 ...
- Mongodb参数详解(参考:http://blog.csdn.net/freebird_lb/article/details/8229567)
MongoDB配置参数详解: #对mongo实例来说,每个host允许链接的最大链接数,这些链接空闲时会放入池中,如果链接被耗尽,任何请求链接的操作会被阻塞等待链接可用,推荐配置10 connecti ...
- Linux 内核红黑树分析
Android binder 内核实现是用红黑树的,理解红黑树我觉得是每一个Linux er的重中之重,感谢格子森同学的投稿,周末愉快. 内核版本为 linux4.2.1 本文主要从红黑树的代码实现入 ...
- 基于Linux内核红黑树的TR069参数解析工具:树形结构+CPE RPC支持
目录 tr069_oid.h tr069_oid.c demo.c tr069_oid.h /* Dependencies -------------------------------------- ...
- 装配 Spring Bean 详解https://blog.csdn.net/kkfd1002/article/details/79936447
装配 Bean 的概述 前面已经介绍了 Spring IoC 的理念和设计,这一篇文章将介绍的是如何将自己开发的 Bean 装配到 Spring IoC 容器中. 大部分场景下,我们都会使用 Appl ...
- .net反射详解 原文://http://blog.csdn.net/wenyan07/article/details/27882363
概述反射 通过反射可以提供类型信息,从而使得我们开发人员在运行时能够利用这些信息构造和使用对象. 反射机制允许程序在执行过程中动态地添加各种功能. 运行时类型标识 运行时类型标识(RTTI),可以在程 ...
- Mysql分析-profile详解(转http://blog.csdn.net/ty_hf/article/details/54895026)
一.前言 当我们要对某一条sql的性能进行分析时,可以使用它. Profiling是从 mysql5.0.3版本以后才开放的. 启动profile之后,所有 查询包括错误的语句都会记录在内. 关闭会话 ...
- MMX指令集(详解)https://blog.csdn.net/dahan_wangtao/article/details/1944153
EMMS MMX状态置空: 将FP特征字置空(全1),使后续浮点指令可以使用浮点寄存器,其他MMX指令自动置FP为全0.本指令应在所有MMX例程结束和调用可含有FP指令的例程时使用,以清除MMX状态. ...
- linux下各种格式软件的安装(引用http://blog.csdn.net/zyz511919766/article/details/7574040)
首先介绍两个简单的方式 第一:sudo apt-get install packagename 命令 如果我们知道我们要安装的软件的确切的名称,那么我们可以简单的通过此条命令来获取和安装软件.apt- ...
最新文章
- ADO.NET并发性
- Win7下U盘安装Ubuntu14.04双系统
- 自定义工作流任务控件
- 【Vue.js】vue用户登录功能
- web.xml隐藏html,web.xml
- Vue + Bootstrap|Element UI——模态框被遮罩层遮盖问题解决方案
- Qt Creator分析QML应用程序
- 移动研发 DevOps 落地实践
- Map Reduce和流处理
- 最流行的 IDE 之争:Eclipse 反超 Visual Studio 成第一
- kubernets 集群和本地环境联调环境打通工具kt-connect
- 数学物理方法pdf_3位物理学家意外发现基础数学新方法,数学天才陶哲轩:我开始压根不相信...
- python实现网页微信登陆_(转帖)网站微信登录-python 实现
- 定义客户类(Customer): 1,客户类的属性包括:姓名、年龄、电话、金钱数量、账号、密码; 2,方法包括:购买商品、付款、显示自己的信息。 3,创建测试类,在main方法中使用客户类创建两个客户
- 终于!这个速算题自动批改程序被我给做出来了!
- 前端开发入门 --摘自慕克网大漠穷秋
- 戴尔易安信引领科技创新,以全面的端到端解决方案助力企业“数”造未来
- easyUI日期框的日期范围限制
- ui设计是干啥的: ui设计师主要是做什么的呢
- 图像识别(二)| 图像的色彩空间
热门文章
- ios客户端发现_华为爱奇艺手机活动开发总结
- 数据大屏之跑马灯功能
- 云课堂 php代码,基于ThinkPHP二开高仿网易云课堂整站PHP源代码
- 激光粒度仪测试原理及详情解答【注解】
- IDCC2018|上海数据港股份有限公司副总裁、数据中心首席架构师王海峰:标杆管理驱动数据中心建设变革...
- Labwindows/CVI 编写CAN通讯的上位机
- 多多自走棋服务器不稳定,《多多自走棋》将登陆Epic平台 PC版数据将与移动版互通...
- 人工智能粒子群优化和群智能
- maven运行Error:(3, 14) java: 程序包不存在
- 光速不变原理引发的诡异现象与时空分量的解释