[백준 1620] 나는야 포켓몬 마스터 이다솜 (Python)

1620번: 나는야 포켓몬 마스터 이다솜

 이번 문제는 굉장한 스토리텔링이 있는 문제기에 접속하여 읽어보면 재밌을 듯하다.

import sys
n, m = map(int, sys.stdin.readline().split())
dogam = [sys.stdin.readline().rstrip() for _ in range(n)]
dogam_dic = {}
for i in range(len(dogam)) :
      dogam_dic[dogam[i]] = i+1

for i in range(m) :
    temp = sys.stdin.readline().rstrip()
    print(dogam[int(temp)-1] if temp.isdigit() else dogam_dic[temp])

 이번에도 시간을 줄이기 위해 여러 방법을 사용했다. 우선 dictionary를 이용하여 단어의 위치를 표현하였고, 주어진 단어가 리스트에 있는지 확인하기 위해 in 문법을 사용했더니 시간이 너무 오래 걸렸다. 반대로 숫자임을 isdigit()으로 우선적으로 파악하여 그 시간을 줄인 덕분에 이번 문제를 해결할 수 있었다. 

 dictionary에도 익숙해지고 시간을 줄이기 위한 방법을 이것저것 알아가는 중이다. :)