经验首页 前端设计 程序设计 Java相关 移动开发 数据库/运维 软件/图像 大数据/云计算 其他经验
当前位置:技术经验 » 程序设计 » C# » 查看文章
图解B树及C#实现(3)数据的删除
来源:cnblogs  作者:黑洞视界  时间:2023/2/6 9:03:42  对本文有异议

前言

本文为系列文章

  1. B树的定义及数据的插入
  2. 数据的读取及遍历
  3. 数据的删除

阅读本文前,建议先复习前两篇文章,以便更好的理解本文。

从删除的数据所在的节点可分为两种情况:

  1. 从叶子节点删除数据
  2. 从非叶子节点删除数据

无论从叶子节点还是非叶子节点删除数据时都需要保证B树的特性:非根节点每个节点的 key 数量都在 [t-1, 2t-1] 之间

借此保证B树的平衡性。

之前介绍的插入数据关注的是这个范围的上限 2t-1,插入时,如果节点的 key 数量大于 2t-1,就需要进行数据的分裂。

而删除数据则关注是下限 t-1,如果节点的 key 数量小于 t-1,就需要进行数据的移动或者合并。

删除数据时,需要考虑的情况比较多,本文会分别讨论这些情况,但一些比较边缘的情况为避免描述过于复杂,不再文中讨论,而是在代码中进行了注释。

因为删除逻辑比较复杂,请结合完整代码进行阅读。
https://github.com/eventhorizon-cli/EventHorizon.BTree/blob/b51881719146a86568669cdc78f8524299bee33d/src/EventHorizon.BTree/BTree.cs#L139

从叶子节点删除数据

如果待删除的数据在叶子节点,且该节点的 Item 数量大于 t-1,那么直接删除该数据即可。

从非叶子节点删除数据

如果待删除的数据在非叶子节点,那么需要先找到该数据的左子节点,然后将左子节点的数据替换到待删除的数据,最后再删除左子节点的数据。

这样能保证被删除数据的节点的 Item 数量不变,保证 B树 有 k 个子节点的非叶子节点拥有 k ? 1 个键的特性不受破坏。

提前扩充只有 t-1 的 Item 的节点:维持 B树 平衡的核心算法

在数据插入的时候,为了避免回溯性的节点分裂,我们提前将已满的子节点进行分裂。

同样的在数据删除,不断往下递归查找时,如果遇到只有 t-1 个 Item 的节点,我们也需要提前将其扩充,以避免回溯性的节点处理。

扩充的节点不一定是最后数据所在的节点,只是向下查找过程中遇到的节点。

节点扩充的分为两类,一个是从兄弟节点借用 Item,一个是合并兄弟节点,被借用的兄弟节点需要满足 Item 数量大于 t-1。具体可分为以下三种情况:

从左兄弟节点借用 Item

待扩充节点的左兄弟节点存在且左兄弟节点的 Item 数量 > t-1 时,从左兄弟节点借用 Item 进行扩充。

为了保证 B树 数据的顺序特性:任意 Item 的左子树中的 Key 均小于该 Item 的 Key,右子树中的 Key 均大于该 Item 的 Key。需要交换左兄弟节点的最右边的 Item 和父节点中对应位置的 Item(位于左兄弟节点右侧)。

以下图为例进行说明:

从右兄弟节点借用 Item

待扩充节点的左兄弟节点不存在或者左兄弟节点的 Item 数量 只有 t-1 时,无法外借。但右兄弟节点存在且右兄弟节点的 Item 数量 > t-1 时,从右兄弟节点借用 Item 进行扩充。

以下图为例进行说明:

从兄弟节点进行扩充可以概括为:借用,交换,插入。

与左兄弟节点或者右兄弟节点合并

如果待扩充节点的左兄弟节点和右兄弟节点都不存在或者都只有 t-1 个 Item 时,无法外借。此时需要与左兄弟节点或者右兄弟节点进行合并。

以下图为例进行说明:

最值的删除

之前章节介绍过 B树 最值的查找:

  1. 最小值:从根节点开始,一直往左子树走,直到叶子节点。
  2. 最大值:从根节点开始,一直往右子树走,直到叶子节点。

