700. 二叉搜索树中的搜索

Posted on | In
Words count in article: 219 | Reading time ≈ 1

题目

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

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

示例 1:

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

示例2：

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

提示：

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

解法

解法一：

递归搜索

JAVA

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
public TreeNode searchBST(TreeNode root, int val) {
        if (null == root) {
            return root;
        }

        if (root.val == val) {
            return root;
        }

        if (root.val > val) {
            return searchBST(root.left, val);
        } else {
            return searchBST(root.right, val);
        }
    }

解法二：

迭代搜索

1
2
3
4
5
6
public TreeNode searchBST(TreeNode root, int val) {
    while (true) {
        if (root == null || root.val == val) return root;
        root = root.val > val ? root.left : root.right;
    }
}
0%