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

669. 修剪二叉搜索树

1. 学习链接：
2. 思路：所剪切的区间，相当于在二叉搜索树上切了两刀，左面一刀 + 右面一刀，只保留中间部分。
3. 代码：

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    public TreeNode trimBST(TreeNode root, int low, int high) {
        //因为不需要在终止条件上进行修改，因此遇空返回就行。
        if (root == null) return root;
        
        //寻找在区间内的节点
        if (root.val < low) return trimBST(root.right, low, high);
        //寻找区间内的节点
        if (root.val > high) return trimBST(root.left, low, high);
        
        //将右子树中符合条件的节点返回给root.right;
        root.right = trimBST(root.right, low, high);
        //将左子树中符合条件的节点返回给root.left;
        root.left = trimBST(root.left, low, high);
        
        return root;
    }
}

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

1. 学习文章链接：
2. 思路：所构建的二叉搜索树为平衡的（更有意义）。将数组分割成左右区间去构造左右子树。
3. 代码：

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    public TreeNode sortedArrayToBST(int[] nums) {
        if (nums == null) return null;
        return travaersal(nums, 0, nums.length);
    }
    public TreeNode travaersal(int[] nums, int start, int end) {
        //区间内无元素，则返回null
        if (end - start < 1) return null;
        //区间内只有一个元素，以该元素创建树节点，并返回。
        if (end - start == 1) return new TreeNode(nums[start]);

        //构建平衡的二叉搜索树更有意义。定义根节点在数组中的中间位置。
        int rootIndex = start + (end - start) / 2;
        TreeNode root = new TreeNode(nums[rootIndex]);

        //接受构建好的左右子树。
        //注意这里使用的区间不变量为左闭右开。
        root.left = travaersal(nums, start, rootIndex);
        root.right = travaersal(nums, rootIndex + 1, end);
        return root;
    }
}

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

1. 学习链接：
2. 思路：在右中左这种反中序遍历的顺序基础上，定义一个sum，sum用来累加所遍历过的数值，并赋值给当前节点。
3. 代码

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    int sum;
    public TreeNode convertBST(TreeNode root) {
        sum = 0;
        traversal(root);
        return root;
    }
    public void traversal(TreeNode root) {
        if (root == null) return;
        traversal(root.right);
        //累加所遍历节点值
        sum += root.val; 
        //并赋值给当前节点
        root.val = sum;
        traversal(root.left);
    }
}