[LeetCode] 724. Find Pivot Index

https://www.pexels.com/ko-kr/photo/4164418/

 

• pivot 인덱스란 인텍스의 모든 왼쪽 요소의 합과 오른쪽의 모든 요소의 합이 같은 지점의 인덱스

class Solution {
    func pivotIndex(_ nums: [Int]) -> Int {
        var rightSum = nums.reduce(0, +) - nums[0]
        var leftSum = 0

        if rightSum == leftSum {
            return 0
        }

        for index in 1..<nums.count {

            leftSum += nums[index - 1]
            rightSum -= nums[index]

            if leftSum == rightSum {
                return index // pivot index
            }
        }

        return -1
    }
}

var rightSum = nums.reduce(0, +) - nums[0]

• 어레이 전체 합을 구함. 처음 피봇 인덱스는 어레이의 0번째 인덱스 -> 피봇 인덱스 기준 왼쪽 합, 오른 쪽 합을 구해야 하므로 전체 하에서 현재 인덱스 값을 빼줌

        if rightSum == leftSum {
            return 0
        }

• 첫 번째 인덱스에서 왼쪽 합, 오른 쪽 합이 같은 경우 첫 번째 인덱스가 피봇 인덱스 이므로 0을 리턴함 (인덱스 0이 피봇 인덱스)
• 전체 합 S에서 현재 피봇 인덱스 값 nums[0] 뺀 값인 rightSum = S - nums[0] 과 leftSum (초기 값 0)을 비교

        for index in 1..<nums.count {

            leftSum += nums[index - 1]
            rightSum -= nums[index]

            if leftSum == rightSum {
                return index // pivot index
            }
        }

• 0 번째 인덱스를 비교 했으므로 1 부터 전체 갯수 - 1 만큼 for문을 돌려 반복함
• 이전 피봇 값을 leftSum에 더해 가면 피봇 인덱스 기준 왼쪽 어레이 합을 구할 수 있음
• rightSum은 현재 피봇 인덱스 값을 빼주면 구할 수 있음
• 만약 leftSum 과 rightSum이 같다면 피봇 인덱스 이므로 이를 리턴

return -1

• 만약 피봇 인덱스가 없다면 -1을 리턴하면 됨

class Solution {
    func pivotIndex(_ nums: [Int]) -> Int {
        var rightSum = nums.reduce(0, +)
        var leftSum = 0

        for index in 0..<nums.count {
            rightSum -= nums[index]
            
            if leftSum == rightSum {
                return index
            }
            
            leftSum += nums[index]
        }

        return -1
    }
}

• for 문을 0 부터 시작하게 하고 싶다면 간단하게 위와 같이 할 수 있음

 

 

참고 사이트 및 도서

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