二叉搜索树中的搜索,leetcode习题

二叉搜索树中的搜索

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

例如，

给定二叉搜索树:

    4
   / \
  2   7
 / \
1   3

和值: 2
你应该返回如下子树:

  2     
 / \   
1   3

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

//迭代法
class Solution {
    public TreeNode searchBST(TreeNode root, int val) {

        while (root != null) {

            if (val<root.val) root = root.left;
            else if (val>root.val) root = root.right;
            else return root;

        }
        return root;

    }
}
思路：
利用二叉搜索树的这个特性，因为二叉搜索树本来就具有有序性
左子树比右子树大，然后进行比较，如果比目标值小那么对左边进行遍历搜索，如果比目标值大那么对右边进行搜索。直到搜到为止

//递归
public TreeNode searchBST1(TreeNode root, int val){
        if (root == null || root.val == val){
            return root;
        }

        if (val < root.val){
            //左边进行遍历
            return searchBST1(root.left,val);
        }
        else{
            //右部分遍历
            return searchBST1(root.right,val);
        }

    }