algorithm notes: leetcode#724 find pivot index

Problem


Analysis


Solution


1
2
3
4
5
6
7
8
9
10
11
12
13
14
class (object):
def pivotIndex(self, nums):
"""
:type nums: List[int]
:rtype: int
"""
leftSum = 0
rightSum = sum(nums)
for idx, num in enumerate(nums):
rightSum -= num
if leftSum == rightSum:
return idx
leftSum += num
return -1

724. Find Pivot Index