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

📖 문제

📖 예제

📖 풀이

import sys
input = sys.stdin.readline

dict = {}

N, M = map(int, input().split())
for i in range(1, N+1):
  name = input().strip()
  dict[name] = i
  dict[i] = name

for _ in range(M):
  name = input().strip()
  if name.isdigit():
    print(dict[int(name)])
  else:
    print(dict[name])

문제에 정성이 정말 많이 담겨있다. . .
딕셔너리 자료구조를 이용해 입력으로 받은 것을 두 가지 경우로 저장한다.
출력에 대해서는 입력으로 받은 것이 숫자일 경우 문자를 출력하고, 문자일 경우 이에 해당하는 숫자를 출력하게 한다.