LeetCode 700. Search in a Binary Search Tree

LeetCode 700. Search in a Binary Search Tree (二叉搜索树中的搜索)

题目

链接

https://leetcode.cn/problems/search-in-a-binary-search-tree/

问题描述

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

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

示例

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

提示

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

思路

直接借助性质来解答，如果相同就返回，不然分开向左右，如果结点为空就是null。

复杂度分析

时间复杂度 O(n)
空间复杂度 O(n)

代码

Java

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