1.编辑器
我使用的是win10+vscode+leetcode+python3
环境配置参见我的博客:
链接
2.第一百零八题
(1)题目
英文:
Given an array where elements are sorted in ascending order, convert it to a height balanced BST.
For this problem, a height-balanced binary tree is defined as a binary tree in which the depth of the two subtrees of every node never differ by more than 1.
中文:
将一个按照升序排列的有序数组,转换为一棵高度平衡二叉搜索树。
本题中,一个高度平衡二叉树是指一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/convert-sorted-array-to-binary-search-tree
(2)解法
使用递归,二分,关键点就是,数组是有序的,所以用二分可以满足节点大于左子树任意节点,小于右子树任意节点。
(耗时:56ms,内存:15.7M)
class Solution:
def sortedArrayToBST(self, nums: List[int]) -> TreeNode:
if nums:
m = len(nums) // 2
r = TreeNode(nums[m])
r.left, r.right = map(self.sortedArrayToBST, [nums[:m], nums[m+1:]])
return r
注意:
1.使用map要使用iterable变量哦,比如list[[1], [2,3]]这样的。
2.如果向sortedArrayToBST函数输入一个None或者空的list[],会return None。
3.若nums=[1],则nums[:0]=[]空,nums[1:]=[]空,而不是None;并且nums[:1]=[1]而不是1。