devbox
凡人多烦事01
这个屌丝很懒，什么也没留下！
关注作者
热门标签
热门文章
当前位置:   article > 正文

Linux内核的红黑树源码实现以及调用_linux内核红黑树源码

作者：凡人多烦事01 | 2024-03-16 01:21:07

赞

踩

linux内核红黑树源码

红黑树可以说是程序员经常遇到的一种数据结构，不管是工作还是面试都会涉及，有时候还会让你写一段红黑树代码。

本文主要是讲Linux中的红黑树，关于红黑树定义参考wiki:https://en.wikipedia.org/wiki/Red%E2%80%93black_tree

其中《算法导论》中的定义最为清晰：

红黑树图示：

 

Linux中的红黑树（rbtree）

 

下面是我翻译的参考Linux内核文档，https://www.kernel.org/doc/Documentation/rbtree.txt 

英文作者：Rob Landley <rob@landley.net>


什么是红黑树，它适用于什么？
------------------------------------------------

红黑树是一种自平衡二叉搜索树，用于存储可排序的键/值数据对。

它不同于基数树（用来有效地存储稀疏数组，因此使用长整数索来插入/存取/删除节点）
和哈希表（它不用进行排序就可以很容易地按序遍历，但必须设定具体大小和散列函数，而红黑树优雅的扩展存储任意键）

红黑树类似于AVL树，但提供更快的实时有界插入和删除，最坏的情况表现是三次旋转（插入仅需2次），虽然查找略慢略慢但是时间仍然在O(log n)以内。

引用Linux Weekly News：

    内核中很多地方使用红黑树：
    deadline和CFQ I/O调度程序使用rbtree跟踪请求; CD/DVD块数据驱动器也是如此。
    高分辨率计时器代码使用rbtree来组织计时器请求。
    Ext3文件系统使用rbtree跟踪目录项。
    虚拟存储区（VMA）的跟踪，以及epoll的文件描述符，加密密钥，“分层令牌桶”调度网络数据包也有rbtree的使用。

本文档介绍了Linux rbtree实现的使用。
更多有关红黑树性质和使用的信息，
请参阅：关于红黑树的Linux Weekly News文章
http://lwn.net/Articles/184495/

维基百科在红黑树上的条目
http://en.wikipedia.org/wiki/Red-black_tree

Linux实现红黑树
---------------------------------------

Linux的rbtree实现

存在于文件“lib/rbtree.c”中，要使用头文件“#include <linux/rbtree.h>”。
linux\lib\rbtree.c
linux\include\linux\rbtree.h

文末有rbtree.c和rbtree.h的源码

Linux rbtree在实现上针对速度进行了优化，用户需要编写自己的树搜索和插入函数调用内核提供的rbtree。

 

创建一个新的rbtree
---------------------

rbtree树中的数据节点是包含struct rb_node成员的结构:

  1.  struct mytype {
  2.       struct rb_node node;
  3.       char * keystring;
  4.   };

当处理嵌入式struct rb_node的指针时，包含数据可以使用标准的container_of()宏访问结构。
此外，可以通过rb_entry（节点，类型，成员）直接访问各个成员。

每个rbtree的根是一个rb_root结构，它被初始化为：struct rb_root mytree = RB_ROOT;

 

在rbtree中搜索值
----------------------------------

为您的树编写搜索功能非常简单：从root开始比较每个值，并根据需要进入左或右分支。

例如：

  1. struct mytype *my_search(struct rb_root *root, char *string)
  2.   {
  3.   	struct rb_node *node = root->rb_node;
  4.  
  5.   	while (node) {
  6.   		struct mytype *data = container_of(node, struct mytype, node);
  7. 		int result;
  8.  
  9. 		result = strcmp(string, data->keystring);
  10.  
  11. 		if (result < 0)
  12.   			node = node->rb_left;
  13. 		else if (result > 0)
  14.   			node = node->rb_right;
  15. 		else
  16.   			return data;
  17. 	}
  18. 	return NULL;
  19.   }

将数据插入rbtree
-----------------------------

在树中插入数据涉及首先搜索要插入的位置新节点，然后插入节点并重新平衡（“重新着色”）树。

插入的搜索与先前的搜索不同，找到了指针的位置，用于移植新节点。
新节点也需要指向其父节点的链接以进行重新平衡。

