跳转至

108.将有序数组转换为二叉搜索树 (Easy)*

题目描述*

将一个按照升序排列的有序数组,转换为一棵高度平衡二叉搜索树。

本题中,一个高度平衡二叉树是指一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1。

示例*

给定有序数组: [-10,-3,0,5,9],

一个可能的答案是:[0,-3,9,-10,null,5],它可以表示下面这个高度平衡二叉搜索树:

      0
     / \
   -3   9
   /   /
 -10  5

代码*

二叉搜索树的中序遍历就是递增数组,因此可以每次选取数组中间点作为根结点递归建树。

class Solution {
public:
    TreeNode *build(vector<int>& nums, int start, int end) {
        if(start > end) {
            return nullptr;
        }
        int mid = start + (end - start) / 2;
        TreeNode *root = new TreeNode(nums[mid]);
        root->left = build(nums, start, mid - 1);
        root->right = build(nums, mid + 1, end);
        return root;
    }
    TreeNode* sortedArrayToBST(vector<int>& nums) {
        if(nums.size() == 0) {
            return nullptr;
        }
        return build(nums, 0, nums.size() - 1);
    }
};

最后更新: July 23, 2022