백준 알고리즘 1620번 나는야 포켓몬 마스터 이다솜(python)

기본적인 해시 문제이다. dict가 확실히 속도가 빠른 것 같다. 그냥 리스트로 풀면 통과가 안된다. 

import sys

N, M = map(int, sys.stdin.readline().rstrip().split())
d1, d2 = {}, {}
for i in range(1, N+1):
    s = sys.stdin.readline().rstrip()
    d1[s] = i
    d2[i] = s
    
for _ in range(M):
    s = sys.stdin.readline().rstrip()
    if s.isdigit():
        print(d2[int(s)])
    else:
        print(d1[s])