LeetCode——修剪二叉搜索树

Q：给定一个二叉搜索树，同时给定最小边界L 和最大边界 R。通过修剪二叉搜索树，使得所有节点的值在[L, R]中 (R>=L) 。你可能需要改变树的根节点，所以结果应当返回修剪好的二叉搜索树的新的根节点。

A：
思路分析：

• 修剪一棵树，如果根结点的值小于给定的左边界L，那么当前结点及其左子树就会被修剪掉，修剪后的树应该是其右子树，但是右子树不一定是符合范围的树，所以要对其右子树叶进行修剪，然后返回修剪后的右子树。
• 同理，根结点的值大于给定的右边界R，修剪后的树应该是其左子树且要对左子树修剪。
• 涉及到改变树的结构，就需要更新链接，如果当前结点值在范围内，那么修建其左右子树，并且更新左右链接。最后将当前修剪好的子树返回。

时间复杂度为O(n)， 空间复杂度与树高成正比。

代码：

    public TreeNode trimBST(TreeNode root, int L, int R) {
        if (root == null)
            return null;
        if (root.val < L)
            return trimBST(root.right, L, R);
        if (root.val > R)
            return trimBST(root.left, L, R);
        root.left = trimBST(root.left, L, R);
        root.right = trimBST(root.right, L, R);
        return root;
    }