BOJ 2503 - 숫자 야구 / C++

문제 링크 : https://www.acmicpc.net/problem/2503

2503번: 숫자 야구

 

 

문제 풀이

1. 숫자는 1~9까지이다.

2. 세 자리 수 모두 겹치는 숫자가 없다.

 

즉, 가능한 숫자는 123 ~ 987이다.

따라서 미리 123부터 987까지 들어있는 배열을 만들고

n개의 질문마다 strike, ball을 직접 해보며 조건에 일치하지 않으면 배열에서 삭제한다.

 

결국 배열에는 가능한 숫자들만 남아있기 때문에 해당 배열의 size를 리턴해주면 된다.

 

#include <iostream>
#include <vector>
#include <string>
using namespace std;

vector<string> candidate;

pair<int, int> compareString(string a, string b){
    int strike = 0, ball = 0;

    for(int i=0; i<3; i++){
        if(a[i] == b[i]) strike++;
    }

    for(int i=0; i<3; i++){
        for(int j=0; j<3; j++){
            if(i==j) continue;

            if(a[i] == b[j]) ball++;
        }
    }

    return make_pair(strike, ball);
}

int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(nullptr); cout.tie(nullptr);

    int n;
    cin >> n;

    for(int i=123; i<1000; i++){
        string tmp = to_string(i);
        char a = tmp[0];
        char b = tmp[1];
        char c = tmp[2];
        
        if(a == b || b == c || a == c || b == '0' || c == '0') continue;

        candidate.push_back(tmp);
    }

    for(int i=0; i<n; i++){
        string num;
        int strike, ball;
        cin >> num >> strike >> ball;

        pair<int, int> tmp = make_pair(strike, ball);
        
        for(int j=0; j<candidate.size(); ){
            if(tmp == compareString(num, candidate[j])) j++;
            else {
                candidate.erase(candidate.begin() + j);
            }
        }
    }

    cout << candidate.size();
}