1382. 将二叉搜索树变平衡_normal

1382. 将二叉搜索树变平衡

难度中等

给你一棵二叉搜索树，请你返回一棵 平衡后 的二叉搜索树，新生成的树应该与原来的树有着相同的节点值。

如果一棵二叉搜索树中，每个节点的两棵子树高度差不超过 1 ，我们就称这棵二叉搜索树是 平衡的 。

如果有多种构造方法，请你返回任意一种。

 

示例：

输入：root = [1,null,2,null,3,null,4,null,null]
输出：[2,1,3,null,null,null,4]
解释：这不是唯一的正确答案，[3,1,4,null,2,null,null] 也是一个可行的构造方案。

 

提示：

• 树节点的数目在 1 到 10^4 之间。
• 树节点的值互不相同，且在 1 到 10^5 之间。

---

解法1：

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */

/*
解法1: 分成两大步

1. 先将原来的二叉搜索树 中序遍历, 变成 升序数组


2. 再根据二分法, 以数组mid为根,左边数组构建左子树,右边数组构建右子树
   变成了第108题:将有序数组转换为二叉搜索树
   二叉搜索树的中序遍历是升序序列，

给定二叉搜索树的中序遍历，是否可以唯一地确定二叉搜索树？
答案是否定的。

如果没有要求二叉搜索树的高度平衡，
则任何一个数字都可以作为二叉搜索树的根节点，
因此可能的二叉搜索树有多个

直观地看，我们可以选择 中间数字 作为二叉搜索树的 根节点，
这样分给左右子树的数字个数相同或只相差 1，可以使得树保持平衡。

如果数组长度是奇数，则根节点的选择是唯一的，
如果数组长度是偶数，则可以选择 中间位置左边的数字 或 中间位置右边的数字 作为根节点，
选择不同的数字作为根节点 则创建的平衡二叉搜索树也是不同的。

确定平衡二叉搜索树的根节点之后，
其余的数字分别位于平衡二叉搜索树的 左子树 和 右子树中，

左子树 和 右子树 分别也是 平衡二叉搜索树，
因此可以通过 递归的方式 创建 平衡二叉搜索树。

递归的基准情形是: 平衡二叉搜索树不包含任何数字，此时平衡二叉搜索树为空。

在给定中序遍历序列数组的情况下，每一个子树中的数字在数组中一定是连续的，
因此可以通过 数组下标范围 确定子树包含的数字，下标范围记为[left,right]。

对于整个中序遍历序列，下标范围从 left = 0 到 right = nums.length−1。
当  left > right 时，平衡二叉搜索树为空。

以下三种方法中，
方法一总是选择中间位置左边的数字作为根节点，
方法二总是选择中间位置右边的数字作为根节点，
方法三是方法一和方法二的结合，选择任意一个中间位置数字作为根节点。

方法一：中序遍历，总是选择中间位置左边的数字作为根节点
选择中间位置左边的数字作为根节点，
则根节点的下标为 mid = (left + right) / 2，此处的除法为整数除法。
*/ 
class Solution {
    public TreeNode balanceBST(TreeNode root) {
        // 第1步, 中序遍历 构建 升序数组
        List<Integer> numArr = new ArrayList<>();
        inOrder(root, numArr);

        // 第2步, 构建
        // 闭区间[left, right]代表整个数组
        int leftIndex = 0;
        int rightIndex = numArr.size() - 1;
        return buildTree(numArr, leftIndex, rightIndex);
    }

    private void inOrder(TreeNode root, List<Integer> numArr) {
        if (root != null) {
            // 标准中序遍历三步曲
            inOrder(root.left, numArr);
            numArr.add(root.val);
            inOrder(root.right, numArr);
        }
    }

    private TreeNode buildTree(List<Integer> numArr, int leftIndex, int rightIndex) {
        // base case 递归结束条件
        if (leftIndex > rightIndex) {
            return null;
        }

        // 总是选择中间位置偏左 作为根, 将数组一分为二,去构建树 (整数除法)
        int middleIndex = (leftIndex + rightIndex) / 2;

        TreeNode rootNode = new TreeNode(numArr.get(middleIndex));

        // 构建左子树
        rootNode.left = buildTree(numArr, leftIndex, middleIndex - 1);
        // 构建右子树
        rootNode.right = buildTree(numArr, middleIndex + 1, rightIndex);

        // 返回构造后的子树
        return rootNode;
    }
}