비밀 코드 해독

---

📌 문제 설명

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

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

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

비밀 코드에 사용된 정수의 범위가 1~10일 때, 아래와 같이 5번의 시도를 했다고 가정해 보겠습니다.

 

입력한 정수	시스템 응답(일치하는 개수)
[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 함수를 완성해 주세요.

 

📌 제한 사항

• 10 ≤ n ≤ 30
• 1 ≤ (q의 길이 = m) ≤ 10
  • q[i]의 길이 = 5
  • q[i]는 i+1번째 시도에서 입력한 5개의 서로 다른 정수를 담고 있으며, 오름차순으로 정렬되어 있습니다.
  • 1 ≤ q[i][j] ≤ n
• ans의 길이 = m
  • ans[i]는 i+1번째 시도에서 입력한 5개의 정수 중 비밀 코드에 포함된 정수의 개수를 나타냅니다.
  • 0 ≤ ans[i] ≤ 5
• 비밀 코드가 존재하지 않는(답이 0인) 경우는 주어지지 않습니다.

📌 입출력 예

n	q	ans	result
10	[[1, 2, 3, 4, 5], [6, 7, 8, 9, 10], [3, 7, 8, 9, 10], [2, 5, 7, 9, 10], [3, 4, 5, 6, 7]]	[2, 3, 4, 3, 3]	3
15	[[2, 3, 9, 12, 13], [1, 4, 6, 7, 9], [1, 2, 8, 10, 12], [6, 7, 11, 13, 15], [1, 4, 10, 11, 14]]	[2, 1, 3, 0, 1]	5

 

🔗 문제 링크

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

프로그래머스

 

---

반응형

💡 풀이 아이디어

• 모든 조합을 생성 후 조건에 맞지 않는 조합은 제거한다.

📝 최종 풀이

• 백트래킹으로 모든 조합을 생성합니다.
• 현재 조합과 각 입력 시도의 교집합을 비교합니다.

---

728x90

👩‍💻 코드

#include <string>
#include <vector>
#include <set>

using namespace std;

void generateCombinations(int n, int start, vector<int>& current, vector<vector<int>>& combinations) {
    if (current.size() == 5) {
        combinations.push_back(current);
        return;
    }
    
    for (int i = start; i <= n; i++) {
        current.push_back(i);
        generateCombinations(n, i + 1, current, combinations);
        current.pop_back();
    }
}

bool isValidCode(const vector<int>& code, const vector<vector<int>>& q, const vector<int>& ans) {
    set<int> codeSet(code.begin(), code.end());

    for (int i = 0; i < q.size(); i++) {
        set<int> querySet(q[i].begin(), q[i].end());
        int matchCount = 0;

        for (int num : codeSet) {
            if (querySet.count(num)) {
                matchCount++;
            }
        }

        if (matchCount != ans[i]) {
            return false;
        }
    }
    
    return true;
}

int solution(int n, vector<vector<int>> q, vector<int> ans) {
    int answer = 0;
    vector<vector<int>> combinations;
    vector<int> current;
    generateCombinations(n, 1, current, combinations);

    for (const auto& code : combinations) {
        if (isValidCode(code, q, ans)) {
            answer++;
        }
    }

    return answer;
}