[백준 2503] 숫자 야구

 

 

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

2503번: 숫자 야구

 

프로그래머스에서 푼 문제와 동일하다.

처음에 계속 틀렸다고 나오길래 다시보니 0은 포함하지 않는다.

 

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

using namespace std;

set<int> possibles;

void makePossible(string num, int cnt, vector<bool>& visited) {
	if (cnt == 3) {
		possibles.insert(atoi(num.c_str()));
		return;
	}

	for (int i = 1; i < 10; i++) {
		if (!visited[i]) {
			visited[i] = true;
			makePossible(num + to_string(i), cnt + 1, visited);
			visited[i] = false;
		}
	}
}

void deleteNotPoss(int predict, int strike, int ball) {
	string predict_str = to_string(predict);
	for (set<int>::iterator it = possibles.begin(); it != possibles.end(); ) {
		string it_str = to_string(*it);
		int S = 0, B = 0;
		
		for (int i = 0; i < 3; i++) {
			for (int j = 0; j < 3; j++) {
				if (predict_str[i] == it_str[j]) {
					if (i == j) S++;
					else B++;
				}
			}
		}

		if (strike != S || ball != B) {
			set<int>::iterator temp = it;
			temp++;
			possibles.erase(*it);
			it = temp;
		}
		else
			it++;
	}
}


int main() {
	int answer = 0, n;
	vector<bool> visited(10, false);
	makePossible("", 0, visited);

	cin >> n;

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

		deleteNotPoss(predict, strike, ball);

	}

	answer = possibles.size();
	printf("%d\n", answer);
	return 0;
}