백준 2503 - 숫자 야구

문제

https://www.acmicpc.net/problem/2503

2503번: 숫자 야구

 

풀이

숫자 야구를 푸는 알고리즘은 크게 2가지가 있다.

1. 배열에 숫자들을 넣고 조건과 비교해서 조건과 맞지 않는 숫자들을 제거

2. 123부터 시작해서 조건과 맞는 숫자 후보들을 증가

 

배열에 숫자를 넣을 경우 0이 들어가거나 중복되는 수가 들어가는 것은 빼야한다. 배열에 숫자를 넣은 후 문제에 조건과 맞지 않는 숫자들을 제거하고 배열에 남은 수를 출력하면 된다.

코드

#include <iostream>
 
using namespace std;
 
int N;
int game[988];
int num;
int s, b;
int result;
 
// 스트라이크
int getS(int num, int cmp)
{
    int strike = 0;
    if ((num / 100) == (cmp / 100)) strike++;
    if ((num % 100 / 10) == (cmp % 100 / 10)) strike++;
    if (num % 10 == cmp % 10) strike++;
    return strike;
}
 
// 볼
int getB(int num, int cmp)
{
    int ball = 0;
    int firstA = num / 100, firstB = cmp / 100;
    int secondA = num % 100 / 10, secondB = cmp % 100 / 10;
    int thirdA = num % 10, thirdB = cmp % 10;
 
    if (firstA == secondB || firstA == thirdB) ball++;
    if (secondA == firstB || secondA == thirdB) ball++;
    if (thirdA == firstB || thirdA == secondB) ball++;
 
    return ball;
}
 
int main()
{
    cin.tie(0);
    ios::sync_with_stdio(false);
 
    cin >> N;
 
    // 중복되거나 0이 들어간 수 제거
    for (int i = 123; i <= 987; i++)
    {
        int i1 = i / 100;
        int i2 = i % 100 / 10;
        int i3 = i % 10;
 
        if (i2 == 0 || i3 == 0 || i1 == i2 || i1 == i3 || i2 == i3) game[i] = 0;
        else game[i] = i;
    }
 
    // 조건과 비교
    for (int i = 0; i < N; i++)
    {
        cin >> num >> s >> b;
        for (int x = 123; x <= 987; x++)
        {
            if (!game[x]) continue;
 
            int cs = getS(x, num);
            int cb = getB(x, num);
 
            if (!((cs == s) && (cb == b))) game[x] = 0;
        }
    }
 
    for (int i = 123; i <= 987; i++)
    {
        if (game[i]) result++;
    }
 
    cout << result;
 
    return 0;
}