最值的删除就是先找到最值的位置并将其删除,在向下寻找的过程中,需要和普通的数据删除一样,对节点进行扩充或者合并。

代码实现

最值删除是删除的特殊情况,我们定义一个枚举用来区分普通数据的删除,最小值的删除以及最大值的删除,这三种方式只在数据查找的时候有所区分,其他的逻辑都是一样的。

  1. internal enum RemoveType
  2. {
  3. Item,
  4. Min,
  5. Max
  6. }
  7. public sealed class BTree<TKey, TValue> : IEnumerable<KeyValuePair<TKey, TValue?>>
  8. {
  9. public bool TryRemove([NotNull] TKey key, out TValue? value)
  10. {
  11. ArgumentNullException.ThrowIfNull(key);
  12. return TryRemove(key, RemoveType.Item, out value);
  13. }
  14. public bool TryRemoveMax(out TValue? value) => TryRemove(default, RemoveType.Max, out value);
  15. public bool TryRemoveMin(out TValue? value) => TryRemove(default, RemoveType.Min, out value);
  16. private bool TryRemove(TKey? key, RemoveType removeType, out TValue? value)
  17. {
  18. if (_root == null || _root.IsItemsEmpty)
  19. {
  20. value = default;
  21. return false;
  22. }
  23. bool removed = _root.TryRemove(key, removeType, out var item);
  24. if (_root.IsItemsEmpty && !_root.IsLeaf)
  25. {
  26. // 根节点原来的两个子节点进行了合并,根节点唯一的元素被移动到了子节点中,需要将合并后的子节点设置为新的根节点
  27. _root = _root.GetChild(0);
  28. }
  29. if (removed)
  30. {
  31. _count--;
  32. value = item!.Value;
  33. return true;
  34. }
  35. value = default;
  36. return removed;
  37. }
  38. }

