[백준]2503번: 숫자 야구

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

2503번: 숫자 야구

---

✅ 아이디어

머리로는 너무 잘 알겠는데 코드로 구현하려니까 아이디어가 떠오르질 않았었다.

 

간단하게 생각해서 브루트포스로 풀면 된다.

 

민혁이가 질문할 때마다

정답이 될 수 있는 모든 수(123 ~ 987)를 가지고 있는 리스트에서 답일 수 없는 수들을 모두 제외하여,

마지막에 남는 수들의 개수를 출력한다.

 

예를 들면 민혁이가 123이라고 질문, 영수가 1 스트라이크 1 볼이라고 대답했다면,

123에 대한 대답이 1스트라이크 1볼이 나올 수 없는 수들은 정답 후보가 될 수 없으므로, 모두 리스트에서 제외한다.

 

 

✅ 주의할점

 - 정답 후보는 111부터 999까지가 아닌, 123부터 987까지이다.

 - 0이 포함되거나 중복된 숫자가 있는 수는 정답 후보에서 제외해야 한다. (예시: 130, 114)

 

✅ 제출 코드

import sys
input = sys.stdin.readline

def check(q, a, b):
  qnum = [q//100, (q%100)//10, q%10] # 영수가 말한 답
  for i in range(123,988):
    if i not in ans_lst:
      continue
    strike, ball = 0, 0
    inum = [i//100, (i%100)//10, i%10]
    for j in range(3):
      if qnum[j] in inum:
        ball += 1
      if qnum[j] == inum[j]:
        strike += 1
        ball -= 1
    if strike == a and ball == b:
      pass
    else:
      if i in ans_lst:
        ans_lst.remove(i)

# 정답 후보 리스트
ans_lst = [i for i in range(123,988)]
for i in range(123,988):
  if len(set([i//100, (i%100)//10, i%10])) != 3 or '0' in str(i):
      if i in ans_lst:
        ans_lst.remove(i)
        
n = int(input())
for _ in range(n):
  x, y, z = map(int,input().split())
  if y == 3:
    print(1)
    exit()
  check(x,y,z)

print(len(ans_lst))