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

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

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

示例:

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

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

      0
     / \
   -3   9
   /   /
 -10  5

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
class Solution(object):
def sortedArrayToBST(self, nums):
"""
:type nums: List[int]
:rtype: TreeNode
"""
if not nums:
return None
if len(nums) == 1:
return TreeNode(nums[0])
mid = len(nums)//2
left = nums[:mid]
right = nums[mid+1:]
root = TreeNode(nums[mid])
root.left = self.sortedArrayToBST(left)
root.right = self.sortedArrayToBST(right)
return root