二叉搜索树(Binary Search Tree,BST)就是为了实现快速,高效的查找,插入,删除数据而设计的。高效的原因依赖于本身的特殊结构。

特点

  • 左右子树也分别是二叉搜索树
  • 左子树的所有节点 key 值都小于它的根节点的 key 值
  • 右子树的所有节点 key 值都大于他的根节点的 key 值
  • 二叉搜索树可以为一棵空树
  • 一般来说,树中的每个节点的 key 值都不相等,但根据需要也可以将相同的 key 值插入树中

操作

插入

  • 如果为空树则将插入节点作为根节点
  • 如果不为空树则从根节点开始,比较插入节点与根节点的 key 值,值相同则不做任何处理直接返回,大于则继续比较右子节点 R,小于则继续比较左子节点 L。
  • 右子节点 R 与插入节点比较,插入节点的 key 值大的话则继续往 R 节点的右子节点比较,小于的话则继续往R节点左子节点比较。
  • 以此类推不断往下寻找,直到找到左子节点指针或右子节点指针为空的节点,将插入节点放进去。

步骤:

  1. 创建 D 节点并与根节点比较
  2. D 小于 E,于是往左子节点继续比较
  3. D 大于 C,应该往右子节点方向,而此时 C 节点的右子节点指针为空,D 节点可以放置进去
  4. 同样的,对于 H 节点,先创建 H 节点并与根节点比较
  5. H 大于 E,于是往右子节点继续比较
  6. H 大于 G,应该往右子节点方向,而此时 G 节点的右子节点指针为空,H 节点可以放置进去

    查询

  • 从根节点开始,比较查询节点与根节点的 key 值,值相同则表示找到该节点,直接返回,大于则继续往右子节点R查找,小于则继续往左子节点L查找。
  • 右子节点R与查询节点比较,查询节点的 key 值大的话则继续往R节点的右子节点查找,小于的话则继续往R节点左子节点查找。
  • 以此类推不断往下寻找,直到找到节点的 key 值与查询节点的相同,则表示查找成功,如果最终找不到则说明不存在该节点。

步骤:

  1. B 与根节点的 key 值比较
  2. B 小于 E,往左子节点继续寻找
  3. B 小于 C,往左子节点继续寻找
  4. B 大于 A,往右子节点继续寻找,两者相等,找到
  5. 继续查询 key 值为 G 的节点,与根节点比较
  6. G 大于 E,往右子节点,两者相等,找到

    删除

    删除操作分三种情况进行:
  • 如果删除的节点为叶子节点,即它的左子节点指针和右子节点指针都为空时,则可以直接删掉该节点,并不会影响整棵树的结构。
  • 如果删除的节点只有一个子节点(左子节点或右子节点),则直接将子节点提升到被删除的节点位置。
  • 如果删除的节点有两个子节点,此时需要找到删除节点的中序后继或中序前驱来填补删除节点,中序后继其实就是所有大于删除节点中最小的那个,而中序前驱就是所有小于删除节点中最大的那个,因为二叉搜索树经过中序遍历后是一个递增序列,所以后继就是删除节点的后面那个节点,大于且大得最少的那个,比如1 2 3 4 5中4就是3的后继。前驱就是删除节点前面那个节点,比如1 2 3 4 5中2就是3的前驱。

参考资料


Javascript代码初始化二叉树:

  1. class Node {
  2. constructor(data) {
  3. this.data = data
  4. this.left = null
  5. this.right = null
  6. }
  7. }
  8. class BST {
  9. constructor() {
  10. this.root = null
  11. }
  12. // 插入
  13. insert(data) {
  14. }
  15. // 查找
  16. search(data) {
  17. }
  18. // 查找最大值
  19. findMax() {
  20. }
  21. // 查找最小值
  22. findMin() {
  23. }
  24. // 删除最小值
  25. removeMin() {
  26. if (this.root) this.root = this._removeMin(this.root)
  27. }
  28. // 删除最大值
  29. removeMax() {
  30. if (this.root) this.root = this._removeMax(this.root)
  31. }
  32. // 删除指定节点
  33. remove(data) {
  34. }
  35. }

插入:

