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

news2024/10/9 5:21:58

669. 修剪二叉搜索树

题目

参考文章

思路：这题其实就是删除不符合上下边界的节点。注意：这里删除不符合上下边界节点时，这个不符合上下边界的节点的左或右子树可能存在符合上下边界的节点，所i有每次比较完之后，要继续遍历其左或右子树，直到把所有不符合上下边界的节点都删除为止

代码：

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) {//当遍历到当前数组的的下标位置相差1时，表示已经在数组边界，所以直接构建节点返回即可
            return new TreeNode(nums[left]);
        }
        int mid = left + (right - left) / 2;
        TreeNode root = new TreeNode(nums[mid]);
        root.left = sortedArrayToBST(nums, left, mid);
        root.right = sortedArrayToBST(nums, mid + 1, right);
        return root;
    }
}

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

题目

参考文章

思路：这题目的意思就是让我们从这个二叉搜索树从大到小遍历，原来左中右的情况是从小到大遍历，所以从大到小遍历就是右中左。了解这个这题目就很好解决了。这里设置一个int sum，用于存储累加值，而且每次累加后，当前记得的值就更新为sum（题目要求），按右中左去遍历即可

代码：

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);
    }
}