[Python] BOJ/백준 1620번 나는야 포켓몬 마스터 이다솜

[문제]

https://www.acmicpc.net/problem/1620

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

[풀이]

입력되는 내용을 딕셔너리 타입의 book에 key : value 값을 각각 {번호 : 이름}, {이름 : 번호} 로 저장한 뒤 key 값을 이용해 value 를 찾아 출력한다.

[코드]

import sys

if __name__ == '__main__':
    input = sys.stdin.readline
    n, m = map(int, input().split())
    book = {}

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

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