Leetcode - Tree - Convert Sorted Array to Binary Search Tree

最新推荐文章于 2021-11-02 12:48:41 发布

原创最新推荐文章于 2021-11-02 12:48:41 发布 · 382 阅读

0 ·

CC 4.0 BY-SA版权

Leetcode 专栏收录该内容

30 篇文章

订阅专栏

本文介绍了一种将已排序的数组转换为高度平衡的二叉搜索树的方法。通过递归方式选取数组中间元素作为根节点，并将左右两侧的子数组分别构造为左子树和右子树。

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

/**
 * Definition for binary tree
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    TreeNode *sortedArrayToBST(vector<int> &num) {
        if(num.empty())
            return NULL;
        int mid=(num.size()-1)/2;
        TreeNode *root=new TreeNode(num[mid]);
        vector<int> leftArray(num.begin(),num.begin()+mid);
        vector<int> rightArray(num.begin()+mid+1,num.end());
        if(mid>0)
            root->left=sortedArrayToBST(leftArray);
        root->right=sortedArrayToBST(rightArray);
        return root;
    }
};