例如：

  1. int my_insert(struct rb_root *root, struct mytype *data)
  2.   {
  3.   	struct rb_node **new = &(root->rb_node), *parent = NULL;
  4.  
  5.   	/* Figure out where to put new node */
  6.   	while (*new) {
  7.   		struct mytype *this = container_of(*new, struct mytype, node);
  8.   		int result = strcmp(data->keystring, this->keystring);
  9.  
  10. 		parent = *new;
  11.   		if (result < 0)
  12.   			new = &((*new)->rb_left);
  13.   		else if (result > 0)
  14.   			new = &((*new)->rb_right);
  15.   		else
  16.   			return FALSE;
  17.   	}
  18.  
  19.   	/* Add new node and rebalance tree. */
  20.   	rb_link_node(&data->node, parent, new);
  21.   	rb_insert_color(&data->node, root);
  22.  
  23. 	return TRUE;
  24.   }

 

删除或替换rbtree中的现有数据
------------------------------------------------

要从树中删除现有节点，请调用:

void rb_erase(struct rb_node * victim,struct rb_root * tree);

例如：

  1. struct mytype *data = mysearch(&mytree, "walrus");
  2.  
  3.   if (data) {
  4.   	rb_erase(&data->node, &mytree);
  5.   	myfree(data);
  6.   }

要使用具有相同键的新节点替换树中的现有节点，请调用:

void rb_replace_node(struct rb_node * old,struct rb_node * new,struct rb_root * tree);

以这种方式替换节点不会对树重新排序：如果新节点不重新排序与旧节点具有相同的key，rbtree可能会损坏。

 

迭代存储在rbtree中的元素（按排序顺序）
-------------------------------------------------- ----------------

提供了四个函数用来迭代rbtree的内容排序顺序。

这些工作在任意树上，不应该是修改或包装（锁用途除外）:

  1. struct rb_node *rb_first(struct rb_root *tree);
  2. struct rb_node *rb_last(struct rb_root *tree);
  3. struct rb_node *rb_next(struct rb_node *node);
  4. struct rb_node *rb_prev(struct rb_node *node);

要开始迭代，请使用指向根的指针调用树的rb_first()或rb_last()，将返回树中的指向包含在其中的节点结构的第一个或最后一个元素指针。
要继续，通过在当前节点上调用rb_next()或rb_prev()来获取下一个或上一个节点。
如果没有剩余节点时返回为NULL。

迭代器函数返回一个指向嵌入式struct rb_node的指针。
可以使用container_of()访问包含数据结构宏，可以通过直接访问个人成员rb_entry（节点，类型，成员）。

例如：

  1. struct rb_node *node;
  2.   for (node = rb_first(&mytree);node;node = rb_next(node))
  3. 	printk("key=%s\n",rb_entry(node,struct mytype,node)->keystring);

缓存的rbtrees
--------------

计算最左边(最小)的节点是二叉搜索树的一个常见任务。

为此，用户可以使用'struct rb_root_cached'优化O(logN) rb_first()调用避免可能很昂贵的树迭代。
虽然内存占用更大，在可忽略的运行时完成维护开销。

与rb_root结构类似，缓存的rbtree被初始化为:
 

struct rb_root_cached mytree = RB_ROOT_CACHED;

缓存的rbtree只是一个常规的rb_root，带有一个额外的指针来缓存它最左边的节点。
这允许rb_root_cached存在于rb_root所在的任何地方，
它允许支持增强树以及仅额外的一些接口：

  1. struct rb_node * rb_first_cached(struct rb_root_cached * tree);
  2. void rb_insert_color_cached(struct rb_node *,struct rb_root_cached *,bool);
  3. void rb_erase_cached(struct rb_node * node,struct rb_root_cached *);

插入和删除调用都有各自的增强对应树：

  1. void rb_insert_augmented_cached(struct rb_node * node,struct rb_root_cached *，bool,struct rb_augment_callbacks *);
  2. void rb_erase_augmented_cached(struct rb_node *,struct rb_root_cached *,struct rb_augment_callbacks *);

支持增强rbtrees
-----------------------------

增强的rbtree是一个rbtree，其中存储了每个节点的“一些”附加数据，其中节点N的附加数据必须是函数子树中所有节点的内容都以N为根。
这个数据可以用于增加rbtree的一些新功能。
增强的rbtree是基于基本rbtree基础结构构建的可选功能。
想要此功能的rbtree用户必须调用增强功能函数与用户在插入时提供了扩充回调和删除节点。

实现增强型rbtree操作的C文件必须包含<linux/rbtree_augmented.h>而不是<linux/rbtree.h>。
注意linux/rbtree_augmented.h公开了一些你不应该依赖的rbtree实现细节;
请坚持使用已记录的API，那里并没有包含头文件中的<linux/rbtree_augmented.h>。
这样的实施细节要么是为了尽量减少用户不小心依赖的机会。

