1382. 将二叉搜索树变平衡

luckycoding

于 2021-05-17 22:51:52 发布

阅读量93

点赞数

CC 4.0 BY-SA版权

分类专栏： leetcode

本文链接：https://blog.youkuaiyun.com/luckycoding/article/details/116953090

leetcode 专栏收录该内容

21 篇文章

订阅专栏

本文介绍了一种通过先序遍历将二叉搜索树转化为有序数组的方法，并在此基础上重建平衡二叉搜索树。该方法首先使用递归将原始树转换为有序节点数组，再通过递归方式从中选取中间元素作为根节点，左右两侧子数组分别构成新的子树。

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

通过先序遍历得到有序数组，然后重建。

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    void tree2vector(TreeNode *root, vector<TreeNode *> &tree) {
        if (root == NULL) {
            return ;
        }
        TreeNode *left = root->left;
        TreeNode *right = root->right;
        root->left = NULL;
        root->right = NULL;
        tree2vector(left, tree);
        tree.push_back(root);
        tree2vector(right, tree);
    }
    TreeNode * solve(vector<TreeNode *> &tree, int l, int r) {
        if (l == r) {
            return tree[l];
        }
        else if (l > r) {
            return NULL;
        }
        TreeNode *ans = NULL;
        int mid = (l + r) >> 1;
        ans = tree[mid];
        ans->left = solve(tree, l, mid - 1);
        ans->right = solve(tree, mid + 1, r);
        return ans;
    }

    TreeNode* balanceBST(TreeNode* root) {
        vector<TreeNode *> tree;
        tree2vector(root, tree);
        TreeNode *ans = solve(tree, 0, tree.size() -1);
        return ans;
    }
};