724. Find Pivot Index

Doyeon Kim·2022년 8월 28일

코딩테스트 공부

목록 보기
111/171

문제 링크 : https://leetcode.com/problems/find-pivot-index/


Pivot 인덱스를 찾는 문제이다.

Pivot index란 어떤 해당 인덱스를 기준으로 왼쪽의 합과 오른쪽의 합이 같은 곳의 인덱스를 말한다.

예를 들어 Ex.nus = [1,7,3,6,5,6] 이 있으면
왼쪽 = (0)/1/ 오른쪽=7,3,6,5,6 합
왼쪽= 1/7/오른쪽 = 3656
..
왼쪽 173/6/56
.
.
.

이렇게 탐색하며 답을 찾아낸다

class Solution:
    def pivotIndex(self, nums: List[int]) -> int:
        l = 0
        t = sum(nums) #12
        
        for i in range(len(nums)):
            r= t - l - nums[i]
            if r == l:
                return i
            else: 
                l += nums[i]
        return -1

갠적으로 약간의 수학적인 센스..?가 필요한 문제였다고 느꼈다.

profile
성장하고 도전하는 개발자. 프로그래밍 좋아하세요?

0개의 댓글