중첩되는 반복문

요약

중첩되는 반복문은 재귀함수를 이용하면 좋다.

문제

한국중학교에 다니는 학생들은 각자 정수 번호를 갖고 있습니다. 이 학교 학생 3명의 정수 번호를 더했을 때 0이 되면 3명의 학생은 삼총사라고 합니다. 예를 들어, 5명의 학생이 있고, 각각의 정수 번호가 순서대로 -2, 3, 0, 2, -5일 때, 첫 번째, 세 번째, 네 번째 학생의 정수 번호를 더하면 0이므로 세 학생은 삼총사입니다. 또한, 두 번째, 네 번째, 다섯 번째 학생의 정수 번호를 더해도 0이므로 세 학생도 삼총사입니다. 따라서 이 경우 한국중학교에서는 두 가지 방법으로 삼총사를 만들 수 있습니다.
한국중학교 학생들의 번호를 나타내는 정수 배열 number가 매개변수로 주어질 때, 학생들 중 삼총사를 만들 수 있는 방법의 수를 return 하도록 solution 함수를 완성하세요.

풀이

3의 조합이 0인지 확인하면 된다.

단순한 풀이
for문을 3번 중첩하여 확인할 수 있다.

유연한 풀이
그러나 위 풀이는 조합의 크기를 셋에서 바꾸게 되면 일일이 중첩되는 반복문으로 코드 상당부분을 수정해줘야 한다.
아래 풀이는 이 문제를 해결했다.

function solution(number) {
    let result = 0;

    const combination = (currentArray, startIndex) => {
        if (currentArray.length === 3) {
            const sum = currentArray.reduce((acc, cur) => acc + cur, 0)
            result += sum === 0 ? 1 : 0
            return
        }
        
        for (let i = startIndex; i < number.length; i++) {
            const updatedArray = [...currentArray, number[i]]
            console.log("i", i, "current number:", number[i], "updated array:", updatedArray)
            combination(updatedArray, i + 1)
        }
    }

    combination([], 0)

    return result;
}

console.log(solution([-3, -2, -1, 0, 1, 2, 3])) // prints 5

번외: for문 초깃값 설정할 때 꼭 let 사용

해당 풀이를 발견해서 감탄하고 실제로 구현을 해보는데 반복문 중첩이 제대로 되지 않았다.
for (i = start; i < currentArray.length; i++)의 조건문을 let i = start;로 바꾸자 정상적으로 작동했다.