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

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

669. 修剪二叉搜索树

题目链接

问题简述：将二叉搜索树中结点值在一定范围内的保留下来，其余的进行删除，保证删除后仍为二叉搜索树。

思考：感觉和二叉树删除很像很像，但还是有点困难，主要是递归该返回什么掌握不好。

算法思路：先序遍历，如果当前结点小于low，当前结点的右子树可能仍存在区间范围内的结点，所以递归处理结点的右子树，后返回上一层；同理如果当前结点大于high，当前结点的左子树可能仍存在区间范围内的结点，所以递归处理结点的左子树，后返回上一层；如果当前节点满足区间范围，则处理左右子树，并连接到当前节点的左右孩子。

class Solution {
    public TreeNode trimBST(TreeNode root, int low, int high) {
        if (root == null) return null;
        //如果当前结点小于low，则将当前结点的右子树递归后返回上一层
        if (root.val < low){
            return trimBST(root.right, low, high);
        }
        //如果当前结点大于high，则将当前结点的左子树删除后返回上一层
        if (root.val > high){
            return trimBST(root.left, low, high);
        }
        //如果当前结点满足条件，连接得到的新的左子树和右子树
        root.left = trimBST(root.left, low, high);
        root.right = trimBST(root.right, low, high);
        //root满足条件则返回了root
        return root;
    }
}

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

题目链接

问题简述：将给定的有序数组转换为二叉搜索树，且为二叉平衡树。

思考：感觉起来不难，实际上也简单，二分法注意下区间是左闭右开还是左闭右闭。

算法思路：每次取左闭右闭的范围，创建新的结点，结点值为闭区间的中间元素。再依次递归，将中间元素左侧和右侧作为新的范围，得到的结点作为当前节点的左右结点。

class Solution {
    public TreeNode sortedArrayToBST(int[] nums) {
        return traversal(0, nums.length - 1, nums);
    }
    //双指针指向剩余的元素范围
    public TreeNode traversal(int left, int right, int nums[]){
        //创建当前结点
        TreeNode root = null;
        //如果存在元素
        if (left <= right){
            //为当前root赋值
            root = new TreeNode(nums[(left + right)/2]);
            //递归出左右子树
            root.left = traversal(left, (left + right)/2 - 1, nums);
            root.right = traversal((left + right)/2 + 1, right, nums);
        }
        return root;
    }
}

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

题目链接

问题简述：将二叉搜索树的值修改为累加树的值。累加树元素值是将元素从大到小，依次加和。

思考：用到了右中左遍历。

算法思路：定义一个sum值，进行右中左的遍历，将每个元素值改为当前值加上sum，最后返回root。

class Solution {
    int sum = 0;
    public TreeNode convertBST(TreeNode root) {
        if (root == null) return null;
        //右中左遍历
        convertBST(root.right);
        //更改当前结点和sum的值
        root.val += sum;
        sum = root.val;
        convertBST(root.left);
        return root;
    }
}

感想

不敢想