[programmers/js] 비밀 코드 해독

비밀 코드 해독

https://school.programmers.co.kr/learn/courses/30/lessons/388352

문제 설명

시스템은 1부터 n까지의 서로 다른 정수 5개가 오름차순으로 정렬된 비밀 코드를 가지고 있으며, 당신은 이 비밀 코드를 맞혀야 합니다.

당신은 비밀 코드를 알아내기 위해 암호 분석 도구를 사용하며, m번의 시도를 할 수 있습니다. 각 시도마다 서로 다른 5개의 정수를 입력하면, 시스템은 그 중 몇 개가 비밀 코드에 포함되어 있는지 알려줍니다.

만약 비밀 코드가 [3, 5, 7, 9, 10]이고, 입력한 정수가 [1, 2, 3, 4, 5]라면 비밀 코드에 포함된 정수는 3, 5 두 개이므로 시스템은 2를 응답합니다.
당신은 m번의 시도 후, 비밀 코드로 가능한 정수 조합의 개수를 알고 싶습니다.

입력한 정수	시스템 응답(일치하는 개수)
[1, 2, 3, 4, 5]	2개
[6, 7, 8, 9, 10]	3개
[3, 7, 8, 9, 10]	4개
[2, 5, 7, 9, 10]	3개
[3, 4, 5, 6, 7]	3개

비밀 코드로 가능한 정수 조합은 아래와 같이 3개가 있습니다.

• [3, 4, 7, 9, 10]
  첫 번째 시도에서 비밀 코드에 포함된 정수가 3, 4로 2개 있습니다.
  두 번째 시도에서 비밀 코드에 포함된 정수가 7, 9, 10으로 3개 있습니다.
  세 번째 시도에서 비밀 코드에 포함된 정수가 3, 7, 9, 10으로 4개 있습니다.
  네 번째 시도에서 비밀 코드에 포함된 정수가 7, 9, 10으로 3개 있습니다.
  다섯 번째 시도에서 비밀 코드에 포함된 정수가 3, 4, 7로 3개 있습니다.

• [3, 5, 7, 8, 9]
  첫 번째 시도에서 비밀 코드에 포함된 정수가 3, 5로 2개 있습니다.
  두 번째 시도에서 비밀 코드에 포함된 정수가 7, 8, 9로 3개 있습니다.
  세 번째 시도에서 비밀 코드에 포함된 정수가 3, 7, 8, 9로 4개 있습니다.
  네 번째 시도에서 비밀 코드에 포함된 정수가 5, 7, 9로 3개 있습니다.
  다섯 번째 시도에서 비밀 코드에 포함된 정수가 3, 5, 7로 3개 있습니다.

• [3, 5, 7, 8, 10]
  첫 번째 시도에서 비밀 코드에 포함된 정수가 3, 5로 2개 있습니다.
  두 번째 시도에서 비밀 코드에 포함된 정수가 7, 8, 10으로 3개 있습니다.
  세 번째 시도에서 비밀 코드에 포함된 정수가 3, 7, 8, 10으로 4개 있습니다.
  네 번째 시도에서 비밀 코드에 포함된 정수가 5, 7, 10으로 3개 있습니다.
  다섯 번째 시도에서 비밀 코드에 포함된 정수가 3, 5, 7로 3개 있습니다.

정수 n, 입력한 정수를 담은 2차원 정수 배열 q와 시스템 응답을 담은 1차원 정수 배열 ans가 매개변수로 주어집니다. 이때, 비밀 코드로 가능한 정수 조합 개수를 return 하도록 solution 함수를 완성해 주세요.

풀이

n이 최대 30이고, 5개를 선택하는 경우의 수는 C(30,5) = 142,506이므로, 완전 탐색을 통해 해결할 수 있습니다.

1. 길이가 5인 가능한 모든 배열을 찾습니다.
2. 배열이 ans 조건과 같다면 count를 증가.

function solution(n, q, ans) {
    let count = 0;
    const dfs = (idx, arr) => {
        if (arr.length === 5) {
            for (let i = 0; i < q.length; i++) {
                const matchCount = q[i].filter(num => arr.includes(num)).length;
                if (matchCount !== ans[i]) return; // 조건 불일치 -> 조기 종료
            }
            count++; // 모든 조건 만족하면 카운트 증가
            return;
        }
        
        for (let i = idx; i <= n; i++) {
            dfs(i + 1, [...arr, i])
        }
    }
    dfs(1, []);
    return count;
}