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

문제 출처 : https://www.acmicpc.net/problem/1620

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

내 풀이

import sys

input = sys.stdin.readline

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

dict = {}

for i in range(1, n + 1):
    a = input().rstrip()
    dict[i] = a
    dict[a] = i

for i in range(m):
    quest = input().rstrip()
    if quest.isdigit():
        print(dict[int(quest)])
    else:
        print(dict[quest])

설명

주어지는 내용을 dict에 key : value로  번호 : 이름, 이름 : 번호로 한번씩 저장한 후에 

그대로 찾아서 출력만 하면 되는 문제였다.