leetcode-1382：将二叉搜索树变平衡

最新推荐文章于 2023-06-21 12:50:48 发布

原创最新推荐文章于 2023-06-21 12:50:48 发布 · 194 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #算法

leetcode 专栏收录该内容

612 篇文章

订阅专栏

该博客介绍了如何通过中序遍历转换数组并重建平衡二叉搜索树的方法，解决LeetCode 1382题。具体步骤包括遍历原始二叉搜索树获取节点值数组，然后根据数组重建平衡的二叉搜索树。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

leetcode-1382：将二叉搜索树变平衡

题目
解题
- 方法一：中序遍历转数组+重建平衡二叉树

题目

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

如果一棵二叉搜索树中，每个节点的两棵子树高度差不超过 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]

解题

方法一：中序遍历转数组+重建平衡二叉树

class Solution {
public:
    vector<int> arr;
    void traversal(TreeNode* cur){
        if(!cur) return;
        traversal(cur->left);
        arr.push_back(cur->val);
        traversal(cur->right);
    }

    TreeNode* getTree(int l,int r){
        if(l>r) return nullptr;
        int mid=(l+r)/2;
        TreeNode* cur=new TreeNode(arr[mid]);
        cur->left=getTree(l,mid-1);
        cur->right=getTree(mid+1,r);
        return cur;
    }


    TreeNode* balanceBST(TreeNode* root) {
        traversal(root);
        return getTree(0,arr.size()-1);
    }
};