LeetCode96|二叉搜索树中的搜索

例如，

给定二叉搜索树:

        4
       / 
      2   7
     / 
    1   3

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

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

public class SearchBSTTest {
    public static void main(String[] args) {
        TreeNode t1 = new TreeNode(4);
        TreeNode t2 = new TreeNode(2);
        TreeNode t3 = new TreeNode(7);
        TreeNode t4 = new TreeNode(1);
        TreeNode t5 = new TreeNode(3);
        t1.left = t2;
        t1.right = t3;
        t2.left = t4;
        t2.right = t5;
        int val = 2;
        TreeNode treeNode = searchBST(t1, val);
        System.out.println("treeNode = " + treeNode);

    }

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