在插入时，用户必须更新路径上的增强信息到插入的节点，然后像往常一样调用rb_link_node()和rb_augment_inserted()代替rb_insert_color()调用。
如果rb_augment_inserted()重新平衡rbtree，它将回调到用户提供的功能来在受影响的子树上更新增强信息。

删除节点时，用户必须调用rb_erase_augmented()而不是rb_erase()。
rb_erase_augmented()回调用户提供的函数更新受影响子树的增强信息。

在这两种情况下，回调都是通过struct rb_augment_callbacks提供的。
必须定义3个回调：

- 传播回调，它更新给定的增强值节点及其祖先，直到给定的停止点（或一直到根更新为NULL）。

- 复制回调，复制给定子树的增强值到新分配的子树根。

- 树旋转回调，复制给定的增强值子树到新分配的子树根并重新计算扩充前子树根的信息。

rb_erase_augmented()的编译代码可以内联传播和复制回调，这导致一个大的函数，所以每个增强rbtree用户应该有一个rb_erase_augmented()调用节点以限制编译代码大小。

 


样例用法
^^^^^^^^^^^^

区间树是增强的rb树的示例，参考《算法导论》。


经典rbtree只有一个key，不能直接用于存储区间范围如[lo：hi]，并快速查找任何重叠新的lo:hi：或者找到与一个新的lo:hi是否完全匹配。

但是，可以扩展rbtree以在结构化中存储这样的间隔范围，这样就可以进行有效的查找和完全匹配。

存储在每个节点中的这个“额外信息”是最大的hi(max_hi)作为其后代的所有节点中的值。
这个只需要查看节点，就可以在每个节点上维护信息及其直系子节点。
这将用于O(log n)查找最低匹配(所有可能匹配中的最低起始地址)。

  1. struct interval_tree_node *
  2.   interval_tree_first_match(struct rb_root *root,
  3. 			    unsigned long start, unsigned long last)
  4.   {
  5. 	struct interval_tree_node *node;
  6.  
  7. 	if (!root->rb_node)
  8. 		return NULL;
  9. 	node = rb_entry(root->rb_node, struct interval_tree_node, rb);
  10.  
  11. 	while (true) {
  12. 		if (node->rb.rb_left) {
  13. 			struct interval_tree_node *left =
  14. 				rb_entry(node->rb.rb_left,
  15. 					 struct interval_tree_node, rb);
  16. 			if (left->__subtree_last >= start) {
  17. 				/*
  18. 				 * Some nodes in left subtree satisfy Cond2.
  19. 				 * Iterate to find the leftmost such node N.
  20. 				 * If it also satisfies Cond1, that's the match
  21. 				 * we are looking for. Otherwise, there is no
  22. 				 * matching interval as nodes to the right of N
  23. 				 * can't satisfy Cond1 either.
  24. 				 */
  25. 				node = left;
  26. 				continue;
  27. 			}
  28. 		}
  29. 		if (node->start <= last) {		/* Cond1 */
  30. 			if (node->last >= start)	/* Cond2 */
  31. 				return node;	/* node is leftmost match */
  32. 			if (node->rb.rb_right) {
  33. 				node = rb_entry(node->rb.rb_right,
  34. 					struct interval_tree_node, rb);
  35. 				if (node->__subtree_last >= start)
  36. 					continue;
  37. 			}
  38. 		}
  39. 		return NULL;	/* No match */
  40. 	}
  41.   }

