LeetCode——700. 二叉搜索树中的搜索(Java)

题目描述

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

        4
       / \
      2   7
     / \
    1   3

和值: 2
你应该返回如下子树:

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

题解思路

返回以指定节点的二叉树,当然我们只需要返回该节点即可,根据二叉搜索树的性质

节点的左孩子永远比它小,右孩子节点永远比它大,所以我们可以递归寻找目标节点

正确代码

    // 递归
    public TreeNode searchBST(TreeNode root, int val) {
        if (root == null) {
            return null;
        }
        if (val == root.val) {
            return root;
        }
        return searchBST(val < root.val ? root.left : root.right, val);
    }

    // 迭代
    public TreeNode searchBST01(TreeNode root, int val) {
        while (root != null) {
            if (val == root.val) {
                return root;
            }
            root = val < root.val ? root.left : root.right;
        }
        return null;
    }

总结

就像剪枝思想一样,递归写法总是会产生多余的计算消耗,在斐波那契数组的经典问题中最高效的解法就是记录每次计算结果

如果文章存在问题或者有更好的题解,欢迎在评论区斧正和评论,各自努力,最高处见
posted @ 2021-11-26 08:35  21岁还不是架构师  阅读(31)  评论(0编辑  收藏  举报