알고리즘 & SQL/백준
백준 1620 : 나는야 포켓몬 마스터 이다솜 (파이썬)
양을 좋아하는 문씨
2023. 3. 15. 01:09
너무 쉬운 문제인데 생각이 마치 어딘가 막힌 것 처럼 한 30분 넘게 붙잡고 있었던 것 같다.
import sys
N, M = map(int, sys.stdin.readline().split())
pokemon_name = {}
pokemon_num = {}
for i in range(1, N + 1):
name = sys.stdin.readline().strip()
pokemon_name[name] = str(i)
pokemon_num[str(i)] = name
for _ in range(M):
q = sys.stdin.readline().strip()
if q in pokemon_name:
print(pokemon_name[q])
else:
print(pokemon_num[q])
딕셔너리를 두 개 만들어서 푼다는 생각을 못하고 있었다...
가끔 이렇게 문제 지독하게 안 풀릴 때가 있다.
감기기운이 있는걸까??