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

< 처음 풀 때 >

dict 의 활용 방법을 알지 못했고 계속 시간 초과가 발생하였다.

 

< 해결 방법 > 

dict 를 활용해서 시간을 줄이고자 하였다. dict 의 활용 방법이 미숙해서 검색을 통해 알아보았고, dict 과 리스트를 둘 다 만드는 판단은 맞았었다.

 
import sys
from collections import Counter
 
a,q = sys.stdin.readline().rstrip().split(" ")
 
b = { }
 
c = [ ]
 
for i in range(int(a)):
    k = sys.stdin.readline().rstrip()
    b[k]=i
    c.append(k)
 
for i in range(int(q)):
    k = sys.stdin.readline().rstrip()
    if k.isdigit()==False:
        print(b[k]+1)
    else:
        print(c[int(k)-1])