LeetCode/Find_Pivot_Index.py at main · techatanu/LeetCode · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
class Solution(object):
    def pivotIndex(self, nums):
        """
        :type nums: List[int]
        :rtype: int
        """
        t = sum(nums)
        l_t = 0
        for i in range(len(nums)):
            r_t = (t - l_t - nums[i])
            if r_t == l_t:
                return i
            l_t += nums[i]
        return -1