将二叉搜索树变平衡（letcode）

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 之间

代码：

 import java.util.*;
 // 中序遍历 二分法实现
class Solution {
    List<TreeNode> midOrder = new ArrayList<>();
    public TreeNode balanceBST(TreeNode root) {
        // 中序遍历
        midSortedOrder(root);
        // 二分法
        return dichotomy(0,midOrder.size() - 1);
    }
    void midSortedOrder(TreeNode root){
        if(root == null) return;
        midSortedOrder(root.left);
        midOrder.add(root);
        midSortedOrder(root.right);
    }
    private TreeNode dichotomy(int start, int end){
        if(start > end) return null;
        int mid = (start + end) / 2;
        TreeNode root = midOrder.get(mid);
        root.left = dichotomy(start, mid - 1);
        root.right = dichotomy(mid + 1, end);
        return root;
    }
}