从根节点开始,依次比较要插入的数据和节点的大小关系。 如果要插入的数据比节点的数据大,并且节点的右子树为空,就将新数据直接插到右子节点的位 置;如果不为空,就再递归遍历右子树,查找插入位置。同理,如果要插入的数据比节点数值 小,并且节点的左子树为空,就将新数据插入到左子节点的位置;如果不为空,就再递归遍历左 子树,查找插入位置。

  1. // 插入递归版本
  2. insert(data) {
  3. if (!this.root) return (this.root = new Node(data))
  4. function _insert(data, node) {
  5. if (data < node.data) {
  6. if (!node.left) node.left = new Node(data)
  7. else _insert(data, node.left)
  8. } else {
  9. if (!node.right) node.right = new Node(data)
  10. else _insert(data, node.right)
  11. }
  12. }
  13. _insert(data, this.root)
  14. }
  15. // 插入非递归版本
  16. insertWhile(data) {
  17. if (!this.root) return (this.root = new Node(data))
  18. // 从根节点开始比较
  19. let node = this.root
  20. while (true) {
  21. // 大于往右侧
  22. if (data > node.data) {
  23. if (!node.right) {
  24. node.right = new Node(data)
  25. break
  26. }
  27. node = node.right
  28. }
  29. // 小于往左侧
  30. if (data < node.data) {
  31. if (!node.left) {
  32. node.left = new Node(data)
  33. break
  34. }
  35. node = node.left
  36. }
  37. }
  38. }

查找:

先取根节点,如果它等于我们要查找的数 据,那就返回。如果要查找的数据比根节点的值小,那就在左子树中递归查找;如果要查找的数 据比根节点的值大,那就在右子树中递归查找。

  1. // 查找:递归版本
  2. search(data, node = this.root) {
  3. if (!node) return null
  4. if (node.data === data) return node
  5. if (data > node.data) return this.search(data, node.right)
  6. else return this.search(data, node.left)
  7. }
  8. //查找:非递归版本
  9. searchWhile(data, node = this.root) {
  10. while (node && node.data !== data) {
  11. if (data > node.data) {
  12. node = node.right
  13. } else if (data < node.data) {
  14. node = node.left
  15. }
  16. }
  17. return node
  18. }

查找搜索二叉树中的最大值、最小值:

因为在删除指定节点的时候需要用到,所以这里把_findMax,_findMin方法作为私有方法标记下。

对于以任意节点为根的二叉树,最大值永远是右子树的最右子节点。最小值永远是左子树的最左子节点。

递归代码实现:

  1. // 查找最大值
  2. findMax() {
  3. if (this.root) return this._findMax(this.root)
  4. }
  5. // 查找以某个节点为根的树的最大值
  6. _findMax(node) {
  7. if (!node.right) return node
  8. return this._findMax(node.right)
  9. }
  10. // 查找最小值
  11. findMin() {
  12. if (this.root) return this._findMin(this.root)
  13. }
  14. // 查找以某个节点为根的树的最小值
  15. _findMin(node) {
  16. if (!node.left) return node
  17. return this._findMin(node.left)
  18. }

非递归代码实现:

  1. // 查找最大值
  2. findMax() {
  3. let node = this.root
  4. while(node){
  5. if(!node.right) break
  6. node = node.right
  7. }
  8. return node
  9. }
  10. // 查找最小值
  11. findMin() {
  12. let node = this.root
  13. while(node){
  14. if(!node.left) break
  15. node = node.left
  16. }
  17. return node
  18. }

删除最大、最小值:

递归代码实现:

递归的思路是:每传入一个节点,删除以该节点为根的二叉树的最小值,并返回删除后的新的二叉搜索树的根

  1. // 删除最小值
  2. removeMin() {
  3. if (this.root) this.root = this._removeMin(this.root)
  4. }
  5. // _removeMin函数是删除以node为根的二叉搜索树的最小节点
  6. // 返回删除node最小节点后新的二分搜索树的根
  7. _removeMin(node) {
  8. // 如果没有左子节点,直接返回右子节点,为null一样适用
  9. if (!node.left) return node.right
  10. node.left = this._removeMin(node.left)
  11. return node
  12. }
  13. // 删除最大值
  14. removeMax() {
  15. if (this.root) this.root = this._removeMax(this.root)
  16. }
  17. _removeMax(node) {
  18. if (!node.right) return node.left
  19. node.right = this._removeMax(node.right)
  20. return node
  21. }

