二叉搜索树如何在 JavaScript 中实现？

前端

2023-10-27 11:54:04

深入浅出，解密二叉搜索树在 JavaScript 中的实现

二叉搜索树，一种高效的数据结构，在计算机科学领域占据着举足轻重的地位。凭借其独特的特性，它不仅在查找、插入和删除操作上表现优异，而且广泛应用于各种复杂算法的实现。

JavaScript，作为一门流行且用途广泛的编程语言，也为二叉搜索树的实现提供了丰富的支持。在本文中，我们将从头开始，一步步解析二叉搜索树在 JavaScript 中的实现，让您对这种数据结构的运作机制有更深入的理解。

二叉搜索树的基本概念

在探讨 JavaScript 中的实现之前，让我们先回顾一下二叉搜索树的基本概念。二叉搜索树是一种特殊的二叉树，其特点在于：

每个节点最多只能有两个子节点，即左子节点和右子节点。
左子节点上的元素值小于其父节点的值。
右子节点上的元素值大于其父节点的值。

JavaScript 中的二叉搜索树实现

基于上述基本概念，JavaScript 中的二叉搜索树可以轻松实现。首先，我们需要定义一个二叉搜索树的类，如下所示：

class BinarySearchTree {
  constructor() {
    this.root = null; // 根节点
  }

  // 插入节点
  insert(value) {
    const newNode = new Node(value); // 创建一个新的节点
    if (this.root === null) {
      this.root = newNode; // 如果根节点为空，则直接将新节点设为根节点
    } else {
      this._insertNode(newNode, this.root); // 否则，递归地插入节点
    }
  }

  // 递归插入节点
  _insertNode(newNode, currentNode) {
    if (newNode.value < currentNode.value) { // 如果新节点的值小于当前节点的值
      if (currentNode.left === null) { // 如果当前节点的左子节点为空
        currentNode.left = newNode; // 将新节点设为当前节点的左子节点
      } else {
        this._insertNode(newNode, currentNode.left); // 否则，递归地插入节点到左子树
      }
    } else { // 如果新节点的值大于或等于当前节点的值
      if (currentNode.right === null) { // 如果当前节点的右子节点为空
        currentNode.right = newNode; // 将新节点设为当前节点的右子节点
      } else {
        this._insertNode(newNode, currentNode.right); // 否则，递归地插入节点到右子树
      }
    }
  }

  // 查找节点
  search(value) {
    return this._searchNode(value, this.root); // 调用递归查找方法
  }

  // 递归查找节点
  _searchNode(value, currentNode) {
    if (currentNode === null) { // 如果当前节点为空，则表示未找到该值
      return false;
    } else if (value === currentNode.value) { // 如果找到该值
      return true;
    } else if (value < currentNode.value) { // 如果要查找的值小于当前节点的值
      return this._searchNode(value, currentNode.left); // 则递归地查找左子树
    } else { // 如果要查找的值大于当前节点的值
      return this._searchNode(value, currentNode.right); // 则递归地查找右子树
    }
  }

  // 删除节点
  delete(value) {
    this.root = this._deleteNode(value, this.root); // 调用递归删除方法
  }

  // 递归删除节点
  _deleteNode(value, currentNode) {
    if (currentNode === null) { // 如果当前节点为空，则表示未找到该值
      return null;
    } else if (value === currentNode.value) { // 如果找到该值
      if (currentNode.left === null && currentNode.right === null) { // 如果该节点是叶子节点
        return null; // 直接删除该节点
      } else if (currentNode.left === null) { // 如果该节点只有右子节点
        return currentNode.right; // 将该节点的右子节点设为其父节点的子节点
      } else if (currentNode.right === null) { // 如果该节点只有左子节点
        return currentNode.left; // 将该节点的左子节点设为其父节点的子节点
      } else { // 如果该节点有两个子节点
        const successor = this._findSuccessor(currentNode); // 找到该节点的后继节点
        currentNode.value = successor.value; // 将后继节点的值赋值给该节点
        currentNode.right = this._deleteNode(successor.value, currentNode.right); // 删除后继节点
        return currentNode; // 返回该节点
      }
    } else if (value < currentNode.value) { // 如果要删除的值小于当前节点的值
      currentNode.left = this._deleteNode(value, currentNode.left); // 则递归地删除左子树中的该值
      return currentNode; // 返回该节点
    } else { // 如果要删除的值大于当前节点的值
      currentNode.right = this._deleteNode(value, currentNode.right); // 则递归地删除右子树中的该值
      return currentNode; // 返回该节点
    }
  }

  // 找到后继节点
  _findSuccessor(node) {
    let currentNode = node.right; // 从该节点的右子节点开始
    while (currentNode.left !== null) { // 一直向左走，直到找到最左边的节点
      currentNode = currentNode.left;
    }
    return currentNode; // 返回最左边的节点
  }

  // 打印二叉搜索树
  print() {
    this._printTree(this.root); // 调用递归打印方法
  }

  // 递归打印二叉搜索树
  _printTree(node) {
    if (node === null) { // 如果当前节点为空，则返回
      return;
    }
    console.log(node.value); // 打印当前节点的值
    this._printTree(node.left); // 递归地打印左子树
    this._printTree(node.right); // 递归地打印右子树
  }
}