使用以下扩充回调定义插入/删除:

  1. static inline unsigned long
  2.   compute_subtree_last(struct interval_tree_node *node)
  3.   {
  4. 	unsigned long max = node->last, subtree_last;
  5. 	if (node->rb.rb_left) {
  6. 		subtree_last = rb_entry(node->rb.rb_left,
  7. 			struct interval_tree_node, rb)->__subtree_last;
  8. 		if (max < subtree_last)
  9. 			max = subtree_last;
  10. 	}
  11. 	if (node->rb.rb_right) {
  12. 		subtree_last = rb_entry(node->rb.rb_right,
  13. 			struct interval_tree_node, rb)->__subtree_last;
  14. 		if (max < subtree_last)
  15. 			max = subtree_last;
  16. 	}
  17. 	return max;
  18.   }
  19.  
  20.   static void augment_propagate(struct rb_node *rb, struct rb_node *stop)
  21.   {
  22. 	while (rb != stop) {
  23. 		struct interval_tree_node *node =
  24. 			rb_entry(rb, struct interval_tree_node, rb);
  25. 		unsigned long subtree_last = compute_subtree_last(node);
  26. 		if (node->__subtree_last == subtree_last)
  27. 			break;
  28. 		node->__subtree_last = subtree_last;
  29. 		rb = rb_parent(&node->rb);
  30. 	}
  31.   }
  32.  
  33.   static void augment_copy(struct rb_node *rb_old, struct rb_node *rb_new)
  34.   {
  35. 	struct interval_tree_node *old =
  36. 		rb_entry(rb_old, struct interval_tree_node, rb);
  37. 	struct interval_tree_node *new =
  38. 		rb_entry(rb_new, struct interval_tree_node, rb);
  39.  
  40. 	new->__subtree_last = old->__subtree_last;
  41.   }
  42.  
  43.   static void augment_rotate(struct rb_node *rb_old, struct rb_node *rb_new)
  44.   {
  45. 	struct interval_tree_node *old =
  46. 		rb_entry(rb_old, struct interval_tree_node, rb);
  47. 	struct interval_tree_node *new =
  48. 		rb_entry(rb_new, struct interval_tree_node, rb);
  49.  
  50. 	new->__subtree_last = old->__subtree_last;
  51. 	old->__subtree_last = compute_subtree_last(old);
  52.   }
  53.  
  54.   static const struct rb_augment_callbacks augment_callbacks = {
  55. 	augment_propagate, augment_copy, augment_rotate
  56.   };
  57.  
  58.   void interval_tree_insert(struct interval_tree_node *node,
  59. 			    struct rb_root *root)
  60.   {
  61. 	struct rb_node **link = &root->rb_node, *rb_parent = NULL;
  62. 	unsigned long start = node->start, last = node->last;
  63. 	struct interval_tree_node *parent;
  64.  
  65. 	while (*link) {
  66. 		rb_parent = *link;
  67. 		parent = rb_entry(rb_parent, struct interval_tree_node, rb);
  68. 		if (parent->__subtree_last < last)
  69. 			parent->__subtree_last = last;
  70. 		if (start < parent->start)
  71. 			link = &parent->rb.rb_left;
  72. 		else
  73. 			link = &parent->rb.rb_right;
  74. 	}
  75.  
  76. 	node->__subtree_last = last;
  77. 	rb_link_node(&node->rb, rb_parent, link);
  78. 	rb_insert_augmented(&node->rb, root, &augment_callbacks);
  79.   }
  80.  
  81.   void interval_tree_remove(struct interval_tree_node *node,
  82. 			    struct rb_root *root)
  83.   {
  84. 	rb_erase_augmented(&node->rb, root, &augment_callbacks);
  85.   }

 

