LeetCode700. 二叉搜索树中的搜索

1 题目描述

给定二叉搜索树（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
输出：[]

来源：力扣（LeetCode）
链接：700. 二叉搜索树中的搜索

2 算法设计

由于二叉搜索树是一个有序树，根据二叉搜索树的特点：

• 若它的左子树不空，则左子树上所有结点的值均小于它的根结点的值；
• 若它的右子树不空，则右子树上所有结点的值均大于它的根结点的值；
  它的左、右子树也分别为二叉搜索树。
  因此在进行遍历时可以判断当前节点的值和目标值的关系，若大于目标值则向左子树进行递归，若小于目标值则向右子树递归。

3 代码实现

 public TreeNode searchBST(TreeNode root, int val) {
        //找到该节点或节点为空就直接返回
        if (root == null || root.val == val) {
            return root;
        }
        TreeNode resNode = null;
        //向左子树递归
        if (root.val > val) {
            resNode = searchBST(root.left,val);
        }
        //向右子树递归
       if (root.val < val) {
           resNode = searchBST(root.right,val);
       }

        return  resNode;
    }

注意，该递归函数有返回值，也就是说如果递归找到了值为val的节点将会被返回，需要一个变量去接受，否则结果就为空了。

4 测试结果