非递归代码实现:

要删除最小节点,关键是找到要删除节点的父节点,然后把父节点的left(最小值)或right(最大值)指向最大值的左子节点或最小值的右子节点。

  1. removeMin() {
  2. // 关键是找到要删除的最小值的父节点,然后让父节点 p.left = null
  3. if(!this.root) return null
  4. if (!this.root.left) return this.root = this.root.right
  5. let node = this.root,
  6. p = this.root
  7. while (node) {
  8. if (!node.left) break;
  9. // p缓存父节点
  10. p = node
  11. node = node.left
  12. }
  13. p.left = node.right
  14. }
  15. removeMax() {
  16. if(!this.root) return null
  17. if(!this.root.right) return this.root = this.root.left
  18. let node = this.root,p = this.root
  19. while(node){
  20. if(!node.right) break;
  21. // p缓存父节点
  22. p = node
  23. node = node.right
  24. }
  25. p.right = node.left
  26. }

删除:

二叉查找树的查找、插入操作都比较简单,但删除操作就比较复杂 。针对要删除 节点的子节点个数的不同,需分三种情况来处理。

第一种情况是,如果要删除的节点没有子节点,我们只需要直接将父节点中,指向要删除节点的 指针置为 null。比如图中的删除节点 55。

第二种情况是,如果要删除的节点只有一个子节点(只有左子节点或者右子节点),我们只需要 更新父节点中,指向要删除节点的指针(可能是指向左子节点的指针,也可能是指向右子节点的指针),让它指向要删除节点的子节点就可以了。比如图中的删 除节点 13。

第三种情况是,如果要删除的节点有两个子节点,这就比较复杂了。我们需要找到这个节点的右 子树中的最小节点,把它替换到要删除的节点上。然后再删除掉这个最小节点,因为最小节点肯 定没有左子节点(如果有左子结点,那就不是最小节点了),所以,我们可以应用上面两条规则 来删除这个最小节点。比如图中的删除节点 18。

二分搜索树 - 图1

实际上,关于二叉查找树的删除操作,还有个非常简单、取巧的方法,就是单纯将要删除的节点 标记为“已删除”,但是并不真正从树中将这个节点去掉。这样原本删除的节点还需要存储在内存 中,比较浪费内存空间,但是删除操作就变得简单了很多。而且,这种处理方法也并没有增加插 入、查找操作代码实现的难度。

删除代码递归实现:

思路类似删除最大最小值,函数_remove接受一个节点node,返回值是删除指定节点后返回的新的以node为根的二叉树

  1. // 删除指定节点
  2. remove(data) {
  3. if (this.root) this.root = this._remove(this.root, data)
  4. }
  5. _remove(node, data) {
  6. if (!node) return null
  7. // 小于node节点
  8. if (data < node.data) {
  9. node.left = this._remove(node.left)
  10. return node
  11. } else if (data > node.data) {
  12. node.right = this._remove(node.right)
  13. return node
  14. } else {
  15. // 如果找到了指定的节点,需要根据该节点的子节点情况删除
  16. // 如果node没有子节点也会走第一个if,node.right===null同样适合
  17. if (!node.left) {
  18. return node.right
  19. } else if (!node.right) {
  20. return node.left
  21. } else {
  22. // 如果自右子节点都存在,需要找到左子树中的最大子节点(前驱节点)
  23. // 或者右子树中的最小子节点(后继节点)
  24. // 这里找右子树中的最小子节点,successorNode叫后继节点
  25. let successorNode = this._findMin(node.right)
  26. // 删除右子树中的最小节点,并将返回的新二叉树返回,赋值给后继节点的右指针
  27. successorNode.right = this._removeMin(node.right)
  28. // 后继节点的左指针指向找到的原node节点的左指针
  29. successorNode.left = node.left
  30. return successorNode
  31. }
  32. }
  33. }

