[LeetCode] Convert Sorted Array to Binary Search Tree

最新推荐文章于 2022-02-19 22:56:38 发布

原创最新推荐文章于 2022-02-19 22:56:38 发布 · 3.5k 阅读

0 ·

CC 4.0 BY-SA版权

LeetCode1 专栏收录该内容

120 篇文章

订阅专栏

本文介绍了一种将已排序的数组转换为高度平衡的二叉搜索树(BST)的方法。通过对数组中间元素的选择作为根节点，并递归地构建左右子树来实现平衡。这种方法能够有效地构造出高度平衡的BST。

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

[Problem]
Given an array where elements are sorted in ascending order, convert it to a height balanced BST.

[Analysis]
注意：题目中的数组是已经排好序的，如果没有排序，则需要用BST的左旋、右旋、左右旋、右左旋等进行调整。

对于已经排好序的数组，构造BST时，使用数组的中间元素array[mid]作为BST的根节点，array[0:mid-1]用于构造其左子树，array[mid+1:end]用于构造其右子树，这两步可以递归进行。

[Solution]

// Definition for binary tree
// struct TreeNode {
//     int val;
//     TreeNode *left;
//     TreeNode *right;
//     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
// };

class Solution {
public:
    // build tree
    TreeNode *build(vector &num, int begin, int end){
        if(begin > end){
            return NULL;
        }
   
        // build left brunch and right brunch
        int mid = (begin+end)/2;
        TreeNode* left = build(num, begin, mid-1);
        TreeNode* right = build(num, mid+1, end);
   
        // build tree
        TreeNode* node = new TreeNode(num[mid]);
        node->left = left;
        node->right = right;
        return node;
    }
   
    // sorted array to BST
    TreeNode *sortedArrayToBST(vector &num) {
        // Start typing your C/C++ solution below
        // DO NOT write int main() function
        if(num.size() == 0){
            return NULL;
        }
        else{
            return build(num, 0, num.size()-1);
        }
    }
};