Python 数组切片是否存在越界

Python 数组切片是否存在越界

今天在刷leetecode时注意到这个问题。
Python 数组切片是否存在越界_第1张图片

# Definition for a binary tree node.
class TreeNode:
     def __init__(self, x):
         self.val = x
         self.left = None
         self.right = None

是一道关于最小高度树的简单题,解题思路很清晰,从root开始顺左右子树递归,不断返回中位数作为新的root直到数组为空。

左右子树的有序数列是通过切片原数组得到的,在最初的实现中,为防止数组越界而后置了条件判断语句:

class Solution:
    def sortedArrayToBST(self, nums: List[int]) -> TreeNode:
        if not len(nums): 
            return 
        mid = len(nums)//2
        root = TreeNode(nums[mid]) 
        root.left = self.sortedArrayToBST(nums[:mid]) if mid > 0 else None
        root.right = self.sortedArrayToBST(nums[mid+1:]) if mid < len(nums) else None
        return root

后来发现python中对数组进行切片时,例如:
len(list) = 5,sub_list = list[4:8],返回的sub_list中实际只包含了list[4],而不是发生数组越界错误。若 sub_list[10:],则返回的是一个空list (注意,不是None)。

故重新简化代码:

class Solution:
    def sortedArrayToBST(self, nums: List[int]) -> TreeNode:
        if not len(nums): 
            return 
        mid = len(nums)//2
        root = TreeNode(nums[mid]) 
        root.left = self.sortedArrayToBST(nums[:mid]) 
        root.right = self.sortedArrayToBST(nums[mid+1:])
        return root

你可能感兴趣的:(Python杂记,python,数据结构)