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

문제링크

딕셔너리에 순서:포켓몬이름, 포켓몬이름:순서를 각각 저장해주면 된다.

import sys

n, m = map(int, input().split())

dn = {}

for i in range(n):
    name = sys.stdin.readline().rstrip()
    
    dn[i] = name
    dn[name] = i
    
for _ in range(m):
    query = sys.stdin.readline().rstrip()
    
    if query.isdigit():
        print(dn[int(query) - 1])
    else:
        print(dn[query] + 1)