[programmers] 완전탐색, 숫자 야구

코딩테스트 연습 - 숫자 야구 | 프로그래머스

 

문제

각자 서로 다른 1~9까지 3자리 임의의 숫자를 정한 뒤 서로에게 3자리의 숫자를 불러서 결과를 확인합니다. 그리고 그 결과를 토대로 상대가 정한 숫자를 예상한 뒤 맞힙니다.

• 숫자는 맞지만, 위치가 틀렸을 때는 볼
• 숫자와 위치가 모두 맞을 때는 스트라이크
• 숫자와 위치가 모두 틀렸을 때는 아웃

제한사항

• 질문의 수는 1 이상 100 이하의 자연수입니다.
• baseball의 각 행은 [세 자리의 수, 스트라이크의 수, 볼의 수] 를 담고 있습니다.

질문한 세 자리의 수, 스트라이크의 수, 볼의 수를 담은 2차원 배열 baseball이 매개변수로 주어질 때, 가능한 답의 개수를 return 하도록 solution 함수를 작성해주세요.

해결

ket point, 각자 서로 다른 1~9라는 말에서 3자리 숫자는 서로 다르고 0을 포함하지 않는 다는 것을 의미합니다.

1. 123부터 987까지의 수 중 오름차순으로 숫자 하나를 고른다.
2. 이때 세 자리 숫자는 서로 같을 수 없고, 0을 포함하지 않는다.
3. 질문 배열의 인덱스에 접근하여 해당 질문에 대해서 선택한 숫자와 스트라이크 수와, 볼 수가 일치하는지 확인한다.
4. 여러 질문 중 하나라도 스트라이크 수와 볼의 수가 같지 않다면 질문 배열 탐색을 중지하고 다음 숫자를 선택한다.
5. 모든 질문을 통과한다면 answer의 수를 증가시킨다.

이렇게 모든 반복문을 거치고나면 answer의 개수가 가능한 답의 개수가 됩니다.

 코드

#include <iostream>
#include <string>
#include <vector>
using namespace std;
 
int solution(vector<vector<int>> baseball) {
    int answer = 0;
 
    for (int i = 123; i <= 987; i++) {
        string num = to_string(i);
        if (num[0] == '0' || num[1] == '0' || num[2] == '0') continue;
        if (num[0] == num[1] || num[1] == num[2] || num[2] == num[0]) continue;
        bool flg = false;
        for (int i = 0; i < baseball.size(); i++) {
            int bb = baseball[i][0], strike = baseball[i][1], ball = baseball[i][2];
            string ballnum = to_string(bb);
            int s = 0, b = 0;
            for (int j = 0; j < 3; j++) {
                for (int k = 0; k < 3; k++) {
                    if (ballnum[j] == num[k]) {
                        if (j == k) {
                            s++;
                        }
                        else {
                            b++;
                        }
                    }
                }
            }
            if (strike != s || ball != b) {
                flg = true; break;
            }
        }
        if (!flg)answer++;
    }
    return answer;
}