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官方题解

©️2020 CSDN 皮肤主题: 大白 设计师:CSDN官方博客 返回首页