LeetCode - 724. Find Pivot Index [Kotlin]

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

Find Pivot Index - LeetCode

 

정답 (Solution)

class Solution {
    fun pivotIndex(nums: IntArray): Int {
        val totalSum = nums.sum()
        var leftSum = 0
        for(i in 0 until nums.size){
            if(leftSum == (totalSum -leftSum - nums[i])){
                return i
            }
            leftSum += nums[i]
        }
        return -1
    }
}

 

분석

• 전체 합계를 현재 i 번째를 제외한 왼쪽 합계와 같은지 계속 비교하여 같은 경우 해당 i 를 리턴한다