700.二叉搜索树中的搜索

题目

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

例如，

给定二叉搜索树:

        4
       / \
      2   7
     / \
    1   3

和值: 2

你应该返回如下子树:

      2     
     / \   
    1   3

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

分析

可以使用正常遍历来找，但是效率较低．因为是二叉搜索树，所以可以使用中序遍历的思路来防止不必要的遍历．

代码

第一种：直接遍历

 /**
     * 正常遍历,效率非常低
     *
     * @param root
     * @param val
     * @return
     */
    public TreeNode searchBST(TreeNode root, int val) {
        TreeNode node = search(root, val);
        return node;
    }

    private TreeNode search(TreeNode node, int val) {
        if (node == null) {
            return null;
        }

        if (node.val == val) {
            return node;
        } else {
            TreeNode left = search(node.left, val);
            TreeNode right = search(node.right, val);

            if (left != null) {
                return left;
            }
            if (right != null) {
                return right;
            }
            return null;
        }
    }

第二种方法：使用中序遍历的思想

    /**
     * 因为是二叉搜索树,可以使用中序遍历的思路，减掉不必要的遍历
     *
     * @param root
     * @param val
     * @return
     */
    public TreeNode searchBST２(TreeNode root, int val) {
        TreeNode node = search(root, val);
        return node;
    }

    private TreeNode search２(TreeNode node, int val) {
        if (node == null) {
            return null;
        }

        if (node.val < val) {
            TreeNode right = search(node.right, val);
            return right;
        } else if (node.val > val) {
            TreeNode left = search(node.left, val);
            return left;
        } else {
            return node;
        }
    }