백준 1620번 - 나는야 포켓몬 마스터 이다솜

import sys

n, m = map(int, sys.stdin.readline().split())
dogam = {}
dogam2 = {}

for i in range(n):
  temp = sys.stdin.readline().rstrip()
  dogam[i+1] = temp
  dogam2[temp] = i+1

for _ in range(m):
  cur = sys.stdin.readline().rstrip()
  try:
    cur = int(cur)
    print(dogam[cur])
  except:
    name = str(cur)
    print(dogam2[name])