手撕二叉搜索树(Binary Search Tree)

x33g5p2x  于2022-06-24 转载在 其他  
字(4.3k)|赞(0)|评价(0)|浏览(351)

⏰1.二叉搜索树的概念

二叉搜索树又称二叉排序树。

🌕二叉搜索树的特性

1. 二叉搜索树的左孩子比父节点小,右孩子比父节点大。

2.二叉搜索树的左子树的全部节点都小于根节点,右子树的全部节点都大于根节点。

3.所有节点的左右子树都为二叉搜索树

4. 键值是唯一的,所以二叉搜索树不能有相同的键值。

🌕二叉搜索树的使用场景:

根据使用场景的不同,二叉搜索树还分为K模型和KV模型:
K模型:即只有key作为关键码,只需要存储Key即可,关键码即为需要搜索到的值。如STL中的set

KV模型:每一个关键码key,都有与之对应的值Value,即<Key, Value>的键值对。如STL中的map

⏰2.二叉搜索树的性能分析

由于二叉搜索树的特征,对有n个结点的二叉搜索树,若每个元素查找的概率相等,则二叉搜索树平均查找长度是结点在二叉搜索树的深度的函数,即结点越深,则比较次数越多。

但对于同一个关键码集合,如果各关键码插入的次序不同,可能得到不同结构的二叉搜索树:

如上图所示,最优情况下,二叉搜索树为完全二叉树,其平均比较次数为:log2N

最差情况下,二叉搜索树退化为单支树,其平均比较次数为:N/2

⏰3.二叉搜索树的实现

这里只介绍最常用的查找、插入、删除三个接口

节点由左右指针和一个值val构成

  1. template<class K>
  2. struct BSTreeNode
  3. {
  4. public:
  5. BSTreeNode(const K& val = K())
  6. :_left(nullptr)
  7. , _right(nullptr)
  8. , _val(val)
  9. {}
  10. BSTreeNode<K>* _left;
  11. BSTreeNode<K>* _right;
  12. K _val;
  13. };

🎄排序

根据二叉搜索树的特性,左子树小于根节点,右子树大于根节点。

所以通过一趟中序遍历,即可获得排序的结果。

根据这条性质,你可以AC leetcode 98.验证二叉搜索树

🎄查找

二叉搜索树的查找十分简单,键值比根节点大则进入右子树,键值比根节点小则进入左子树,他的思路有点类似于

二分查找,平均时间复杂度为O(log2N)。

但如果构建时树为有序数列,则二叉搜索树会退化为单支树,时间复杂度则会变为O(N)

退化成单支树时间复杂度剧增怎么办?
为搜索二叉树加上平衡二叉树的属性,也就是我们通常所说的AVL树。

查找操作的实现

从根节点出发,比根节点大则查找右子树,比根节点小则查找左子树,相同则返回。如果遍历完还没找到,则说明不存在此树中,返回nullptr

  1. Node* Find(const K& key)
  2. {
  3. //根据二叉搜索树的性质,从根节点出发,比根节点大则查找右子树,比根节点小则查找左子树
  4. Node* cur = _root;
  5. while (cur)
  6. {
  7. //比根节点大则查找右子树
  8. if (key > cur->_key)
  9. {
  10. cur = cur->_right;
  11. }
  12. //比根节点小则查找左子树
  13. else if (key < cur->_key)
  14. {
  15. cur = cur->_left;
  16. }
  17. //相同则返回
  18. else
  19. {
  20. return cur;
  21. }
  22. }
  23. //遍历完则说明查找不到,返回false
  24. return nullptr;
  25. }

🎄插入

首先要找到我们需要插入的位置(为了保证能够找到父节点还需要用一个指针指向父节点),找到了合适的位置后,我们需要判断当前的键值比父节点大还是比父节点小,来决定应该插入到父节点的左子树还是右子树

1.如果是空树,直接插入。

2.如果树非空,先找到要插入的结点位置,再插入。

🌕插入操作的实现

  • 插入节点一定是插入值为NULL节点的位置
  • 插入节点后记得父节点要与插入节点连接
  • 插入的节点比当前节点大往右子树走,反之往左子树走,相等返回false(插入失败)
  1. bool Insert(const K& key)
  2. {
  3. //如果此时树为空,则初始化根节点
  4. if (_root == nullptr)
  5. {
  6. _root = new Node(key);
  7. return true;
  8. }
  9. Node* cur = _root;
  10. Node* parent = cur;
  11. //找到合适的插入位置
  12. while (cur)
  13. {
  14. //比根节点大则查找右子树
  15. if (key > cur->_key)
  16. {
  17. parent = cur;
  18. cur = cur->_right;
  19. }
  20. //比根节点小则查找左子树
  21. else if (key < cur->_key)
  22. {
  23. parent = cur;
  24. cur = cur->_left;
  25. }
  26. //相同则返回false,因为搜索树不能存在相同数据
  27. else
  28. {
  29. return false;
  30. }
  31. }
  32. cur = new Node(key);
  33. //判断cur要插入到parent的左子树还是右子树
  34. if (parent->_key > key)
  35. {
  36. parent->_left = cur;
  37. }
  38. else if (parent->_key < key)
  39. {
  40. parent->_right = cur;
  41. }
  42. return true;
  43. }

