当前位置：首页 > article >正文

LeetCode：700.二叉搜索树中的搜索

article 2025/3/4 22:05:27

跟着carl学算法，本系列博客仅做个人记录，建议大家都去看carl本人的博客，写的真的很好的！
代码随想录

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

递归法

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

迭代法

	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;
    }