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

포켓몬 도감에 수록되어 있는 포켓몬의 개수N

내가 맞춰야하는 문제의 개수 M

N개의 줄에 번호가 1번인 포켓몬부터 N번인 포켓몬까지 저장

문자로 들어오면 해당 숫자 출력,

숫자로 들어오면 해당 문자 출력

import sys
input=sys.stdin.readline
n,m=map(int,input().split())
dic={}

for i in range(1,n+1):  #1번부터 n번까지
    a=input().rstrip()
    dic[a]=str(i)  #a=str(키값)
    dic[str(i)]=a  #키값=a

for i in range(m):
    print(dic[input().rstrip()])