Save my data

백준 1620 : 나는야 포켓몬 마스터 이다솜 (파이썬) 본문

알고리즘/백준

백준 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])

딕셔너리를 두 개 만들어서 푼다는 생각을 못하고 있었다...


가끔 이렇게 문제 지독하게 안 풀릴 때가 있다.

감기기운이 있는걸까??

Comments