[파이썬] 리트코드 724. Find Pivot Index 풀이

https://leetcode.com/problems/find-pivot-index/submissions/

Find Pivot Index - LeetCode

숫자 리스트가 주어졌을 떄 

어느 한 요소보다 왼쪽에 있는 숫자의 합과 오른쪽에 있는 숫자의 합이 동일해지는 요소의 인덱스를 찾아야 한다 

 

Nums = [1,7,3,6,5,6] 이라면 

인덱스가 3일 때 좌우의 합이 모두 11로 동일해지므로 3이 답이 되는 것이다 

 

1. 왼쪽에서부터의 합을 left, 오른쪽에서부터의 합을 right라고 두면 

2. left = sum(nums[0,i-1]) # 인덱스 바로 전 숫자까지의 합 

3. right = sum(nums) - left - nums[i] # 전체의 합에서 왼쪽에서부터의 합, 인덱스를 뺀 수치 

4. 일 때의 left == right가 되는 인덱스를 찾으면 된다 

class Solution:
    def pivotIndex(self, nums: List[int]) -> int:
        left = 0 
        total = sum(nums)
        for i, x in enumerate(nums):
            if left == total - left - x:
                return i 
            left += x 
        return -1