主要的逻辑定义在 Node 中,不断向下递归

  1. internal class Node<TKey, TValue>
  2. {
  3. public bool TryRemove(TKey? key, RemoveType removeType, [MaybeNullWhen(false)] out Item<TKey, TValue?> item)
  4. {
  5. int index = 0;
  6. bool found = false;
  7. if (removeType == RemoveType.Max)
  8. {
  9. if (IsLeaf)
  10. {
  11. if (_items.Count == 0)
  12. {
  13. item = default;
  14. return false;
  15. }
  16. // 如果是叶子节点,直接删除最后一个元素,就是删除最大的 Item
  17. item = _items.RemoveLast();
  18. return true;
  19. }
  20. // 当前节点不是叶子节点,需要找到最大的子节点,继续向下查找并删除
  21. index = ItemsCount;
  22. }
  23. if (removeType == RemoveType.Min)
  24. {
  25. if (IsLeaf)
  26. {
  27. if (_items.Count == 0)
  28. {
  29. item = default;
  30. return false;
  31. }
  32. // 当前节点是叶子节点,直接删除第一个元素,就是删除最小的 Item
  33. item = _items.RemoveAt(0);
  34. return true;
  35. }
  36. // 当前节点不是叶子节点,需要找到最小的子节点,继续向下查找并删除
  37. index = 0;
  38. }
  39. if (removeType == RemoveType.Item)
  40. {
  41. // 如果没有找到,index 表示的是 key 可能在的子树的索引
  42. found = _items.TryFindKey(key!, out index);
  43. if (IsLeaf)
  44. {
  45. // 如果是叶子节点,能找到就删除,找不到就返回 false,表示删除失败
  46. if (found)
  47. {
  48. item = _items.RemoveAt(index);
  49. return true;
  50. }
  51. item = default;
  52. return false;
  53. }
  54. }
  55. // 如果当前节点的左子节点的 Item 个数小于最小 Item 个数,就需要进行合并或者借元素
  56. // 这个处理对应两种情况:
  57. // 1. 要删除的 Item 不在当前节点的子节点中,为避免删除后导致数据所在节点的 Item 个数小于最小 Item 个数,需要先进行合并或者借元素。
  58. // 2. 要删除的 Item 就在当前节点中,为避免删除后导致当前节点的 Item 个数小于最小 Item 个数,需要先从左子节点中借一个 Item 过来,保证当前节点的 Item 数量不变。
  59. // 为此先要保证左子节点被借用后的 Item 个数不小于最小 Item 个数。
  60. if (_children[index].ItemsCount <= _minItems)
  61. {
  62. return GrowChildrenAndTryRemove(index, key!, removeType, out item);
  63. }
  64. var child = _children[index];
  65. if (found)
  66. {
  67. // 如果在当前节点找到了,就删除当前节点的 Item,然后将 左子节点 中的最大的 Item 移动到当前节点中
  68. // 以维持当前节点的 Item 个数不变,保证 B树 有 k 个子节点的非叶子节点拥有 k ? 1 个键的特性。
  69. item = _items[index];
  70. child.TryRemove(default!, RemoveType.Max, out var stolenItem);
  71. _items[index] = stolenItem;
  72. return true;
  73. }
  74. return child.TryRemove(key!, removeType, out item);
  75. }
  76. private bool GrowChildrenAndTryRemove(
  77. int childIndex,
  78. TKey key,
  79. RemoveType removeType,
  80. [MaybeNullWhen(false)] out Item<TKey, TValue?> item)
  81. {
  82. if (childIndex > 0 && _children[childIndex - 1].ItemsCount > _minItems)
  83. {
  84. // 如果左边的子节点存在且左边的子节点的item数量大于最小值,则从左边的子节点借一个item
  85. var child = _children[childIndex];
  86. var leftChild = _children[childIndex - 1];
  87. var stolenItem = leftChild._items.RemoveLast();
  88. child._items.InsertAt(0, _items[childIndex - 1]);
  89. _items[childIndex - 1] = stolenItem;
  90. if (!leftChild.IsLeaf)
  91. {
  92. // 非叶子节点的子节点需要保证数量比item多1,item数量变了,子节点数量也要变
  93. // 所以需要从左边的子节点中移除最后一个子节点,然后插入到当前子节点的第一个位置
  94. child._children.InsertAt(0, leftChild._children.RemoveLast());
  95. }
  96. }
  97. else if (childIndex < ChildrenCount - 1 && _children[childIndex + 1].ItemsCount > _minItems)
  98. {
  99. // 如果右边的子节点存在且右边的子节点的item数量大于最小值,则从右边的子节点借一个item
  100. var child = _children[childIndex];
  101. var rightChild = _children[childIndex + 1];
  102. var stolenItem = rightChild._items.RemoveAt(0);
  103. child._items.Add(_items[childIndex]);
  104. _items[childIndex] = stolenItem;
  105. if (!rightChild.IsLeaf)
  106. {
  107. // 非叶子节点的子节点需要保证数量比item多1,item数量变了,子节点数量也要变
  108. // 所以需要从右边的子节点中移除第一个子节点,然后插入到当前子节点的最后一个位置
  109. child.AddChild(rightChild._children.RemoveAt(0));
  110. }
  111. }
  112. else
  113. {
  114. // 如果当前节点左右两边的子节点的item数量都不大于最小值(例如正好等于最小值 t-1 ),则合并当前节点和右边的子节点或者左边的子节点
  115. // 优先和右边的子节点合并,如果右边的子节点不存在,则和左边的子节点合并
  116. if (childIndex >= ItemsCount)
  117. {
  118. // ItemCount 代表最的子节点的索引,如果 childIndex 大于等于 ItemCount,说明右边的子节点不存在,需要和左边的子节点合并
  119. childIndex--;
  120. }
  121. var child = _children[childIndex];
  122. var mergeItem = _items.RemoveAt(childIndex);
  123. var mergeChild = _children.RemoveAt(childIndex + 1);
  124. child._items.Add(mergeItem);
  125. child._items.AddRange(mergeChild._items);
  126. child._children.AddRange(mergeChild._children);
  127. }
  128. return TryRemove(key, removeType, out item);
  129. }
  130. }

Benchmarks:与 优先队列 PriorityQueue 的比较

我们实现的 BTree 支持自定义排序规则,也实现最值的删除,意味着可以充当优先队列使用。