非递归实现:
第一步:找到该节点,及该节点的父节点,并且记录下父节点的方向。
第二步:根据该节点的子节点情况进行删除

  1. remove(data) {
  2. // 先要找到该节点,和该节点的父节点
  3. let node = this.root,
  4. // 记录父节点
  5. p = this.root,
  6. directive = null
  7. while (node && node.data !== data) {
  8. p = node
  9. if (data < node.data) {
  10. node = node.left
  11. directive = 'left'
  12. } else {
  13. node = node.right
  14. directive = 'right'
  15. }
  16. }
  17. if (!node) return '没有该节点'
  18. // 找到node后,需要根据node子节点的不同情况处理
  19. if (!node.left) {
  20. p[directive] = node.right
  21. } else if (!node.right) {
  22. p[directive] = node.left
  23. } else {
  24. // 如果有两个子节点
  25. // 这里找左子树的最大值
  26. let maxNode = node.left,
  27. maxNodeParent = node
  28. while (maxNode) {
  29. if (!maxNode.right) break;
  30. maxNodeParent = maxNode
  31. maxNode = maxNode.right
  32. }
  33. // 说明要删除的node的左子节点就是我们要找的最大值
  34. if (maxNodeParent === node) {
  35. p[directive] = maxNode
  36. maxNode.right = node.right
  37. } else {
  38. maxNodeParent.right = null
  39. p[directive] = maxNode
  40. maxNode.right = node.right
  41. maxNode.left = node.left
  42. }
  43. }
  44. }

二叉搜索树的遍历:

前序中序后序的本质都是深度优先,只是操作的具体位置不同而已。

前序中序后序遍历的概念已经在前面有提到。需要记得两点的是:二叉搜索树的中序遍历其实就是从小到大排序的结果。后序适合一些特殊的操作,比如树的释放,因为释放都需要先释放子节点,再释放父节点,刚好是满足的。

  1. preTraversal(node = this.root) {
  2. if (node) {
  3. console.log(node.data);
  4. this.preTraversal(node.left)
  5. this.preTraversal(node.right)
  6. }
  7. }
  8. midTraversal(node=this.root) {
  9. if(node){
  10. this.midTraversal(node.left)
  11. console.log(node.data);
  12. this.midTraversal(node.right)
  13. }
  14. }
  15. postTraversal(node=this.root) {
  16. if(node){
  17. this.postTraversal(node.left)
  18. this.postTraversal(node.right)
  19. console.log(node.data);
  20. }
  21. }

二分搜索数还有其他的操作,比如求某个数的floor,ceil?

重复二叉搜索树如何处理?

上面的操作方法都是基于二叉树中不存在健值相同的情况,如果相同的话呢?这个时候我们可能就需要把这些方法都稍作调整。

思路:

  • 对于插入的话,我们可以把相同的数据都存储在一个节点上,这样的话,我们的节点上存储的就不再是简单的数字,必须是一个对象。
  • 仍然每个节点只存储一个数据,插入的时候,如果碰到相同的值,就把这个数据插入到右子树中,就是相当于把相等的值当大于这个节点的值处理。如果要查找的话,也一样,遇到相同的节点,我们要继续在右子树中查找,直到找到所有等于这个值的节点为止。删除也是。

二叉搜索树时间复杂度:

二叉搜索树的时间复杂度的分析其实跟我们在快排排序的分析很类似,都是类似的递归树的分析,如果二叉树根节点的左右子树不平衡,就会影响到时间复杂度的计算,最差情况退化为链表,时间复杂度变成O(n)。满二叉树、完全二叉树就是O(logn)。其实时间复杂度就是跟树的高度成正比。所以在实际应用中,这也是平衡二叉树会比较受欢迎的原因。

