【二叉搜索树】将二叉搜索树变平衡-力扣 1382 题

news2024/11/23 23:38:17

💝💝💝欢迎来到我的博客，很高兴能够在这里和您见面！希望您在这里可以感受到一份轻松愉快的氛围，不仅可以获得有趣的内容和知识，也可以畅所欲言、分享您的想法和见解。

推荐:kuan 的首页,持续学习,不断总结,共同进步,活到老学到老
导航
檀越剑指大厂系列:全面总结 java 核心技术点,如集合,jvm,并发编程 redis,kafka,Spring,微服务,Netty 等
常用开发工具系列:罗列常用的开发工具,如 IDEA,Mac,Alfred,electerm,Git,typora,apifox 等
数据库系列:详细总结了常用数据库 mysql 技术点,以及工作中遇到的 mysql 问题等
懒人运维系列:总结好用的命令,解放双手不香吗?能用一个命令完成绝不用两个操作
数据结构与算法系列:总结数据结构和算法,不同类型针对性训练,提升编程思维,剑指大厂

非常期待和您一起在这个小小的网络世界里共同探索、学习和成长。💝💝💝 ✨✨ 欢迎订阅本专栏 ✨✨

博客目录

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

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

public class E09Leetcode1382 {
    List<Integer> temp = new ArrayList<>();

    /**
     * 前序遍历
     *
     * @param root
     */
    private void travesal(TreeNode root) {
        if (root == null) {
            return;
        }
        travesal(root.left);
        temp.add(root.val);//存储中序遍历的结果
        travesal(root.right);
    }

    private TreeNode toTree(int left, int right) {
        if (left > right) {
            return null;
        }
        //获取中间index
        int mid = (left + right) >> 1;
        //中间index的值就是根节点
        TreeNode node = new TreeNode(temp.get(mid));
        node.left = toTree(left, mid - 1);
        node.right = toTree(mid + 1, right);
        return node;
    }

    public TreeNode balanceBST(TreeNode root) {
        travesal(root);
        return toTree(0, temp.size() - 1);
    }
}