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

669. 修剪二叉搜索树

完成

代码

class Solution {
    public TreeNode trimBST(TreeNode root, int low, int high) {
        if (root == null) {
            return null;
        }
        if (root.val < low) {
            return trimBST(root.right, low, high);
        }
        if (root.val > high) {
            return trimBST(root.left, low, high);
        }
        // root在[low,high]范围内
        root.left = trimBST(root.left, low, high);
        root.right = trimBST(root.right, low, high);
        return root;
    }
}

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

完成

代码

递归地挑选[left, right)区间的中点值作为根节点

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

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

完成

代码

最右节点的值不变，其余节点的值从右向左累加。
按照右中左的遍历顺序即可从大到小地遍历节点。

class Solution {
    int sum;
    public TreeNode convertBST(TreeNode root) {
        sum = 0;
        convert(root);
        return root;
    }

    // 按右中左顺序遍历
    public void convert(TreeNode root) {
        if (root == null) {
            return;
        }
        convert(root.right);
        sum += root.val;
        root.val = sum;
        convert(root.left);
    }
}

总结