python 寻找数组的中心索引_yiduobo的每日leetcode 724.寻找数组的中心索引

python 寻找数组的中心索引_yiduobo的每日leetcode 724.寻找数组的中心索引_第1张图片

祖传的手艺不想丢了,所以按顺序写一个leetcode的题解。计划每日两题,争取不卡题吧。

724.寻找数组的中心索引

力扣​leetcode-cn.com

比较基础的题目,首先计算出数组中元素的总和,然后从头开始往后遍历同时进行累加,若当前的累加和乘以2等于元素总和减去当前元素值,那么当前下标即为答案,否则将当前元素累加进累加和。

最后附上python代码:

class Solution(object):
    def pivotIndex(self, nums):
        """
        :type nums: List[int]
        :rtype: int
        """

        if not nums:
            return -1

        s = sum(nums)
        now = 0
        for index, num in enumerate(nums):
            if s - num == now + now:
                return index
            now += num
        
        return -1

你可能感兴趣的:(python,寻找数组的中心索引)