将一个按照升序排列的有序数组,转换为一棵高度平衡二叉搜索树。
本题中,一个高度平衡二叉树是指一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1。
示例:
给定有序数组: [-10,-3,0,5,9], 一个可能的答案是:[0,-3,9,-10,null,5],它可以表示下面这个高度平衡二叉搜索树: 0 / \ -3 9 / / -10 5
方法:
采用递归的形式,由于目标是实现高度平衡的二叉搜索树,因此需要先确定根节点的位置。实际上根节点的位置就是列表中间的元素,之后再分别对左半部分和右半部分进行相同的操作。
特别注意一点,博主一开始出现了一个很愚蠢的错误,导致代码一致不能AC,后来才发现是一个很基础的问题
列表的 [] != None
[] 应该利用 列表的长度进行判断或者直接用 == [] 进行判断。
最终的AC解为:
# Definition for a binary tree node.
# class TreeNode(object):
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution(object):
def sortedArrayToBST(self, nums):
"""
:type nums: List[int]
:rtype: TreeNode
"""
length = len(nums)
if length == 0:
return None
if length == 1:
return TreeNode(nums[0])
root = TreeNode(nums[length // 2])
root.left = self.sortedArrayToBST(nums[:length // 2])
root.right = self.sortedArrayToBST(nums[length // 2 + 1:])
return root
版权声明:本文为jhlovetll原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。