代码随想录刷题day23｜修剪二叉搜索树&将有序数组转换为二叉搜索树&把二叉搜索树转换为累加树

---

day23学习内容

day23主要内容

• 修剪二叉搜索树
• 将有序数组转换为二叉搜索树
• 把二叉搜索树转换为累加树

---

声明
本文思路和文字，引用自《代码随想录》

一、修剪二叉搜索树

669.原题链接

1.1、思路

• 二叉搜索树（BST）的特性是，对于每个节点，其左子树中的所有节点都小于该节点，其右子树中的所有节点都大于该节点。
• 因此，如果一个节点的值小于给定范围的下限 low，那么该节点及其左子树都可以被舍弃；如果一个节点的值大于给定范围的上限 high，那么该节点及其右子树都可以被舍弃。
• 递归逻辑：
  • 如果 root 的值大于 high，则说明 root 及其右子树都不在范围内，应该返回修剪后的左子树：return trimBST(root.left, low, high);。
    • 为什么：对于二叉搜索树中的任意节点 root，如果 root 的值大于给定范围的上限 high，那么根据二叉搜索树的性质，root 及其右子树中的所有节点的值都大于 high，因此这些节点都不在给定范围内，应该被舍弃。
  • 如果 root 的值小于 low，则说明 root 及其左子树都不在范围内，应该返回修剪后的右子树：return trimBST(root.right, low, high);。
  • 如果 root 的值在范围内，需要同时修剪左右子树，
    • 并更新root的左子节点：root.left = trimBST(root.left, low, high);
    • 更新root的右子节点root.right = trimBST(root.right, low, high);

1.2、正确写法

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.原题链接

2.1、思路

• 因为题目要求是高度平衡的二叉搜索树，
• 所以一种思路就是
  • 在数组中选择中间元素作为树的根节点，
  • 然后递归地在左右子数组中选择中间元素作为左右子树的根节点。

2.2、正确写法-递归

class Solution {
    public TreeNode sortedArrayToBST(int[] nums) {
        return recursion(nums, 0, nums.length - 1);
    }

    private TreeNode recursion(int[] nums, int left, int right) {
        // 递归终止条件
        if (left > right) {
            return null;
        }

        // 单层递归逻辑
        int mid = left + (right - left) / 2;
        TreeNode root = new TreeNode(nums[mid]);
        root.left = recursion(nums, left, mid - 1);
        root.right = recursion(nums, mid + 1, right);
        return root;
    }
}

2.2.1、数组是偶数还是奇数会影响代码逻辑吗？

• 奇数长度的数组
  当数组的长度是奇数时，选择中间元素作为根节点后，左右子数组的长度将相等。这意味着树的左右子树将具有相同的高度，从而确保树是高度平衡的。
• 偶数长度的数组
  当数组的长度是偶数时，存在两个“中间”元素。通常，算法会选择两个中间元素中的较小者或较大者作为根节点。在这段代码中，它选择了较小的一个（通过向下取整实现）。这将导致左子树的元素少于右子树的元素，但左右子树的高度差不会超过 1，因此仍然保持了树的高度平衡。
• 哪段代码是向下取整呢
  • 这里，mid 是通过对 (right - left) / 2 的结果进行向下取整来计算的。在 Java 中，当两个整数相除时，结果会自动向下取整。因此，如果 right - left 是一个奇数，(right - left) / 2 的结果会被向下取整。
  • 例如，假设 left = 0，right = 3，则 right - left = 3。计算 mid 的值为：

int mid = 0 + (3 - 0) / 2;
在这个例子中，(3 - 0) / 2 的结果是 1.5，但由于我们在 Java 中使用整数除法，结果会被向下取整为 1。这就是通过向下取整实现的部分。
mid的最后结果是mid = 0 + 1 = 1

三、把二叉搜索树转换为累加树

538.原题链接

3.1 思路-使用反向中序遍历

• 根据题意，在累加树中，每个节点的新值是原始树中所有大于或等于该节点值的节点值之和。
• 为了实现这个目标，需要按照从大到小的顺序遍历树中的每个节点，并且在遍历的过程中，累加遇到的节点值
• 如何反向遍历？中序遍历（左-中-右）会按照从小到大的顺序访问所有节点。因此，如果我们想要按照从大到小的顺序访问所有节点，我们可以使用反向中序遍历（右-中-左）

3.2 代码-反向中序遍历

class Solution {
    TreeNode cur;
    TreeNode pre;
    int sum = 0;

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

    private void bianli(TreeNode root) {

        if (root == null) {
            return;
        }

        // 右
        convertBST(root.right);
        // 中
        sum += root.val;
        root.val = sum;
        // 左
        convertBST(root.left);
    }
}

总结

1.感想

• 题目都不是很难。
• 今天二叉树刷完了，周末还是要抽时间复习一遍，整理一下。

2.思维导图

本文思路引用自代码随想录，感谢代码随想录作者。