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

Overview

BOJ 1620번 나는야 포켓몬 마스터 이다솜 Python 문제 풀이
분류: Data Structure (자료구조)

---

문제 페이지

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

문제 설명이 길지만 마지막 3줄만 읽어도 풀 수 있다.

---

풀이 코드

from sys import stdin


def input():
    return stdin.readline().rstrip()


n, m = map(int, input().split())
by_id = {}
by_name = {}
for i in range(1, n + 1):
    pokemon = input()
    by_id[i] = pokemon
    by_name[pokemon] = i

for _ in range(m):
    x = input()
    if x.isdigit():
        print(by_id[int(x)])
    else:
        print(by_name[x])

딕셔너리 2개에 포켓몬 정보를 저장한다. 딕셔너리 하나는 '포켓몬 번호-이름'으로, 다른 하나는 '포켓몬 이름-번호'로 매핑한다. 그리고 찾아야 하는 것이 포켓몬 번호인지, 이름인지에 따라 각 딕셔너리에서 찾아 출력한다.