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

1. 题目

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

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

示例:

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

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

      0
     / \
   -3   9
   /   /
 -10  5

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

2. 解题

• 二叉搜索树要平衡，那么左右子树节点数量接近相等最好
• 每次取数组中间点为root
• 递归划分数组即可

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

python3 解答

class Solution:
    def sortedArrayToBST(self, nums: List[int]) -> TreeNode:
        def build(l,r):
            if l > r:
                return None
            mid = (l+r)>>1
            root = TreeNode(nums[mid])
            root.left = build(l,mid-1)
            root.right = build(mid+1,r)
            return root
        return build(0,len(nums)-1)