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

669. 修剪二叉搜索树⭐️

• 题目链接:代码随想录

收获点：如果遇到不符合条件的值，大胆返回，因为只要设计了当前节点左右子树接住的代码，那么当前节点一定能被正确的接住并返回

• 删除结点固定思路：
  如果遍历到当前层需要删除，那么处理当前层之后可以返回上一层。也可以对当前层进行下一步的递归操作
• 解题思路：（根据区间逐层遍历）
  ①返回值和参数：返回值要设置为TreeNode，用来向上传递结点
  ②处理逻辑：
  1.当前层
  如果当前节点小于区间最大值，但是右子树还有希望，修改好右子树后返回
  如果当前节点大于区间最小值，但是左子树还有希望，修改好右子树后返回
  2.上一层逻辑
  处理从下一层修建好，归上来的节点
  ③终止条件：递归到树为空或者是有不满足区间的值

public TreeNode trimBST(TreeNode root, int low, int high) {
    //终止条件
    if(root == null){
        return null;
    }

    //处理逻辑
    //当前节点小于区间值，但是右子树还有希望，修改好右子树后返回
    //注意：这里不能就直接返回了右子树，因为有可能右子树还有符合区间的值
    if(root.val < low){
        //修剪右子树，然后返回修剪后的根节点
        TreeNode right = trimBST(root.right, low, high);
        return right;
    }

    if(root.val > high){
        //修剪左子树，然后返回修剪后的根节点
        TreeNode left = trimBST(root.left, low, high);
        return left;
    }

    //处理从下一层修建好，归上来的节点
    root.left = trimBST(root.left, low, high);
    root.right = trimBST(root.right, low, high);

    return root;
}

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

题目链接:代码随想录

建议结合前面的前序中序生成二叉树题目来看构建过程，删除节点题来看返回节点过程

• 解题思路：
  ①返回值和参数：要传入要切割的数组，通过不断改变下标来达到切割目的，而不是重复申请新数组
  ②终止条件：要切割的数组中左右边界的值不符合条件
  ③处理逻辑：
  每次根节点都是切割后的数组序列的中间值，保证了平衡性
  每次通过递归左右子树，不断改变切割的区间，达到建树的目的

public TreeNode sortedArrayToBST(int[] nums) {
    TreeNode root = traversal(nums,0,nums.length - 1);
    return root;
}


private 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 - 1);//通过划分数组完成对左树的构建
    root.right = traversal(nums, mid + 1, right);

    return root;
}

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

题目链接:代码随想录

本题难点在于看懂题…

• 解题思路：
  整体采取右 – 中 – 左的遍历
  设计一个全局变量记录累加和,然后遍历到该节点就加上该节点的值

public int sum = 0;
public TreeNode convertBST(TreeNode root) {
    //终止条件
    if(root == null){
        return null;
    }
    //右中左
    root.right = convertBST(root.right);

    sum += root.val;
    root.val = sum;
    root.left = convertBST(root.left);

    return root;
}

• 双指针法

public int pre = 0;

public TreeNode convertBST(TreeNode root) {
    traversal(root);
    return root;
}

private void traversal(TreeNode root){
    if(root == null){
        return;
    }
    //处理逻辑
    //右中左
    traversal(root.right);
    int cur = pre + root.val;
    root.val = cur;
    pre = cur;//更新pre的值
    traversal(root.left);
}