力扣刷题记录-700.二叉树中的搜索

700.二叉树中的搜索

一、题目描述

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

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

例如：

示例2：

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

提示:

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

二、解题思路

因为这是一颗二叉搜索树，利用搜索树的特性我们可以更加快速地查找目标，而不需要完全遍历整棵树。

利用 递归 或者 迭代 都可以很容易地解此题。

1.递归法

2.迭代法

三、AC代码

1.递归法

class Solution {

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

}

2.迭代法

class Solution {

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

}

参考代码随想录：https://programmercarl.com/