我们使用 PriorityQueue 与 BTree 进行性能对比来看看 B树 能否充当优先队列使用。

入队性能

  1. public class BTree_PriorityQueue_EnequeueBenchmarks
  2. {
  3. [Params(1000, 1_0000, 10_0000)] public int DataSize;
  4. [Params(2, 4, 8, 16)] public int Degree;
  5. private HashSet<int> _data;
  6. [IterationSetup]
  7. public void Setup()
  8. {
  9. var random = new Random();
  10. _data = new HashSet<int>();
  11. while (_data.Count < DataSize)
  12. {
  13. var value = random.Next();
  14. _data.Add(value);
  15. }
  16. }
  17. [Benchmark]
  18. public void BTree_Add()
  19. {
  20. var btree = new BTree<int, int>(Degree);
  21. foreach (var value in _data)
  22. {
  23. btree.Add(value, value);
  24. }
  25. }
  26. [Benchmark]
  27. public void PriorityQueue_Enqueue()
  28. {
  29. var priorityQueue = new PriorityQueue<int, int>(DataSize);
  30. foreach (var value in _data)
  31. {
  32. priorityQueue.Enqueue(value, value);
  33. }
  34. }
  35. }

出队性能

  1. public class BTree_PriorityQueue_DequeueBenchmarks
  2. {
  3. [Params(1000, 1_0000, 10_0000)] public int DataSize;
  4. [Params(2, 4, 8, 16)] public int Degree;
  5. private BTree<int, int> _btree;
  6. private PriorityQueue<int, int> _priorityQueue;
  7. [IterationSetup]
  8. public void Setup()
  9. {
  10. var random = new Random();
  11. _btree = new BTree<int, int>(Degree);
  12. _priorityQueue = new PriorityQueue<int, int>(DataSize);
  13. while (_btree.Count < DataSize)
  14. {
  15. var value = random.Next();
  16. _btree.Add(value, value);
  17. _priorityQueue.Enqueue(value, value);
  18. }
  19. }
  20. [Benchmark]
  21. public void BTree_Remove()
  22. {
  23. while (_btree.Count > 0)
  24. {
  25. _btree.RemoveMin();
  26. }
  27. }
  28. [Benchmark]
  29. public void PriorityQueue_Dequeue()
  30. {
  31. while (_priorityQueue.Count > 0)
  32. {
  33. _priorityQueue.Dequeue();
  34. }
  35. }
  36. }

可以看到,B树 虽然在入队性能上比 PriorityQueue 差。但在数据量和 degree 较大时,出队性能比 PriorityQueue 好,是有能力充当优先队列使用的。

总结

B树 在 degree 较大时,树的高度较低,删除的效率较高,可充当优先队列使用。

B树 的插入,删除,查找都是基于递归的,递归的深度为树的高度。

B树 对数据的查找基于二分查找,时间复杂度为 O(log n),B树 的插入和删除基于 B树的查找算法,都要找到数据所在的节点,然后在该节点进行插入和删除。因此,B树 的插入和删除的时间复杂度也为 O(log n)。

B树 是对二叉树的一种优化,使得树的高度更低,但是在插入,删除的过程中,需要进行大量的节点分裂,合并,借用,交换等操作,使得算法的复杂度更高。

参考资料

Google 用 Go 实现的内存版 B树 https://github.com/google/btree

B树 维基百科 https://zh.m.wikipedia.org/zh-hans/B树

原文链接:https://www.cnblogs.com/eventhorizon/p/17092336.html

 友情链接:直通硅谷  点职佳  北美留学生论坛

本站QQ群:前端 618073944 | Java 606181507 | Python 626812652 | C/C++ 612253063 | 微信 634508462 | 苹果 692586424 | C#/.net 182808419 | PHP 305140648 | 运维 608723728

W3xue 的所有内容仅供测试,对任何法律问题及风险不承担任何责任。通过使用本站内容随之而来的风险与本站无关。
关于我们  |  意见建议  |  捐助我们  |  报错有奖  |  广告合作、友情链接(目前9元/月)请联系QQ:27243702 沸活量
皖ICP备17017327号-2 皖公网安备34020702000426号