代码随想录day20 Java版

669. 修剪二叉搜索树

由于递归函数有返回值，对于不在范围内的节点，可以通过左右孩子递归来实现修剪。

接下来要将下一层处理完左子树的结果赋给root->left，处理完右子树的结果赋给root->right。

最后返回root节点。

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

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

吸取了之前构造二叉树的经验，递归函数需要利用开始索引和结束索引

注意这里二分采用左闭右开的写法，因此调用递归函数右边是nums.length

class Solution {
    public TreeNode sortedArrayToBST(int[] nums) {
        return BST(nums, 0, nums.length);
    }
    TreeNode BST(int[] nums, int left, int right) {
        if (left >= right) return null;
        if (right-left == 1) return new TreeNode(nums[left]);
        TreeNode root = new TreeNode(nums[(right+left)/2]);
        root.left = BST(nums,left, (right+left)/2);
        root.right= BST(nums,(right+left)/2+1,right);
        return root;
    }
}

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

直接暴力做了，先求到总和，再按照中序遍历构建

class Solution {
    int sum = 0;
    public TreeNode convertBST(TreeNode root) {
        getSum(root);
        addTree(root);
        return root;
    }
    void getSum(TreeNode root) {
        if (root == null) return;
        sum += root.val;
        getSum(root.left);
        getSum(root.right);
    }
    void addTree(TreeNode root) {
        if (root == null) return ;
        addTree(root.left);
        int temp = root.val;
        root.val = sum;
        sum -= temp;
        addTree(root.right);
    }
}

看题解发现按照右中左顺序来遍历就可以了

class Solution {
    int sum;
    public TreeNode convertBST(TreeNode root) {
        sum = 0;
        convertBST1(root);
        return root;
    }
    public void convertBST1(TreeNode root) {
        if (root == null) return;
        convertBST1(root.right);
        sum += root.val;
        root.val = sum;
        convertBST1(root.left);
    }
}