BOJ 2503 : 숫자야구 (Silver 5)

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

int number;
int strike, ball;
vector<int> array;
void Filtering(int num, int s, int b)
{
    vector<int> newarray;
    vector<int>::iterator iter;
    int strike = 0, ball = 0;
    for (iter = array.begin(); iter != array.end(); iter++)
    {
        int strike = 0, ball = 0;
        // cout << *iter << "검사\n";
        int a1 = num / 100;
        int b1 = num / 10 % 10;
        int c1 = num % 10;
        int a2 = *iter / 100;
        int b2 = *iter / 10 % 10;
        int c2 = *iter % 10;
        // cout << a1 << a2 << "/" << b1 << b2 << "/" << c1 << c2 << "\n";
        if (a1 == a2)
            strike++;
        else if (a1 == b2)
            ball++;
        else if (a1 == c2)
            ball++;
        if (b1 == b2)
            strike++;
        else if (b1 == a2)
            ball++;
        else if (b1 == c2)
            ball++;
        if (c1 == c2)
            strike++;
        else if (c1 == a2)
            ball++;
        else if (c1 == b2)
            ball++;
        if (strike == s && ball == b && a2 != b2 && b2 != c2 && a2 != c2 && a2 != 0 && b2 != 0 && c2 != 0)
        {
            newarray.push_back(*iter);
            //     cout << *iter << "\n";
        }
    }
    array = newarray;
    return;
}

int main()
{
    int T;
    cin >> T;
    for (int i = 123; i <= 987; i++)
    {
        array.push_back(i);
    }
    for (int i = 0; i < T; i++)
    {
        cin >> number >> strike >> ball;
        Filtering(number, strike, ball);
        // cout << "****************************\n";
    }
    cout << array.size();
}

이제 맨 처음엔 123~987 로 array를 업데이트 ㅐㅎ주고, 이 안에서 각 자리 숫자를 구해서 strike랑 ball을 구한 뒤, 함수 인자로 넘겨준 정답 strike, ball과 값이 일치하는지 검사함. 그래서 일치하면 그 숫자는 가능한 숫자이므로 newArray에 추가해줌. 이 짓을 모든 array에 있는 값에 대해 해준 뒤, newArray로 array를 대체해줌. (업데이트 해주는것!)

이 때, 각 자리 숫자에 0이 있거나, 같은 숫자가 있는 경우는 추가해 주면 안된다.