题目描述
给定一个有序整数数组,元素各不相同且按升序排列,编写一个算法,创建一棵高度最小的二叉搜索树。
示例:
给定有序数组: [-10,-3,0,5,9],
一个可能的答案是:[0,-3,9,-10,null,5],它可以表示下面这个高度平衡二叉搜索树:
0
/ \
-3 9
/ /
-10 5
算法思路
创建二叉搜索树,要求高度最小,则是二叉平衡搜索树。
对给定的有序数组,根即是数组中位数。然后分而治之,拆分为左子树和右子树。
递归的终点是数组长度为1或0,为1时return TreeNode(nums[0])
,为0时return
class Solution:
def sortedArrayToBST(self, nums) -> TreeNode:
def helper(ls):
if not ls:return
if len(ls)==1:
return TreeNode(ls[0])
mid=len(ls)//2
r=TreeNode(ls[mid])
r.left=helper(ls[:mid])
r.right=helper(ls[mid+1:])
return r
return helper(nums)
执行用时 :64 ms, 在所有 Python3 提交中击败了49.38%的用户
内存消耗 :15.9 MB, 在所有 Python3 提交中击败了100.00%的用户
将辅助函数简化删掉:
class Solution:
def sortedArrayToBST(self, ls) -> TreeNode:
if not ls:return
if len(ls)==1:return TreeNode(ls[0])
mid=len(ls)//2
r=TreeNode(ls[mid])
r.left=self.sortedArrayToBST(ls[:mid])
r.right=self.sortedArrayToBST(ls[mid+1:])
return r
执行用时 :48 ms, 在所有 Python3 提交中击败了97.50%的用户
将传递数组改为传递左右标:
class Solution:
def sortedArrayToBST(self, ls) -> TreeNode:
b,e=0,len(ls)
def helper(b,e):
if b<e:
mid=(b+e)//2
r=TreeNode(ls[mid])
r.left=helper(b,mid)
r.right=helper(mid+1,e)
return r
return helper(b,e)
执行用时 :68 ms, 在所有 Python3 提交中击败了41.29%的用户
并没有什么用就是了。