프로그래머스 - 등차수열의 특정한 항만 더하기 - js (reduce 함수 초기값?)

프로그래머스 - 등차수열의 특정한 항만 더하기 - js

문제

두 정수 a, d와 길이가 n인 boolean 배열 included가 주어집니다. 첫째항이 a, 공차가 d인 등차수열에서 included[i]가 i + 1항을 의미할 때, 이 등차수열의 1항부터 n항까지 included가 true인 항들만 더한 값을 return 하는 solution 함수를 작성해 주세요.

 

제한 사항

1 ≤ a ≤ 100
1 ≤ d ≤ 100
1 ≤ included의 길이 ≤ 100
included에는 true가 적어도 하나 존재합니다.

 

입 출력 예

a	b	included	result
3	4	[true, false, false, true, true]	37
7	1	[false, false, false, true, false, false, false]	10

 

입출력 예 설명

예제 1번은 a와 d가 각각 3, 4이고 included의 길이가 5입니다. 이를 표로 나타내면 다음과 같습니다.

	1항	2항	3항	4항	5항
등차수열	3	7	11	15	19
included	true	false	false	true	true

 

따라서 true에 해당하는 1항, 4항, 5항을 더한 3 + 15 + 19 = 37을 return 합니다.

 

 

solution.js

function solution(a, d, included) {
    const arr = [a]
    for(let i = 1 ; i < included.length; i ++) {
        arr[i] = arr[i-1] + d
    }
    
    return arr.reduce((acc, cur, idx) => included[idx] ? acc+cur : acc , 0)
}

 

- return arr.reduce((acc, cur, idx) => included[idx] ? acc + cur : acc, 0): 배열 arr을 순회하며 included 배열의 값에 따라 항을 선택하여 합을 계산한다.

- arr.reduce((acc, cur, idx) => ... , 0): reduce 메서드를 이용하여 배열 arr의 모든 항목을 순회하며, 초기값 0을 가지는 acc 변수에 합을 누적한다.

- included[idx] ? acc + cur : acc: 현재 항(cur)의 인덱스(idx)를 기반으로 included 배열의 해당 인덱스가 true라면 합(acc + cur)에 항을 추가하고, false라면 누적합(acc)을 그대로 반환다.

 

reduce () 초기값?

 

return arr.reduce((acc, cur, idx) => included[idx] ? acc + cur : acc, 0)

 

- reduce 메서드 끝에 있는 , 0은 accumulator의 초기값입니다. 감소가 시작되기 전에 'acc'의 초기 값이 무엇인지 지정합니다. 이 경우 '0'으로 설정되어 있으므로 'acc'의 초기값이 '0'인 상태에서 축소가 시작됩니다.

 

초기 값 '0'을 제공함으로써 'arr' 배열이 비어 있거나 포함된 요소가 없는 경우(모든 'included' 값이 'false'임) 'reduce' 메서드가 여전히 '0'을 반환하도록 한다. 

요약하면 reduce 메소드는 누산기의 초기값으로 , 0을 사용하는데, 배열이 비어있거나 sum에 원소가 하나도 포함되지 않은 경우에도 유효한 결과를 보장한다.