[BOJ/python] 1620: 나는야 포켓몬 마스터 이다솜

문제

silver 4 / 해시

문제 흐름

포켓몬을 입력받으면 names와 numbers 딕셔너리에 각각 값을 저장한다.
주의해야할 점은 input받은 값에 rstrip()을 해줘야 한다는 점..
그리고 포켓몬 번호가 1부터 시작한다는 점이다.
이 두가지를 주의한다면 크게 어려운 점은 없다.
딕셔너리에 저장한 값을 토대로 퀴즈의 답을 찾아낸다.

코드

# 나는야 포켓몬 마스터 이다솜
# hash

import sys
input = sys.stdin.readline

if __name__ == "__main__":
    n, m = map(int, input().split())
    names = {}
    numbers = {}
    for i in range(n):
        name = input().rstrip()
        names[name] = i+1
        numbers[i+1] = name
    for i in range(m):
        quize = input().rstrip()
        try:
            number = int(quize)
            print(numbers[number])
        except:
            print(names[quize])

마무리

사실 1부터 시작한다는 점을 확인 안해서 조금 헤맸다.
문제를 똑바로 읽자!