JS Practice) 프로그래머스 Lv.1 : 삼총사

 

삼총사

문제설명 :

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

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

 

제한 사항

3 ≤ number의 길이 ≤ 13

-1,000 ≤ number의 각 원소 ≤ 1,000

서로 다른 학생의 정수 번호가 같을 수 있습니다.

 

입출력 예

number	result
[-2, 3, 0, 2, -5]	2
[-3, -2, -1, 0, 1, 2, 3]	5
[-1, 1, -1, 1]	0

 

접근

: 3개의 반복문을 돌며, 서로 다른 index의 요소를 가져와 해당 숫자들의 합이 0이면 answer에 1을 더해주었다. 

function solution(number) {
    let answer = 0;
    
    number.forEach((n1, idx1) => {
        for(let i = idx1+1; i<number.length; i++){
            for(let j = i+1; j<number.length; j++){
                if(n1 + number[i] + number[j] === 0) {
                    answer++
                }
            }
        }
    })
    
    return answer
}

 

 

solution

function solution(number) {
    let answer = 0;
    
    // dfs: 현재 학생을 선택할지 말지 결정하는 재귀함수
    // idx: 현재 학생의 index, sum: 선택한 학생들의 정수 번호의 합, count: 선택한 학생의 수
    function dfs(idx, sum, count) { 
        if (count === 3) { // 선택한 학생이 3명이면 정수 번호의 합이 0인지 확인
            if (sum === 0) {
                answer++;
            }
            return;
        }
        
        // idx와 number 배열의 길이가 같으면 모든 학생을 고려했으므로 함수 종료
        if (idx === number.length) { 
            return;
        }
        
        dfs(idx + 1, sum + number[idx], count + 1); // 현재 학생을 선택한 경우
        dfs(idx + 1, sum, count); // 현재 학생을 선택하지 않은 경우
    }
    
    dfs(0, 0, 0);
    
    return answer;
}

// DFS(깊이우선탐색) 알고리즘 문제