[JavaScript] LeetCode : Find Pivot Index

문제

Given an array of integers nums, calculate the pivot index of this array.

The pivot index is the index where the sum of all the numbers strictly to the left of the index is equal to the sum of all the numbers strictly to the index's right.

If the index is on the left edge of the array, then the left sum is 0 because there are no elements to the left. This also applies to the right edge of the array.

Return the leftmost pivot index. If no such index exists, return -1.

 

정수 배열을 매개변수로 받아 해당 배열의 피봇 인덱스를 찾아 반환하는 문제.
피봇 인덱스는 배열 내 인덱스 왼쪽에 있는 모든 숫자의 합이 인덱스 오른쪽의 모든 숫자의 합과 같은 인덱스이다.
만약 가장 왼쪽 끝에 있는 인덱스가 있다면 왼쪽 인덱스의 합은 0이 된다. (왼쪽에 다른 요소가 더 없기 때문에. 오른쪽도 동일)
가장 왼쪽에 있는 피봇 인덱스를 반환하며 인덱스가 없을 경우에는 -1을 반환한다.

 

 

풀이

var pivotIndex = function(nums) {
    let pivot = 0;
    let sum = nums.reduce((prev, curr) => {
        return prev + curr;
    });

    let leftSum = 0;
    for (let i = 0; i < nums.length; i++) {
        pivot = i;

        sum -= nums[i];
        if (sum === leftSum) return pivot;
        leftSum += nums[i];
  }
    return -1;
};

 

생각한 과정은 아래와 같다.

• 배열의 총합 구하기
• 왼쪽부터 인덱스 하나당 총합에서 뺀 값과 왼쪽 인덱스 누적값 구하기
• 총합에서 뺀 값과 왼쪽 인덱스가 일치하면 인덱스 반환하기

먼저 배열의 총합은 for문으로 구할 수도 있지만, 좀 더 코드를 직관적이고 간소화 시킬 수 있도록,

지난 번에 사용해보았던 reduce 메서드를 이용해봤다.

배열의 모든 요소를 따져봐야 하므로 하나의 for문을 통해 인덱스를 구하기로 했다.

피봇 인덱스는 i가 증가할 때마다 새롭게 지정하고 조건에 일치하면 반환하는 방식이다.

하나의 인덱스를 기준으로 총합에서 현재 인덱스 값을 뺐을 때 누적값과 같은지 비교하고,

총합이 더 크다면 누적값에 인덱스값을 더해주기로 했다.

 

 

다른 방법

LeetCode에서 본 다른 풀이 방식 중 하나이다.

var pivotIndex = function(nums) {
    let sum = nums.reduce((a,b)=>a+b, 0);
    let sumL = 0;
    for(let i=0, len=nums.length; i<len; i++){
        if(sumL === (sum-nums[i])/2) return i;
        sumL += nums[i];
    }
    return -1;
};

나와 동일하게 reduce 메서드로 총합을 구하고,
왼쪽 누적값이 총합에서 현재 인덱스값을 뺀 것을 2로 나눈 값과 일치하면 반환하는 식이다.
2로 나눈 것은 왼쪽과 오른쪽의 누적값이 동일해야 하는 부분을 차용한 것 같다.

 

 

추가로 다른 방법이 하나 더 있는데, 아래 페이지에서 볼 수 있다.

https://leetcode.com/problems/find-pivot-index/discuss/138027/javascript-solution%3A-beats-100.00-runtime-76ms

javascript solution: beats 100.00%, runtime 76ms - LeetCode Discuss

 

 

 

 

---

🤸‍♀️🤸‍♀️🤸‍♀️🤸‍♀️🤸‍♀️🤸‍♀️

방문해주셔서 감사합니다! 🙌

포스팅들은 공부중인 내용을 글로 작성한 것이라 부족한 점이 많으니 참고 부탁드립니다. 

부족한 부분에 대한 코멘트는 언제나 환영입니다.

좋은 하루 되세요, 감사합니다! 🙂