724. 尋找數(shù)組的中心索引

給你一個整數(shù)數(shù)組 nums，請編寫一個能夠返回數(shù)組 **“中心索引” **的方法。

數(shù)組** 中心索引 **是數(shù)組的一個索引，其左側(cè)所有元素相加的和等于右側(cè)所有元素相加的和。

如果數(shù)組不存在中心索引，返回 -1 。如果數(shù)組有多個中心索引，應該返回最靠近左邊的那一個。

注意：中心索引可能出現(xiàn)在數(shù)組的兩端。

示例 1：
輸入：nums = [1, 7, 3, 6, 5, 6]
輸出：3
解釋：索引 3 (nums[3] = 6) 的左側(cè)數(shù)之和 (1 + 7 + 3 = 11)，與右側(cè)數(shù)之和 (5 + 6 = 11) 相等。
同時, 3 也是第一個符合要求的中心索引。

示例 2：
輸入：nums = [1, 2, 3]
輸出：-1
解釋：數(shù)組中不存在滿足此條件的中心索引。

示例 3：
輸入：nums = [2, 1, -1]
輸出：0
解釋：索引 0 左側(cè)不存在元素，視作和為 0 ；右側(cè)數(shù)之和為 1 + (-1) = 0 ，二者相等。

示例 4：
輸入：nums = [0, 0, 0, 0, 1]
輸出：4
解釋：索引 4 左側(cè)數(shù)之和為 0 ；右側(cè)不存在元素，視作和為 0 ，二者相等。

提示：

• nums 的長度范圍為 [0, 10000]。
• 任何一個 nums[i] 將會是一個范圍在 [-1000, 1000]的整數(shù)。

解題思路：

左邊逐個相加，右邊為總數(shù)減左邊減當前值

代碼：

class Solution:
    def pivotIndex(self, nums: List[int]) -> int:
        n = len(nums)
        s = sum(nums)
        left = 0
        for i in range(n):
            right = s - left - nums[i]
            if right == left:
                return i
            left += nums[i]
        return -1