Leetcode 108 将有序数组转换为二叉搜索树

地址  https://leetcode-cn.com/problems/convert-sorted-array-to-binary-search-tree/

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

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

示例:

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

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

      0
     / 
   -3   9
   /   /
 -10  5

 

解答

平衡搜索二叉树 高度差不超过1  且左子树所有节点的值比根节点小 右子树所有节点的值比根节点大

那么我们每次取数组中间的值作为当前的根节点 左边比他小的区间作为当前节点的左子树 右边比他大的区间作为当前节点的右子树

class Solution {
public:
    TreeNode* dfs(vector<int>& nums, int l, int r)
    {
        if (l > r) return NULL;
        int mid = l + r >> 1;
        TreeNode* p = new  TreeNode(nums[mid]);
        p->left = dfs(nums, l, mid - 1);
        p->right = dfs(nums, mid + 1, r);

        return p;
    }

    TreeNode* sortedArrayToBST(vector<int>& nums) {
        //sort(nums.begin(), nums.end());
        TreeNode* root = dfs(nums, 0, nums.size() - 1);
        return root;
    }
};