leetCode——将二叉搜索树变平衡

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

如果一棵二叉搜索树中，每个节点的两棵子树高度差不超过 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] 也是一个可行的构造方案。

class Solution {
    ArrayList<TreeNode> arr = new ArrayList<>();
    public TreeNode balanceBST(TreeNode root) {
        inorder(root);
        if(arr.size() < 3) return root;
        return buildBST(0, arr.size()-1);
    }
//根据中序遍历的结果 构建树
    public TreeNode buildBST(int left, int right){
        if(left > right) return null;

        int curIndex = (left+right)/2;
        TreeNode cur = arr.get(curIndex);
        cur.left = buildBST(left, curIndex-1);
        cur.right = buildBST(curIndex+1, right);
        return cur;
    }
//中序遍历
    public void inorder(TreeNode node){
        if(node == null){
            return;
        }
        inorder(node.left);
        arr.add(node);
        inorder(node.right);
    }
}

本题思路：将原有的二叉搜索树进行中序遍历，存放在一个Array数组中，取数组的中间点为根节点，左侧为左子树，右侧为右子树。（这样构建的二叉搜索树是平衡二叉树，具体的证明可以查看Leetcode官方题解）leetcode官方题解