题目描述:
将一个按照升序排列的有序数组,转换为一棵高度平衡二叉搜索树。
本题中,一个高度平衡二叉树是指一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1。
示例:
给定有序数组: [-10,-3,0,5,9],
一个可能的答案是:[0,-3,9,-10,null,5],它可以表示下面这个高度平衡二叉搜索树:
0
/ \
-3 9
/ /
-10 5
个人思路:
emmm,还是递归吧!遵循一个原则:在递归主体中只处理root,剩下来的交给左右节点。这个题目要求“高度平衡”,实际上只要每次将该有序数组“平分”,即可保证这一点。因此将该“平分点”作为root,小于root的数组作为左子树,大于root的数组作为右子树即可。
具体代码如下:
class Solution:
def sortedArrayToBST(self, nums: List[int]) -> TreeNode:
if not nums:
return None
def helper(nums):
mid = len(nums) // 2 # 确保“平衡”
root = TreeNode(nums[mid])
root.left = helper(nums[:mid])
root.right = helper(nums[mid + 1:])
return root
return helper(nums)