본문 바로가기
PRO

프로그래머스 1단계 : 삼총사

by 코빈_ 2022. 12. 5.

프로그래머스 1단계 : 삼총사

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

한국중학교 학생들의 번호를 나타내는 정수 배열 number가 매개변수로 주어질 때,
학생들 중 삼총사를 만들 수 있는 방법의 수를 return 하도록 solution 함수를 완성하세요.


제한 사항

3 ≤ number의 길이 ≤ 13
-1,000 ≤ number의 각 원소 ≤ 1,000
서로 다른 학생의 정수 번호가 같을 수 있습니다.

나의 문제 풀이

이 문제는 풀다가 도저히 몰라서 다른 분의 코드를 참고했습니다.

function solution(number) {
    let result = 0;
    let length = number.length;

    if(length === 3) {
        return number.reduce((a, b) => a + b) === 0 ? 1 : 0
    }

    for(let i = 0; i < length - 2; i++) {
        for(let j = i + 1; j <  length - 1; j++) {
            for(let k = j + 1; k < length; k++) {
                if(number[i] + number[j] + number[k] === 0) {
                    result++;
                }
            }
        }
    }
    return result;
}

다른 사람 문제 풀이

function solution(number) {
    let result = 0;

    const combination = (current, start) => {
        if (current.length === 3) {
            result += current.reduce((acc, cur) => acc + cur, 0) === 0 ? 1 : 0;
            return;
        }

        for (let i = start; i < number.length; i++) {
            combination([...current, number[i]], i + 1);
        }
    }
    combination([], 0);
    return result;
}

댓글


INFORMATION

javascript

css

html