724.尋找數組的中心索引

問題描述

給定一個整數類型的數組 nums，請編寫一個可以返回數組「中心索引」的方法。

咱們是這樣定義數組中心索引的：數組中心索引的左側全部元素相加的和等於右側全部元素相加的和。

若是數組不存在中心索引，那麼咱們應該返回 -1。若是數組有多箇中心索引，那麼咱們應該返回最靠近左邊的那一個。

示例 1:

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

示例 2:

輸入: 
nums = [1, 2, 3]
輸出: -1
解釋: 
數組中不存在知足此條件的中心索引。

說明:

• nums 的長度範圍爲 [0, 10000]。
• 任何一個 nums[i] 將會是一個範圍在 [-1000, 1000]的整數。

解決方案

• 建立兩個變量，分別保存當前索引左邊之和，以及右邊之和（left_sum ,right_sum）
• 開始遍歷數組
• right_sum——減去當前遍歷數值
• 判斷right_sum、left_sum 是否相等，是，返回當前索引
• left_sum ——加上當前遍歷數值

時間複雜度：O(n)

show me the code

class Solution(object):
    def pivotIndex(self, nums):
        """
        :type nums: List[int]
        :rtype: int
        """
        left_sum ,right_sum = 0,sum(nums)
        for index, val in enumerate(nums):
            right_sum -= val
            if left_sum == right_sum:
                return index
            left_sum += val
        return -1