🎄删除

🌕针对被删除节点分4种情况:
1.空树
2.叶子结点(比如删除9)
3.只有一个孩子(比如删除8)
4.有两个孩子(比如删除1)

对于第一种情况,即空树,返回false即可;

对于第二种情况,直接删除叶子结点;

对于第三种情况,若要删除的结点只有左孩子结点,则删除该结点且使被删除节点的双亲结点指向被删除节点的左孩子结点,若要删除的结点只有右孩子结点,则删除该结点且使被删除节点的双亲结点指向被删除结点的右孩子结点。
对于第四种情况,采用替代法删除结点,即找到该节点左子树中key值最大的结点或者找到其右子树key值最小的结点,把该节点的值赋给要删除的结点,然后将这个替代的结点删除即可。

再接着进一步分析,这里的第二、三种情况可以合并处理,因为叶子节点的左右子树都为空,即使让父节点指向这两个空节点,也没有任何问题。

🌕删除操作的实现(非递归):

  1. bool erase(const K& key)
  2. {
  3. Node* cur = _root;
  4. Node* parent = cur;
  5. while (cur)
  6. {
  7. if (key > cur->_key)
  8. {
  9. parent = cur;
  10. cur = cur->_right;
  11. }
  12. else if (key < cur->_key)
  13. {
  14. parent = cur;
  15. cur = cur->_left;
  16. }
  17. else
  18. {
  19. //二、三种情况合并处理,如果当前结点只有一个子树,则让父节点指向他的子树
  20. //处理只有右子树时
  21. if (cur->_left == nullptr)
  22. {
  23. //如果当前节点为根节点,则让右子树成为新的根节点
  24. if (cur == _root)
  25. {
  26. _root = cur->_right;
  27. }
  28. else
  29. {
  30. //判断当前节点是他父节点的哪一个子树
  31. if (parent->_right == cur)
  32. {
  33. parent->_right = cur->_right;
  34. }
  35. else
  36. {
  37. parent->_left = cur->_right;
  38. }
  39. }
  40. delete cur;
  41. }
  42. //处理只有左子树时
  43. else if (cur->_right == nullptr)
  44. {
  45. //如果当前节点为根节点,则让左子树成为新的根节点
  46. if (cur == _root)
  47. {
  48. _root = cur->_left;
  49. }
  50. else
  51. {
  52. if (parent->_right == cur)
  53. {
  54. parent->_right = cur->_left;
  55. }
  56. else
  57. {
  58. parent->_left = cur->_left;
  59. }
  60. }
  61. delete cur;
  62. }
  63. //处理左右子树都不为空时,选取左子树的最右节点或者右子树的最左节点
  64. else
  65. {
  66. //这里选取的是左子树的最右节点
  67. Node* LeftMax = cur->_left;
  68. Node* LeftMaxParent = cur;
  69. //找到左子树的最右节点
  70. while (LeftMax->_right)
  71. {
  72. LeftMaxParent = LeftMax;
  73. LeftMax = LeftMax->_right;
  74. }
  75. //替换节点
  76. std::swap(cur->_kv, LeftMax->_kv);
  77. //判断当前节点是他父节点的哪一个子树, 因为已经是最右子树了,所以这个节点的右子树为空,但是左子树可能还有数据,所以让父节点指向他的左子树
  78. //并且删除最右节点
  79. if (LeftMax == LeftMaxParent->_left)
  80. {
  81. LeftMaxParent->_left = LeftMax->_left;
  82. }
  83. else
  84. {
  85. LeftMaxParent->_right = LeftMax->_left;
  86. }
  87. delete LeftMax;
  88. }
  89. return true;
  90. }
  91. }
  92. return false;
  93. }

🌕删除操作的实现(递归):

  1. void DeleteNodeR(Node*& root, const K& val)
  2. {
  3. //删除前先递归找到该节点
  4. if (root == nullptr)
  5. return;
  6. if (root->_val < val)
  7. {
  8. DeleteNodeR(root->_right, val);
  9. }
  10. else if (root->_val>val)
  11. {
  12. DeleteNodeR(root->_left, val);
  13. }
  14. else
  15. {
  16. Node* del = root;
  17. if (root->_left == nullptr)
  18. {
  19. //情况1/2/3
  20. //传引用的妙用
  21. root = root->_right;
  22. }
  23. else if (root->_right == nullptr)
  24. {
  25. //情况1/2/3
  26. root = root->_left;
  27. }
  28. else
  29. {
  30. //情况4
  31. Node* cur = root->_right;
  32. while (cur->_left)
  33. {
  34. cur = cur->_left;
  35. }
  36. //swap(cur->_val, root->_val); -->破坏了树的结构
  37. root->_val = cur->_val;
  38. DeleteNodeR(root->_right, cur->_val);
  39. return;
  40. }
  41. delete del;
  42. }
  43. }
  44. void DeleteNodeR(const K& val)
  45. {
  46. DeleteNodeR(_root, val);
  47. }

🌕递归与非递归对比:

在情况4当中,找到了替换节点,实际上一步就可以把他删除( 迭代写法一步就删除了),但是采用递归的话又需要重新在子树当中去找到新的删除的节点,所以实际上没有迭代的写法优。并且递归自带弊端:栈帧的开销,栈溢出的危险。

相关文章