leetcode 724 寻找数组的中心索引 python3

 

思路:用切片分割数组,再用sum求和比较。

代码:

class Solution:
    def pivotIndex(self, nums: List[int]) -> int:
        i = 0
        while i < len(nums):
            if sum(nums[:i]) == sum(nums[i+1:]):
                return i
            else :
                i += 1
        return -1

体会:

评论区有人指出用sum暴力求和会超时。虽然这个暴力方法没有超时但是每次计算都切片两次还是造成了时间的浪费。从这方面入手有更好的思路:切片左边的和*2+当前指向的数==总和。

猜你喜欢

转载自www.cnblogs.com/hinsbugdeveloper/p/12526265.html