딕셔너리에 저장된 키와 값을 반대로 만드려면 딕셔너리 컴프리헨션을 사용하자.
dict_inv = {val:key for key, val in dict_original.items()}
dictionary.items()
는 키 값을 튜플 형태로 반환한다.
import sys
n, m = map(int, sys.stdin.readline().rstrip().split())
pokemon = {}
for i in range(1, n+1):
pokemon[i] = sys.stdin.readline().rstrip()
pokemon_inv = {val:key for key, val in pokemon.items()}
for _ in range(m):
question = sys.stdin.readline().rstrip()
if question.isdigit():
print(pokemon.get(int(question)))
else:
print(pokemon_inv.get(question))