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

-코드

from sys import stdin
arr = {}
arr2 = {}
n, m = map(int, stdin.readline().split())
for i in range(1, n + 1):
    name = stdin.readline().strip()
    arr[i] = name
    arr2[name] = i
for _ in range(m):
    q = stdin.readline().strip()
    if q.isdigit():
        print(arr[int(q)])
    else:
        print(arr2[q])

처음에는 isdigit을 이용해서 리스트에서 찾는 방법을 사용하려 했으나 문자가 나오는 경우 문제가 생겨서 딕셔너리로 변경하여 만들었습니다. 그러나 런타임 오류가 생겨서 찾아보니 strip로 해결이 되어서 사용해서 풀었습니다.