代码随想录二叉树——二叉搜索树中的搜索

题目

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

应该返回如下子树：

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

思路

二叉搜索树的特点：二叉树搜索树是一个有序树

● 若它的左子树不空，则左子树上所有结点的值均小于它的根结点的值；
● 若它的右子树不空，则右子树上所有结点的值均大于它的根结点的值；
● 它的左、右子树也分别为二叉搜索树。

1. 递归法（三部曲）
（1）确定递归函数的参数和返回值

传入的参数是根节点和要搜索的数值，返回的是这个搜索的数值所在的节点

（2）确定终止条件

如果root为空或者找到这个数值的节点了，就返回root

（3）确定单层递归的逻辑

二叉搜索树的节点是有序的，如果root.val > val，则搜索左子树；如果root.val < val，则搜索右子树，如果最后都没有搜索到，则返回null。

java代码如下：

class Solution {
    public TreeNode searchBST(TreeNode root, int val){
        if(root == null || root.val == val)
            return root;
        if(root.val > val){
            return searchBST(root.left, val);
        } else{
            return searchBST(root.right, val);
        }
    }
}

2. 迭代法

对于二叉搜索树的特点，即节点的有序性，可以不使用辅助栈或者队列就可以写出迭代法。

例如要搜索元素为3的节点，我们不需要搜索其他节点，也不需要做回溯，查找的路径已经规划好了，因为节点的有序性就帮我们确定了搜索的方向。

java代码如下：

class Solution {
	public TreeNode searchBST(TreeNode root,int val){
		while(root != null){
			if(root.val > val){
				root = root.left;
			}
			else if(root.val < val){
				root = root.right;
			}
			else {
				return root;
			}
		}
		return null;
	}
}