【LeetCode】将有序数组转换为二叉搜索树

题目描述

这道题目对应于 LeetCode 的 108. 将有序数组转换为二叉搜索树。

题目的大概描述是输入一个已经升序排序的数组，将这个数组转换为平衡的二叉搜索树。

在一个二叉搜索树中，父结点的左儿子的值一定比父结点小，而右结点的值比父结点大。解决这道题的手段是通过递归，每次首先选取数组的中间结点，以中间结点作为根结点，递归地建立左子树和右子树。假设中间结点的下标为 mid，左边界是 left，右边界是 right，有mid = left + right >> 1;。此时根据nums[mid]的值new一个根结点，再递归地对[left, mid - 1]和[mid + 1, right]这两个部分进行建树，即可完成将有序数组转为二叉搜索树。

解决方案

代码以 leetcode 的 C++ 风格给出：

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