백준 1620번: 나는야 포켓몬 마스터 이다솜 (파이썬)

import sys

n, m = map(int, sys.stdin.readline().split())
n_dict = dict()
for i in range(n):
  name = sys.stdin.readline().rstrip()
  n_dict[i+1] = name
    
n_dict2 = { v:k for k,v in n_dict.items()}
for i in range(m):
  qes = sys.stdin.readline().rstrip()
  if qes.isnumeric():
    print(n_dict[int(qes)])
  else :
    print(n_dict2[qes])