[백준 2503번] 숫자 야구

📖문제:

 

📙풀이:

문제를 너무 어렵게 생각하고 다른 방향으로 풀다 보니 시간이 너무 오래 소요되었다. (1시간 반..)

결국 구글링을 하여 다른 사람의 코드를 참고함.

기존에 코드를 짠 방식은 아래와 같다.

브루트포스 방식으로 짜려고 의도했으나, 결과에 맞게 코드를 작성해 경우의 수를 좁혀나가려다 보니까 완전 하드 코딩이 되어 버렸다..

어떻게 코드를 간소화 시킬 수 있을지, 반복되는 부분을 줄일 수 있을지를 많이 고민해야 할 것 같다.

 

'''
https://www.acmicpc.net/problem/2503
문제: 숫자 야구
난이도: 실버3
'''

answer_box = []
# 가능한 모든 경우의 수
for i in range(1, 10):
    for j in range(1, 10):
        for k in range(1, 10):
            if i != j and j != k and i != k:
                answer_box.append(str(i*100 + j *10 + k))

three_strike = False

testCase = int(input())
for _ in range(testCase):
    num, s, b = map(str, input().split())
    this_case_box = []
    if s == '3': # 3스트라이크: 정답이 하나로 확정됨
        answer_box = [num]
        break
    elif s == '2': # 2스트라이크
        for i in range(1, 10):
            if num[0] != str(i):
                this_case_box.append(str(i)+num[1]+num[2])
            if num[1] != str(i):
                this_case_box.append(num[0]+str(i)+num[2])
            if num[2] != str(i):
                this_case_box.append(num[0]+num[1]+str(i))
    elif s == '1': # 1스트라이크
        # 1스트라이크 & 2볼
        if b == '2':
            this_case_box.append(num[0]+num[2]+num[1])
            this_case_box.append(num[2]+num[1]+num[0])
            this_case_box.append(num[1]+num[0]+num[2])
        # 1스트라이크 & 1볼
        elif b == '1':
            for i in range(1, 10):
                this_case_box.append(num[0]+num[2]+str(i))
                this_case_box.append(num[0]+str(i)+num[1])
                this_case_box.append(str(i)+num[1]+num[0])
                this_case_box.append(num[2]+num[1]+str(i))
                this_case_box.append(str(i)+num[0]+num[2])
                this_case_box.append(num[1]+str(i)+num[2])
        # 1스트라이크 & 0볼
        elif b == '0':
            for i in range(1, 10):
                for j in range(1, 10):
                    if str(i) != num[1] or str(j) != num[2] or str(i) != num[2] or str(j) != num[1]:
                        this_case_box.append(num[0]+str(i)+str(j))
                    if str(i) != num[0] or str(j) != num[2] or str(i) != num[2] or str(j) != num[0]:
                        this_case_box.append(str(i)+num[1]+str(j))
                    if str(i) != num[0] or str(j) != num[1]:
                        this_case_box.append(str(i)+str(j)+num[2])
    elif s == '0': # 0스트라이크
        # 0스트라이크 & 3볼
        if b == '3':
            this_case_box.append(num[1]+num[2]+num[0])
            this_case_box.append(num[2]+num[0]+num[1])
        # 0스트라이크 & 2볼
        if b == '2':
            for i in range(1, 10):
                this_case_box.append(str(i)+num[0]+num[1])
                this_case_box.append(str(i)+num[2]+num[0])
                this_case_box.append(str(i)+num[2]+num[1])
                this_case_box.append(num[1]+str(i)+num[0])
                this_case_box.append(num[2]+str(i)+num[0])
                this_case_box.append(num[1]+num[0]+str(i))
                this_case_box.append(num[1]+num[2]+str(i))
                this_case_box.append(num[2]+num[0]+str(i))
        # 0스트라이크 & 1볼
        if b == '1':
            for i in range(1, 10):
                for j in range(1, 10):
                    if str(i) != num[1] or str(j) != num[2] or str(i) != num[2] or str(j) != num[1]:
                        this_case_box.append(str(i)+num[0]+str(j))
                    if str(i) != num[1] or str(j) != num[2] or str(i) != num[2] or str(j) != num[1]:
                        this_case_box.append(str(i)+str(j)+num[0])
                    if str(i) != num[0] or str(j) != num[2] or str(i) != num[2] or str(j) != num[0]:
                        this_case_box.append(num[1]+str(i)+str(j))
                    if str(i) != num[0] or str(j) != num[2] or str(i) != num[2] or str(j) != num[0]:
                        this_case_box.append(str(i)+str(j)+num[1])
                    if str(i) != num[0] or str(j) != num[1] or str(i) != num[1] or str(j) != num[0]:
                        this_case_box.append(num[2]+str(i)+str(j))
                    if str(i) != num[0] or str(j) != num[1] or str(i) != num[1] or str(j) != num[0]:
                        this_case_box.append(str(i)+num[2]+str(j))
        # 0스트라이크 & 0볼
        if b == '0':
            for i in range(1, 10):
                for j in range(1, 10):
                    for k in range(1, 10):
                        if i != num[0] or i != num[1] or i != num[2] or j != num[0] or j != num[1] or j != num[2] or k != num[0] or k != num[1] or k != num[2]:
                            this_case_box.append(str(i)+str(j)+str(k))

    answer_box = list(set(answer_box) & set(this_case_box))
    print(answer_box)


print(len(answer_box))

 

✏️정답 코드:

'''
https://www.acmicpc.net/problem/2503
문제: 숫자 야구
난이도: 실버3
'''

import sys
from itertools import permutations
input = sys.stdin.readline

n = int(input())
nums = list(permutations(list(range(1, 10)), 3))	# 1부터 9까지의 서로 다른 3자리 숫자 모음

for _ in range(n):
    num, s, b = map(int, input().split())
    tmp = []	# 현재 조건에 맞는 숫자가 들어갈 배열

    for check in nums:		# 3자리 숫자 모음 탐색
        cnt_s, cnt_b = 0, 0	# 스트라이크와 볼의 갯수

        for i, str_num in enumerate(str(num)):	# 입력한 숫자를 문자열로 바꾸어 숫자의 위치와 값을 한 번에 확인
            if int(str_num) == check[i]:		# 현재 탐색중인 숫자와 입력한 숫자 중 위치와 값이 같은 수가 있다면
                cnt_s += 1						# 스트라이크 갯수 증가
            if int(str_num) != check[i] and int(str_num) in check:	# 위치는 다르나 숫자가 포함되어 있다면
                cnt_b += 1						# 볼 갯수 증가

        if s == cnt_s and b == cnt_b:			# 스트라이크와 볼의 갯수가 입력값과 같다면
            tmp.append(check)					# tmp에 현재 숫자 추가
    nums = tmp									# 기존 3자리 숫자의 모음을 현재까지의 조건에 맞춰 최신화

print(len(nums))								# 최종 조건에 맞는 수들의 길이 출력

 

🔗링크:

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