Search

Find Pivot Index

문제 설명 : 현재 인덱스 이전의 합과 현재 인덱스 이후 합이 동일한 피벗 인덱스를 반환
풀이 방법
전체 합을 미리 구한 후, 재 순회 하며 누적 값을 사용하였다.
시간복잡도 : O(N)O(N)
성공 코드
class Solution: def pivotIndex(self, nums: List[int]) -> int: all_sum = sum(nums) current_sum = 0 for i, num in enumerate(nums): if all_sum - num - current_sum == current_sum: return i current_sum += num return -1
Python
복사