1620 나는야 포켓몬 마스터

문제

포켓몬 이름이 여러줄에 걸쳐 주어지는데

포켓몬 이름의 경우 도감의 index를
숫자의 경우 포켓몬 이름을 말하면 된다.

코드

import sys

len_of_poketmon, num_of_problem = map(int, sys.stdin.readline().split())
data = []
dictionary = {}
for i in range(len_of_poketmon):
    poketmon_name = sys.stdin.readline().strip()
    data.append(poketmon_name)
    dictionary[poketmon_name] = i
for i in range(num_of_problem):
    input = sys.stdin.readline().strip()
    if(input.isnumeric()):
        print(data[int(input)-1])
    else:
        print(int(dictionary[input])+1)

풀이

숫자의 경우 도감배열에서 쉽게 접근 가능했다.
포켓몬 이름의 경우, 순회하면서 찾으면 시간 초과가 나와서 파이썬의 딕셔너리를 사용하였다.

strip()라는 함수로 개행문자를 쉽게 처리할 수 있었다.