代码随想录算法训练营第 20 天 | 669. 修剪二叉搜索树、108. 将有序数组转换为二叉搜索树、538. 把二叉搜索树转换为累加树

669. 修剪二叉搜索树

题目链接

class Solution {
    public TreeNode trimBST(TreeNode root, int low, int high) {
        if (root == null) {
            return null;
        }
        if (root.val > low && root.val > high) {
            return trimBST(root.left, low, high);
        }
        if (root.val < low && root.val < high) {
            return trimBST(root.right, low, high);
        }

        // 如果该节点满足边界要求
        root.left = trimBST(root.left, low, high); // 妙，再调用递归
        root.right = trimBST(root.right, low, high);
        return root;
    }
}

---

108. 将有序数组转换为二叉搜索树

题目链接

指针法：

class Solution {
    public TreeNode sortedArrayToBST(int[] nums) {
        return traversal(nums, 0, nums.length);
    }

    // 左闭右开
    TreeNode traversal(int[] nums, int left, int right) {
        if (left >= right) {
            return null;
        }

        int mid = (left + right) / 2;
        TreeNode root = new TreeNode(nums[mid]);

        root.left = traversal(nums, left, mid);
        root.right = traversal(nums, mid + 1, right);
        return root;
    }
}

---

评论区，拷贝数组法：

class Solution {
    public TreeNode sortedArrayToBST(int[] nums) {
        //每次将中点作为根节点，其左边的所有元素作为左子树，右边的所有元素作为右子树
        if(nums.length == 0) {
            return null;
        }
        int mid = nums.length / 2;
        TreeNode root = new TreeNode(nums[mid]);
        root.left = sortedArrayToBST(Arrays.copyOfRange(nums,0,mid));
        root.right = sortedArrayToBST(Arrays.copyOfRange(nums,mid+1,nums.length));
        return root;
    }
}

---

538. 把二叉搜索树转换为累加树

题目链接

联想成数组求累加数组，如 [2, 5, 6] -> [13, 11, 6]
倒序双指针遍历相加即可。

所以对树而言，右中左遍历树即可。

class Solution {
    int pre = 0;

    public TreeNode convertBST(TreeNode root) {
        pre = 0; // 全局变量，调用方法前最好重新赋值一下
        traversal(root);
        return root;
    }

    public void traversal(TreeNode cur) {
        if (cur == null) {
            return;
        }

        // 右
        traversal(cur.right);

        // 中
        cur.val += pre;
        pre = cur.val;

        // 左
        traversal(cur.left);
    }
}

附：对于到底是用全局变量，还是局部变量放在递归函数参数中的问题，见<<重构:改善既有代码的设计>>

---