1382. 将二叉搜索树变平衡 中等 先递归打出中序序列,再根据中序递归构建平衡二叉搜索树

  1. 将二叉搜索树变平衡
    给你一棵二叉搜索树,请你返回一棵 平衡后 的二叉搜索树,新生成的树应该与原来的树有着相同的节点值。如果有多种构造方法,请你返回任意一种。

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

示例 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] 也是一个可行的构造方案。
示例 2:

输入: root = [2,1,3]
输出: [2,1,3]

和108题一样,只是多了一步构造中序序列

没见过这种题目的是挺难想到的,见过就简单了。。

/**
 * 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 balanceBST(TreeNode root) {
        int num[]=zon(root);
        return build(num,0,num.length-1);
    }
    int [] zon(TreeNode root){
        if(root==null)return new int[0];
        int []l=zon(root.left);
        int []r=zon(root.right);
        int []ans=new int [l.length+r.length+1];
        int p=0;
        for(int x:l)ans[p++]=x;
        ans[p++]=root.val;
        for(int x:r)ans[p++]=x;
        return ans;
    }

    TreeNode build(int []num,int l,int r){
        if(l>r)return null;
        int mid=l+r>>1;
        TreeNode root=new TreeNode(num[mid]);
        root.left=build(num,l,mid-1);
        root.right=build(num,mid+1,r);
        return root;
    }
}
posted @ 2022-11-17 23:01  林动  阅读(3)  评论(0编辑  收藏  举报