完整代码:

  1. class Node {
  2. constructor(data) {
  3. this.data = data
  4. this.left = null
  5. this.right = null
  6. }
  7. }
  8. class BST {
  9. constructor() {
  10. this.root = null
  11. }
  12. // 插入递归版本
  13. insert(data) {
  14. if (!this.root) return (this.root = new Node(data))
  15. function _insert(data, node) {
  16. if (data < node.data) {
  17. if (!node.left) node.left = new Node(data)
  18. else _insert(data, node.left)
  19. } else {
  20. if (!node.right) node.right = new Node(data)
  21. else _insert(data, node.right)
  22. }
  23. }
  24. _insert(data, this.root)
  25. }
  26. // 插入非递归版本
  27. insertWhile(data) {
  28. if (!this.root) return (this.root = new Node(data))
  29. // 从根节点开始比较
  30. let node = this.root
  31. while (true) {
  32. // 大于往右侧
  33. if (data > node.data) {
  34. if (!node.right) {
  35. node.right = new Node(data)
  36. break
  37. }
  38. node = node.right
  39. }
  40. // 小于往左侧
  41. if (data < node.data) {
  42. if (!node.left) {
  43. node.left = new Node(data)
  44. break
  45. }
  46. node = node.left
  47. }
  48. }
  49. }
  50. // 查找:递归版本
  51. search(data, node = this.root) {
  52. if (!node) return null
  53. if (node.data === data) return node
  54. if (data > node.data) return this.search(data, node.right)
  55. else return this.search(data, node.left)
  56. }
  57. //查找:非递归版本
  58. searchWhile(data, node = this.root) {
  59. while (node && node.data !== data) {
  60. if (data > node.data) {
  61. node = node.right
  62. } else if (data < node.data) {
  63. node = node.left
  64. }
  65. }
  66. return node
  67. }
  68. // 查找最大值
  69. findMax() {
  70. if (this.root) return this._findMax(this.root)
  71. }
  72. _findMax(node) {
  73. if (!node.right) return node
  74. return this._findMax(node.right)
  75. }
  76. // 查找最小值
  77. findMin() {
  78. if (this.root) return this._findMin(this.root)
  79. }
  80. _findMin(node) {
  81. if (!node.left) return node
  82. return this._findMin(node.left)
  83. }
  84. // 删除最小值
  85. removeMin() {
  86. if (this.root) this.root = this._removeMin(this.root)
  87. }
  88. // _removeMin函数是删除以node为根的二叉搜索树的最小节点
  89. // 返回删除node最小节点后新的二分搜索树的根
  90. _removeMin(node) {
  91. // 如果没有左子节点,直接返回右子节点,为null一样适用
  92. if (!node.left) return node.right
  93. node.left = this._removeMin(node.left)
  94. return node
  95. }
  96. // 删除最大值
  97. removeMax() {
  98. if (this.root) this.root = this._removeMax(this.root)
  99. }
  100. _removeMax(node) {
  101. if (!node.right) return node.left
  102. node.right = this._removeMax(node.right)
  103. return node
  104. }
  105. // 删除指定节点
  106. remove(data) {
  107. if (this.root) this.root = this._remove(this.root, data)
  108. }
  109. _remove(node, data) {
  110. if (!node) return null
  111. // 小于node节点
  112. if (data < node.data) {
  113. node.left = this._remove(node.left)
  114. return node
  115. } else if (data > node.data) {
  116. node.right = this._remove(node.right)
  117. return node
  118. } else {
  119. // 如果找到了指定的节点,需要根据该节点的子节点情况删除
  120. if (!node.left) {
  121. return node.right
  122. } else if (!node.right) {
  123. return node.left
  124. } else {
  125. // 如果自右子节点都存在,需要找到左子树中的最大子节点(前驱节点)
  126. // 或者右子树中的最小子节点(后继节点)
  127. // 这里找右子树中的最小子节点,successorNode叫后继节点
  128. let successorNode = this._findMin(node.right)
  129. // 删除右子树中的最小节点,并将返回的新二叉树返回,赋值给后继节点的右指针
  130. successorNode.right = this._removeMin(node.right)
  131. // 后继节点的左指针指向找到的原node节点的左指针
  132. successorNode.left = node.left
  133. return successorNode
  134. }
  135. }
  136. }
  137. }
  138. let bst = new BST()
  139. bst.insertWhile(34)
  140. bst.insertWhile(32)
  141. bst.insertWhile(56)
  142. bst.insertWhile(43)
  143. bst.insertWhile(51)
  144. bst.insertWhile(45)
  145. bst.insertWhile(25)
  146. bst.insertWhile(67)
  147. bst.insertWhile(40)
  148. bst.insertWhile(39)
  149. bst.insertWhile(33)
  150. console.log(bst)
  151. console.log(bst.search(10))
  152. console.log(bst.findMin())
  153. console.log(bst.findMax())
  154. // 删除最大、最小值
  155. bst.removeMin()
  156. bst.removeMax()
  157. console.log(bst)
  158. // 删除指定节点
  159. bst.remove(56)
  160. console.log(bst)