JS趣味算法学习- 实现二叉排序树

前言

我们知道dom结构也是以树的形式存在的，所以了解树的这种数据结构对于我们分析前端代码还是很重要的。

（当然这里跟前端沾边也是为了吸引大家学习的兴趣，真相其实是我就单纯的想写这一章 ...(｡•ˇ‸ˇ•｡) ...）

废话不多说，我们先从二叉排序树开始学起吧。

二叉排序树（BST Binary Search Tree）

什么是二叉排序树？

简单来说，就是每一个点只能最多有两个子节点。它有下面三个属性：

• 若左子树不空，则左子树上所有结点的值均小于它的根结点的值；
• 若右子树不空，则右子树上所有结点的值均大于或等于它的根结点的值；
• 左、右子树也分别为二叉排序树，这点很重要，希望大家记住。

对于这颗树来说，23是它的根节点。而一个父节点的两个子节点分别称为左节点跟右节点。在

二叉树中，左边的节点一定都小于23，右边节点都大于等于23。

实现BST

BST由节点组成。所以先来看看节点的实现：

function Node(data, left, right) {
  this.data = data;
  this.left = left;
  this.right = right;
  this.show = function () {
    return this.data;
  }
}

这个节点里，传入的data就是节点的数据。left，right分别为左右子节点。

现在需要定义一个二叉查找树（下面简称BST）的类，这个类有一个根节点 root，和一个插入函数insert

function BST() {
  this.root = null;
  this.insert = insert;
}

难点在于这个插入函数，先抛开这个函数是如何写的，我们先来想想BST是如何长成的。假设我需要在BST里依次插入23,45,16,37,3,99,22这些数字，那么插入的顺序如下：

1. 第一个数字为根节点，即 this.root = 23
2. 第二个数字为45，对比23，如果比23大，放在23的右边节点。此时BST如下：

3.第三个数字为16，对比23，比23小，成为23的左节点。此时BST如下：

4.第四个数字为37，先对比23，比23大，放在右边。右边再看23的右节点有值，是45，

37比45小，所以放在45的左边。

5.第五个数字为3，对比23，比23小，放左边，再看左边23的左节点16，3比16小，

所以放在16的左边。

6.第六个数字为99，对比23，比23大，看23的右节点45，比45大，放在45的右节点

7. 第七个数字为22，对比23，比23小，看23的左节点16，比16大，放在16的右节点。

8.如果最后一个数字是100，对比23，比23大，看23的右节点45，比45大，再看45的右节点99，比99大，所以放在99的右节点处。

以上，就是一颗BST生成的过程。

接下来我们用计算机的语言描述一下上述过程:（以插入右边节点为例）

1. 针对第一个节点，直接将它设置成根节点。
2. 设置一个指针parent用于追踪当前的父节点。
3. 节点跟当前的parent进行对比，如果 > parent, 看parent.right 是否有值，没有的话，成功找到当前节点n的位置，即parent.right = n，此时终止程序。
4. 如果parent.right有值，这么这个parent.right就会成为最新的parent，在重复3步骤。

下面我们用JS来实现一下上述过程：

JS代码实现

function insert(data) {
  var n = new Node(data, null, null); 
  if(!this.root) {
    this.root = n;
  } else {
    // 使用current的指针来探索子节点
    var current = this.root;
    while(true) {
      var parent = current;
      if(data < parent.data) {
        current = current.left;
        if(!current) {
          parent.left = n;
          break;
        }

      } else {
        current = current.right;
        if(!current) {
          parent.right = n;
          break;
        }
      }
    }
  }
}

这样我们这个BST的类就完成啦~~

运行上面的代码：

var nums = new BST();
nums.insert(23);
nums.insert(45);
nums.insert(16);
nums.insert(37);
nums.insert(3);
nums.insert(99);
nums.insert(100);
console.log(nums)

在将结果nums打印出来,如果你得到了这样的数据结构，那么恭喜你，你的BST生成成功啦。

( >﹏<。)～

遍历

遍历分为前序，中序，以及后序遍历，他们的命名是以根节点的访问次序划分的：

前序遍历：根节点->左子树->右子树
中序遍历：左子树->根节点->右子树
后序遍历：左子树->右子树->根节点

以我们刚刚生成的BST为例：

前序遍历：23 16 3 22 45 37 99

中序遍历：3 16 22 23 37 45 99 （这时候就会发现：卧槽，6的飞起啊，这不就是升序的排列么）

后序遍历：3 22 16 37 99 45 23

怎么理解呢？以中序遍历为例，由于中序遍历遵循：左子树->根节点->右子树

所以这颗树的排列顺序为 16（左子树） -> 23(根节点) -> 45(右子树)

/ \ / \

3 22 37 99

在看左子树，开篇的时候我们说过，左右字数都依旧是BST，所以对于左右子树，都还是遵循这个排序：左子树->根节点->右子树；

那么，对于左字树来说，排序顺序为3 -> 16 -> 22, 对于右字树，排列顺序是 37 -> 45 -> 99。

连起来就是 3 16 22 23 37 45 99。

前序跟后序同理。

下一篇，我们将了解如何使用算法完成这三种遍历，而且要从递归函数讲起，大家可以先自己思考一下~

 

本文有帮助的话，点个赞鼓励一下作者呗 ლ(＾ω＾ლ)

参考资料：

1、《数据结构与算法JS描述》

 

原文https://zhuanlan.zhihu.com/p/46920212