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

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

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

import sys
m, n = map(int, sys.stdin.readline().split())
pkmn = []
pkmn_dic = {}

for i in range(m):
    pk = sys.stdin.readline().rstrip()
    pkmn.append(pk)
    pkmn_dic[pk] = i + 1

for _ in range(n):
    search = sys.stdin.readline().rstrip()
    if search.isdigit():
        print(pkmn[int(search) - 1])
    else:
        print(pkmn_dic[search])

 

해설

딕셔너리나 isdigit()과 같은 내장 함수에 대한 기본 개념 복기를 시켜준 문제였다.

입력과 관련해서 rstrip() 때문에 key error가 발생했다.. 검색해서 찾아본 결과 오른쪽 공백을 제거하는 rstrip()을 입력에 사용해야했다. 여기서 꽤 시간을 낭비했다..😥