力扣题目——700. 二叉搜索树中的搜索

注：本文的实现思路主要是基于JS（JavaScript），涉及到的一些函数都是JS中的自带函数

题目描述

给定二叉搜索树（BST）的根节点和一个值。 你需要在BST中找到节点值等于给定值的节点。 返回以该节点为根的子树。 如果节点不存在，则返回 NULL。

---

输入示例

给定二叉搜索树:

        4
       / \
      2   7
     / \
    1   3

和值: 2

你应该返回如下子树:

      2     
     / \   
    1   3

在上述示例中，如果要找的值是 5，但因为没有节点值为 5，我们应该返回 NULL。

---

解题思路

1. 暴力破解法

由于题目中所给的是二叉树，所以直接遍历二叉树中的每一个节点即可，代码如下，思路说明在每一行中的注释中。这种暴力波破解的方法也能够顺利通过所有测试用例。

var searchBST = function(root, val) {
	// 当前节点为叶子节点，或找到了给定值的节点
    if(root === null || root.val === val) return root
    // 当前节点有子节点，且当前节点不是目标节点
    else{
    	// 查找左子树
        let leftNode = searchBST(root.left, val)
        // 查找右子树
        let rightNode = searchBST(root.right, val)
        // 如果左子树返回的不是null， 则说明找到了目标节点， 返回
        if(leftNode) return leftNode
        // 左子树中没有找到目标节点，尝试从右子树中查找
        else return rightNode
    }
};

2. 剪枝法

但是题目中的二叉树是二叉搜索树，这个条件我们还没使用上。首先说明一下什么是二叉搜索树（BST），又称二叉查找树、二叉排序树。定义是：
根节点的值大于其左子树中根节点的值，小于其右子树中根节点的值。

根据这个性质，我们能够对上述算法进行优化，上述算法中变量所有的左右子树。但是根据BST的特点，我们能够根据根节点的数值与目标值的大小比较，判断到底遍历左右子树的哪一个。

var searchBST = function(root, val) {
	// 当前节点为叶子节点，或找到了给定值的节点，则返回root
    if(root === null || root.val === val) 
        return root
    // 判断，若目标值小于根节点的值，则遍历左子树，否则遍历右子树
    else
        return val < root.val ? searchBST(root.left,val)  : searchBST(root.right,val)
};

3. 迭代法

前两种方法都是递归使用函数的方式，其实也可以使用迭代的方法实现，也比较容易理解。

var searchBST = function(root, val) {
	// 给定的节点为空树，则直接返回
    if(root === null) return root
    // 迭代
    while(root){
    	// 找到目标子树， 返回该子树
        if(val === root.val) return root
        // 若目标值小于根节点的值，则遍历左子树
        else if(val < root.val) root = root.left
        // 若目标值大于根节点的值，则遍历右子树
        else root = root.right
    }
    // 没找到目标子树， 返回空
    return null
}