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

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

풀이 코드 및 설명

import sys

# 포켓몬의 개수 N과 문제의 개수 M을 입력받음
N, M = map(int, sys.stdin.readline().split())

# 포켓몬 이름과 번호를 매핑하는 두 딕셔너리
name_to_number = {}
number_to_name = {}

# 포켓몬 이름을 입력받고, 두 딕셔너리에 저장
for number in range(1, N + 1):
    name = sys.stdin.readline().strip()
    name_to_number[name] = number
    number_to_name[number] = name

# 문제를 하나씩 입력받아 답을 출력
for _ in range(M):
    query = sys.stdin.readline().strip()
    if query.isdigit():  # 숫자인 경우
        print(number_to_name[int(query)])
    else:  # 문자인 경우
        print(name_to_number[query])

각각 번호 - 포켓몬 이름(문자열) 로 저장하면 좋을 것 같은 문제이다. 이는 딕셔너리를 사용하면 쉽게 풀 수 있다.

문자일 경우, 포켓몬 이름이기에 숫자를 출력하고 숫자일 경우, 해당하는 숫자의 포켓몬 이름을 출력한다.