【算法题解】700. 二叉搜索树中的搜索 - Java递归和迭代法

1. 题目描述。LeetCode700题

给定二叉搜索树（BST）的根节点和一个值。 你需要在BST中找到节点值等于给定值的节点。 返回以该节点为根的子树。 如果节点不存在，则返回 NULL。

2. 分析

重点：利用BST的特性：左边的小于中间的，中间的小于右边的。

3. JAVA代码递归版本

    // 递归
    public TreeNode searchBST(TreeNode root, int val) {
        // base case ，空树返回空
        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);
    }

• 时间复杂度：O(logN)
• 空间复杂度：O(lonN), 最坏情况下为 O(N)。

4. JAVA代码迭代版本

    // 迭代
    public TreeNode searchBST(TreeNode root, int val) {
        // base case ，空树返回空
        if (root == null) return null;
        // 迭代，迭代的变量是root
        while (root != null) {
            // 当前节点就是查找值，返回当前节点
            if (root.val == val)  return root ;
            // 当前节点值小于查找值，去查找右子树
            else if (root.val < val) 
                root = root.right;
            // 当前节点值大于查找值，去查找左子树
            else 
                root = root.left;
        }
        return null;
    }

• 时间复杂度：O(logN)
• 空间复杂度：O(1)