백준 1620 [나는야 포켓몬 마스터 이다솜] 파이썬

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

1620번: 나는야 포켓몬 마스터 이다솜

# 백준 1620 나는야 포켓몬 마스터 이다솜
import sys
input=sys.stdin.readline
N,M=map(int,input().split())
mon_to_digit={}
digit_to_mon={}
for i in range(1,N+1):
    monster=str(input().rstrip())
    mon_to_digit[monster]=i
    digit_to_mon[str(i)]=monster
for _ in range(M):
    val=input().rstrip()
    if mon_to_digit.get(val):
        print(mon_to_digit[val])
    else:
        print(digit_to_mon[val])

딕셔너리를 두개 사용해서 양쪽으로 참조가 가능하도록 구성하였다. 또, 딕셔너리의 get 함수를 이용해서 딕셔너리에 해당 key 값이 존재하는지 확인하였다.

 

풀고나서, 찾아보니 더 좋은 풀이를 푸신 분이 계셔서 참조한다.

import sys

input = sys.stdin.readline

n, m = map(int, input().split())

dict = {}

for i in range(1, n + 1):
    a = input().rstrip()
    dict[i] = a
    dict[a] = i

for i in range(m):
    quest = input().rstrip()
    if quest.isdigit():
        print(dict[int(quest)])
    else:
        print(dict[quest])

딕셔너리에 값을 다 저장하고, 들어오는 값에 대해 정수인지 판단한다.

https://gudwns1243.tistory.com/63 <- 참조 블로그