算法-力扣700 二叉搜索树中的搜索 Java版本

题目描述

给定二叉搜索树（BST）的根节点 root 和一个整数值 val。

你需要在 BST 中找到节点值等于 val 的节点。 返回以该节点为根的子树。 如果节点不存在，则返回 null 。

示例 1:

输入：root = [4,2,7,1,3], val = 2
输出：[2,1,3]
示例 2:

输入：root = [4,2,7,1,3], val = 5
输出：[]

提示：

树中节点数在 [1, 5000] 范围内
1 <= Node.val <= 107
root 是二叉搜索树
1 <= val <= 107

代码

class Solution {
    
    public TreeNode searchBST(TreeNode root, int val) {
    
        //二叉搜索树是比当前节点小的值就去左子树中找，比当前节点大的值就去右子树中找
        //如果最后找到空节点就表明没有这个值
        //使用递归的方法
        if (root == null) {
    
            return null;
        }

        if (val < root.val) {
    
           return searchBST(root.left, val);
        }
        if (val> root.val){
    
            return searchBST(root.right,val);
        }
        //找到值了
        return root;
    }
}