下面是内核的rbtree源码：

  1. /*
  2.   Red Black Trees
  3.   (C) 1999  Andrea Arcangeli <andrea@suse.de>
  4.   
  5.   This program is free software; you can redistribute it and/or modify
  6.   it under the terms of the GNU General Public License as published by
  7.   the Free Software Foundation; either version 2 of the License, or
  8.   (at your option) any later version.
  10.   This program is distributed in the hope that it will be useful,
  11.   but WITHOUT ANY WARRANTY; without even the implied warranty of
  12.   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13.   GNU General Public License for more details.
  15.   You should have received a copy of the GNU General Public License
  16.   along with this program; if not, write to the Free Software
  17.   Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  19.   linux/include/linux/rbtree.h
  21.   To use rbtrees you'll have to implement your own insert and search cores.
  22.   This will avoid us to use callbacks and to drop drammatically performances.
  23.   I know it's not the cleaner way,  but in C (not in C++) to get
  24.   performances and genericity...
  26.   Some example of insert and search follows here. The search is a plain
  27.   normal search over an ordered tree. The insert instead must be implemented
  28.   int two steps: as first thing the code must insert the element in
  29.   order as a red leaf in the tree, then the support library function
  30.   rb_insert_color() must be called. Such function will do the
  31.   not trivial work to rebalance the rbtree if necessary.
  33. -----------------------------------------------------------------------
  34. static inline struct page * rb_search_page_cache(struct inode * inode,
  35. 						 unsigned long offset)
  36. {
  37. 	struct rb_node * n = inode->i_rb_page_cache.rb_node;
  38. 	struct page * page;
  40. 	while (n)
  41. 	{
  42. 		page = rb_entry(n, struct page, rb_page_cache);
  44. 		if (offset < page->offset)
  45. 			n = n->rb_left;
  46. 		else if (offset > page->offset)
  47. 			n = n->rb_right;
  48. 		else
  49. 			return page;
  50. 	}
  51. 	return NULL;
  52. }
  54. static inline struct page * __rb_insert_page_cache(struct inode * inode,
  55. 						   unsigned long offset,
  56. 						   struct rb_node * node)
  57. {
  58. 	struct rb_node ** p = &inode->i_rb_page_cache.rb_node;
  59. 	struct rb_node * parent = NULL;
  60. 	struct page * page;
  62. 	while (*p)
  63. 	{
  64. 		parent = *p;
  65. 		page = rb_entry(parent, struct page, rb_page_cache);
  67. 		if (offset < page->offset)
  68. 			p = &(*p)->rb_left;
  69. 		else if (offset > page->offset)
  70. 			p = &(*p)->rb_right;
  71. 		else
  72. 			return page;
  73. 	}
  75. 	rb_link_node(node, parent, p);
  77. 	return NULL;
  78. }
  80. static inline struct page * rb_insert_page_cache(struct inode * inode,
  81. 						 unsigned long offset,
  82. 						 struct rb_node * node)
  83. {
  84. 	struct page * ret;
  85. 	if ((ret = __rb_insert_page_cache(inode, offset, node)))
  86. 		goto out;
  87. 	rb_insert_color(node, &inode->i_rb_page_cache);
  88.  out:
  89. 	return ret;
  90. }
  91. -----------------------------------------------------------------------
  92. */
  93.  
  94. #ifndef	_LINUX_RBTREE_H
  95. #define	_LINUX_RBTREE_H
  96.  
  97. #include <linux/kernel.h>
  98. #include <linux/stddef.h>
  99.  
  100. struct rb_node
  101. {
  102. 	unsigned long  rb_parent_color;
  103. #define	RB_RED		0
  104. #define	RB_BLACK	1
  105. 	struct rb_node *rb_right;
  106. 	struct rb_node *rb_left;
  107. } __attribute__((aligned(sizeof(long))));
  108.     /* The alignment might seem pointless, but allegedly CRIS needs it */
  109.  
  110. struct rb_root
  111. {
  112. 	struct rb_node *rb_node;
  113. };
  114.  
  115.  
  116. #define rb_parent(r)   ((struct rb_node *)((r)->rb_parent_color & ~3))
  117. #define rb_color(r)   ((r)->rb_parent_color & 1)
  118. #define rb_is_red(r)   (!rb_color(r))
  119. #define rb_is_black(r) rb_color(r)
  120. #define rb_set_red(r)  do { (r)->rb_parent_color &= ~1; } while (0)
  121. #define rb_set_black(r)  do { (r)->rb_parent_color |= 1; } while (0)
  122.  
  123. static inline void rb_set_parent(struct rb_node *rb, struct rb_node *p)
  124. {
  125. 	rb->rb_parent_color = (rb->rb_parent_color & 3) | (unsigned long)p;
  126. }
  127. static inline void rb_set_color(struct rb_node *rb, int color)
  128. {
  129. 	rb->rb_parent_color = (rb->rb_parent_color & ~1) | color;
  130. }
  131.  
  132. #define RB_ROOT	(struct rb_root) { NULL, }
  133. #define	rb_entry(ptr, type, member) container_of(ptr, type, member)
  134.  
  135. #define RB_EMPTY_ROOT(root)	((root)->rb_node == NULL)
  136. #define RB_EMPTY_NODE(node)	(rb_parent(node) == node)
  137. #define RB_CLEAR_NODE(node)	(rb_set_parent(node, node))
  138.  
  139. extern void rb_insert_color(struct rb_node *, struct rb_root *);
  140. extern void rb_erase(struct rb_node *, struct rb_root *);
  141.  
  142. /* Find logical next and previous nodes in a tree */
  143. extern struct rb_node *rb_next(const struct rb_node *);
  144. extern struct rb_node *rb_prev(const struct rb_node *);
  145. extern struct rb_node *rb_first(const struct rb_root *);
  146. extern struct rb_node *rb_last(const struct rb_root *);
  147.  
  148. /* Fast replacement of a single node without remove/rebalance/add/rebalance */
  149. extern void rb_replace_node(struct rb_node *victim, struct rb_node *new, 
  150. 			    struct rb_root *root);
  151.  
  152. static inline void rb_link_node(struct rb_node * node, struct rb_node * parent,
  153. 				struct rb_node ** rb_link)
  154. {
  155. 	node->rb_parent_color = (unsigned long )parent;
  156. 	node->rb_left = node->rb_right = NULL;
  157.  
  158. 	*rb_link = node;
  159. }
  160.  
  161. #endif	/* _LINUX_RBTREE_H */
  1. /*
  2.   Red Black Trees
  3.   (C) 1999  Andrea Arcangeli <andrea@suse.de>
  4.   (C) 2002  David Woodhouse <dwmw2@infradead.org>
  5.   
  6.   This program is free software; you can redistribute it and/or modify
  7.   it under the terms of the GNU General Public License as published by
  8.   the Free Software Foundation; either version 2 of the License, or
  9.   (at your option) any later version.
  11.   This program is distributed in the hope that it will be useful,
  12.   but WITHOUT ANY WARRANTY; without even the implied warranty of
  13.   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14.   GNU General Public License for more details.
  16.   You should have received a copy of the GNU General Public License
  17.   along with this program; if not, write to the Free Software
  18.   Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  20.   linux/lib/rbtree.c
  21. */
  22.  
  23. #include <linux/rbtree.h>
  24. #include <linux/module.h>
  25.  
  26. static void __rb_rotate_left(struct rb_node *node, struct rb_root *root)
  27. {
  28. 	struct rb_node *right = node->rb_right;
  29. 	struct rb_node *parent = rb_parent(node);
  30.  
  31. 	if ((node->rb_right = right->rb_left))
  32. 		rb_set_parent(right->rb_left, node);
  33. 	right->rb_left = node;
  34.  
  35. 	rb_set_parent(right, parent);
  36.  
  37. 	if (parent)
  38. 	{
  39. 		if (node == parent->rb_left)
  40. 			parent->rb_left = right;
  41. 		else
  42. 			parent->rb_right = right;
  43. 	}
  44. 	else
  45. 		root->rb_node = right;
  46. 	rb_set_parent(node, right);
  47. }
  48.  
  49. static void __rb_rotate_right(struct rb_node *node, struct rb_root *root)
  50. {
  51. 	struct rb_node *left = node->rb_left;
  52. 	struct rb_node *parent = rb_parent(node);
  53.  
  54. 	if ((node->rb_left = left->rb_right))
  55. 		rb_set_parent(left->rb_right, node);
  56. 	left->rb_right = node;
  57.  
  58. 	rb_set_parent(left, parent);
  59.  
  60. 	if (parent)
  61. 	{
  62. 		if (node == parent->rb_right)
  63. 			parent->rb_right = left;
  64. 		else
  65. 			parent->rb_left = left;
  66. 	}
  67. 	else
  68. 		root->rb_node = left;
  69. 	rb_set_parent(node, left);
  70. }
  71.  
  72. void rb_insert_color(struct rb_node *node, struct rb_root *root)
  73. {
  74. 	struct rb_node *parent, *gparent;
  75.  
  76. 	while ((parent = rb_parent(node)) && rb_is_red(parent))
  77. 	{
  78. 		gparent = rb_parent(parent);
  79.  
  80. 		if (parent == gparent->rb_left)
  81. 		{
  82. 			{
  83. 				register struct rb_node *uncle = gparent->rb_right;
  84. 				if (uncle && rb_is_red(uncle))
  85. 				{
  86. 					rb_set_black(uncle);
  87. 					rb_set_black(parent);
  88. 					rb_set_red(gparent);
  89. 					node = gparent;
  90. 					continue;
  91. 				}
  92. 			}
  93.  
  94. 			if (parent->rb_right == node)
  95. 			{
  96. 				register struct rb_node *tmp;
  97. 				__rb_rotate_left(parent, root);
  98. 				tmp = parent;
  99. 				parent = node;
  100. 				node = tmp;
  101. 			}
  102.  
  103. 			rb_set_black(parent);
  104. 			rb_set_red(gparent);
  105. 			__rb_rotate_right(gparent, root);
  106. 		} else {
  107. 			{
  108. 				register struct rb_node *uncle = gparent->rb_left;
  109. 				if (uncle && rb_is_red(uncle))
  110. 				{
  111. 					rb_set_black(uncle);
  112. 					rb_set_black(parent);
  113. 					rb_set_red(gparent);
  114. 					node = gparent;
  115. 					continue;
  116. 				}
  117. 			}
  118.  
  119. 			if (parent->rb_left == node)
  120. 			{
  121. 				register struct rb_node *tmp;
  122. 				__rb_rotate_right(parent, root);
  123. 				tmp = parent;
  124. 				parent = node;
  125. 				node = tmp;
  126. 			}
  127.  
  128. 			rb_set_black(parent);
  129. 			rb_set_red(gparent);
  130. 			__rb_rotate_left(gparent, root);
  131. 		}
  132. 	}
  133.  
  134. 	rb_set_black(root->rb_node);
  135. }
  136. EXPORT_SYMBOL(rb_insert_color);
  137.  
  138. static void __rb_erase_color(struct rb_node *node, struct rb_node *parent,
  139. 			     struct rb_root *root)
  140. {
  141. 	struct rb_node *other;
  142.  
  143. 	while ((!node || rb_is_black(node)) && node != root->rb_node)
  144. 	{
  145. 		if (parent->rb_left == node)
  146. 		{
  147. 			other = parent->rb_right;
  148. 			if (rb_is_red(other))
  149. 			{
  150. 				rb_set_black(other);
  151. 				rb_set_red(parent);
  152. 				__rb_rotate_left(parent, root);
  153. 				other = parent->rb_right;
  154. 			}
  155. 			if ((!other->rb_left || rb_is_black(other->rb_left)) &&
  156. 			    (!other->rb_right || rb_is_black(other->rb_right)))
  157. 			{
  158. 				rb_set_red(other);
  159. 				node = parent;
  160. 				parent = rb_parent(node);
  161. 			}
  162. 			else
  163. 			{
  164. 				if (!other->rb_right || rb_is_black(other->rb_right))
  165. 				{
  166. 					rb_set_black(other->rb_left);
  167. 					rb_set_red(other);
  168. 					__rb_rotate_right(other, root);
  169. 					other = parent->rb_right;
  170. 				}
  171. 				rb_set_color(other, rb_color(parent));
  172. 				rb_set_black(parent);
  173. 				rb_set_black(other->rb_right);
  174. 				__rb_rotate_left(parent, root);
  175. 				node = root->rb_node;
  176. 				break;
  177. 			}
  178. 		}
  179. 		else
  180. 		{
  181. 			other = parent->rb_left;
  182. 			if (rb_is_red(other))
  183. 			{
  184. 				rb_set_black(other);
  185. 				rb_set_red(parent);
  186. 				__rb_rotate_right(parent, root);
  187. 				other = parent->rb_left;
  188. 			}
  189. 			if ((!other->rb_left || rb_is_black(other->rb_left)) &&
  190. 			    (!other->rb_right || rb_is_black(other->rb_right)))
  191. 			{
  192. 				rb_set_red(other);
  193. 				node = parent;
  194. 				parent = rb_parent(node);
  195. 			}
  196. 			else
  197. 			{
  198. 				if (!other->rb_left || rb_is_black(other->rb_left))
  199. 				{
  200. 					rb_set_black(other->rb_right);
  201. 					rb_set_red(other);
  202. 					__rb_rotate_left(other, root);
  203. 					other = parent->rb_left;
  204. 				}
  205. 				rb_set_color(other, rb_color(parent));
  206. 				rb_set_black(parent);
  207. 				rb_set_black(other->rb_left);
  208. 				__rb_rotate_right(parent, root);
  209. 				node = root->rb_node;
  210. 				break;
  211. 			}
  212. 		}
  213. 	}
  214. 	if (node)
  215. 		rb_set_black(node);
  216. }
  217.  
  218. void rb_erase(struct rb_node *node, struct rb_root *root)
  219. {
  220. 	struct rb_node *child, *parent;
  221. 	int color;
  222.  
  223. 	if (!node->rb_left)
  224. 		child = node->rb_right;
  225. 	else if (!node->rb_right)
  226. 		child = node->rb_left;
  227. 	else
  228. 	{
  229. 		struct rb_node *old = node, *left;
  230.  
  231. 		node = node->rb_right;
  232. 		while ((left = node->rb_left) != NULL)
  233. 			node = left;
  234.  
  235. 		if (rb_parent(old)) {
  236. 			if (rb_parent(old)->rb_left == old)
  237. 				rb_parent(old)->rb_left = node;
  238. 			else
  239. 				rb_parent(old)->rb_right = node;
  240. 		} else
  241. 			root->rb_node = node;
  242.  
  243. 		child = node->rb_right;
  244. 		parent = rb_parent(node);
  245. 		color = rb_color(node);
  246.  
  247. 		if (parent == old) {
  248. 			parent = node;
  249. 		} else {
  250. 			if (child)
  251. 				rb_set_parent(child, parent);
  252. 			parent->rb_left = child;
  253.  
  254. 			node->rb_right = old->rb_right;
  255. 			rb_set_parent(old->rb_right, node);
  256. 		}
  257.  
  258. 		node->rb_parent_color = old->rb_parent_color;
  259. 		node->rb_left = old->rb_left;
  260. 		rb_set_parent(old->rb_left, node);
  261.  
  262. 		goto color;
  263. 	}
  264.  
  265. 	parent = rb_parent(node);
  266. 	color = rb_color(node);
  267.  
  268. 	if (child)
  269. 		rb_set_parent(child, parent);
  270. 	if (parent)
  271. 	{
  272. 		if (parent->rb_left == node)
  273. 			parent->rb_left = child;
  274. 		else
  275. 			parent->rb_right = child;
  276. 	}
  277. 	else
  278. 		root->rb_node = child;
  279.  
  280.  color:
  281. 	if (color == RB_BLACK)
  282. 		__rb_erase_color(child, parent, root);
  283. }
  284. EXPORT_SYMBOL(rb_erase);
  285.  
  286. /*
  287.  * This function returns the first node (in sort order) of the tree.
  288.  */
  289. struct rb_node *rb_first(const struct rb_root *root)
  290. {
  291. 	struct rb_node	*n;
  292.  
  293. 	n = root->rb_node;
  294. 	if (!n)
  295. 		return NULL;
  296. 	while (n->rb_left)
  297. 		n = n->rb_left;
  298. 	return n;
  299. }
  300. EXPORT_SYMBOL(rb_first);
  301.  
  302. struct rb_node *rb_last(const struct rb_root *root)
  303. {
  304. 	struct rb_node	*n;
  305.  
  306. 	n = root->rb_node;
  307. 	if (!n)
  308. 		return NULL;
  309. 	while (n->rb_right)
  310. 		n = n->rb_right;
  311. 	return n;
  312. }
  313. EXPORT_SYMBOL(rb_last);
  314.  
  315. struct rb_node *rb_next(const struct rb_node *node)
  316. {
  317. 	struct rb_node *parent;
  318.  
  319. 	if (rb_parent(node) == node)
  320. 		return NULL;
  321.  
  322. 	/* If we have a right-hand child, go down and then left as far
  323. 	   as we can. */
  324. 	if (node->rb_right) {
  325. 		node = node->rb_right; 
  326. 		while (node->rb_left)
  327. 			node=node->rb_left;
  328. 		return (struct rb_node *)node;
  329. 	}
  330.  
  331. 	/* No right-hand children.  Everything down and left is
  332. 	   smaller than us, so any 'next' node must be in the general
  333. 	   direction of our parent. Go up the tree; any time the
  334. 	   ancestor is a right-hand child of its parent, keep going
  335. 	   up. First time it's a left-hand child of its parent, said
  336. 	   parent is our 'next' node. */
  337. 	while ((parent = rb_parent(node)) && node == parent->rb_right)
  338. 		node = parent;
  339.  
  340. 	return parent;
  341. }
  342. EXPORT_SYMBOL(rb_next);
  343.  
  344. struct rb_node *rb_prev(const struct rb_node *node)
  345. {
  346. 	struct rb_node *parent;
  347.  
  348. 	if (rb_parent(node) == node)
  349. 		return NULL;
  350.  
  351. 	/* If we have a left-hand child, go down and then right as far
  352. 	   as we can. */
  353. 	if (node->rb_left) {
  354. 		node = node->rb_left; 
  355. 		while (node->rb_right)
  356. 			node=node->rb_right;
  357. 		return (struct rb_node *)node;
  358. 	}
  359.  
  360. 	/* No left-hand children. Go up till we find an ancestor which
  361. 	   is a right-hand child of its parent */
  362. 	while ((parent = rb_parent(node)) && node == parent->rb_left)
  363. 		node = parent;
  364.  
  365. 	return parent;
  366. }
  367. EXPORT_SYMBOL(rb_prev);
  368.  
  369. void rb_replace_node(struct rb_node *victim, struct rb_node *new,
  370. 		     struct rb_root *root)
  371. {
  372. 	struct rb_node *parent = rb_parent(victim);
  373.  
  374. 	/* Set the surrounding nodes to point to the replacement */
  375. 	if (parent) {
  376. 		if (victim == parent->rb_left)
  377. 			parent->rb_left = new;
  378. 		else
  379. 			parent->rb_right = new;
  380. 	} else {
  381. 		root->rb_node = new;
  382. 	}
  383. 	if (victim->rb_left)
  384. 		rb_set_parent(victim->rb_left, new);
  385. 	if (victim->rb_right)
  386. 		rb_set_parent(victim->rb_right, new);
  387.  
  388. 	/* Copy the pointers/colour from the victim to the replacement */
  389. 	*new = *victim;
  390. }
  391. EXPORT_SYMBOL(rb_replace_node);

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/凡人多烦事01/article/detail/246020
推荐阅读
相关标签

Copyright © 2003-2013 www.wpsshop.cn 版权所有，并保留所有权利。

  

闽ICP备14008679号