Leetcode 108

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

本题中,一个高度平衡二叉树是指一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 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版权协议,转载请附上原文出处链接和本声明。