백준 / 나는야포켓몬마스터 / 1620

Question

문제링크
Silver 4

Logic

기본 구조 : dictionary
1. 딕셔너리를 두 개 생성한다
2. 입력된 포켓몬에 대해, 숫자를 키로 하는 경우와 포켓몬 이름을 키로 하는 경우로 각각 입력한다.
3. 입력이 끝나면 명령어의 종류(숫자/문자)에 따라 다른 딕셔너리로 출력한다.

Code

from sys import stdin

N,M = map(int,stdin.readline().strip().split())
pokenum = {}
pokelis = {}
for i in range(1,N+1):
    s = stdin.readline().strip()
    pokenum[str(i)]=s
    pokelis[s]=str(i)

for i in range(M):
    s = stdin.readline().strip()
    if s.isalpha() : print(pokelis[s])
    else : print(pokenum[s])