LeetCode108，将有序数组转换为二叉搜索树

题目

给你一个整数数组 nums ，其中元素已经按 升序 排列，请你将其转换为一棵 高度平衡 二叉搜索树。

高度平衡 二叉树是一棵满足「每个节点的左右两个子树的高度差的绝对值不超过 1 」的二叉树。

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/convert-sorted-array-to-binary-search-tree
题目著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

题解

思路

设数组左端点下标为l，右端点下标为r，中点为mid，每次将数组分成三份，[l, mid - 1], [mid, mid], [mid + 1, r]
nums[mid]为当前节点的值，nums[l, mid - 1]为右子树，nums[mid + 1, r] 为左子树

时间复杂度

O(n)

空间复杂度

O(1)无需额外的空间

代码

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    TreeNode* build(int l, int r, vector<int>& num) {
        if(l > r) return nullptr; // end
        int mid = (l + r) >> 1;
        TreeNode* node = new TreeNode(num[mid]);
        node -> left = build(l, mid - 1, num);
        node -> right = build(mid + 1, r, num);
        // cout << (node -> val) << endl;
        return node;
    }
    TreeNode* sortedArrayToBST(vector<int>& nums) {
        return build(0, nums.size() - 1, nums);
    }
};