BST 700_701

700、在BST中找到一个子树

给一个根结点和一个数val，在BST中找到等于val的结点，然后返回以这个结点为根结点的子树。

可以用递归或迭代，差不多，这里用递归，从根结点搜索等于val的结点，找到时返回该结点。这里主要返回值可能会写错：

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

701、在BST插入一个结点

给一个BST插入值为val的结点。

迭代比较容易想到，这里用递归，关键点就是将创建的新结点插入为上个递归中的子结点

其实从大的方向看就是三步：

1. 创建一个新结点
2. 如果要插入的val比当前结点小，就交给子任务做（子任务就是把它插入到左子树中），并且调用递归返回的根结点插入到当前结点的左子树中，如果val比当前结点大，反过来
3. 返回根结点

public TreeNode insertIntoBST(TreeNode root, int val) {
    if (root == null)
        return new TreeNode(val);

    if (val < root.val)
        root.left = insertIntoBST(root.left, val);
    else
        root.right = insertIntoBST(root.right, val);

    return root;
}