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

Junyoung Park·2022년 3월 4일
0

코딩테스트

목록 보기
183/631
post-thumbnail

1. 문제 설명

나는야 포켓몬 마스터 이다솜

2. 문제 분석

딕셔너리에 저장된 키와 값을 반대로 만드려면 딕셔너리 컴프리헨션을 사용하자. dict_inv = {val:key for key, val in dict_original.items()} dictionary.items()는 키 값을 튜플 형태로 반환한다.

3. 나의 풀이

import sys

n, m = map(int, sys.stdin.readline().rstrip().split())
pokemon = {}

for i in range(1, n+1):
    pokemon[i] = sys.stdin.readline().rstrip()
pokemon_inv = {val:key for key, val in pokemon.items()}

for _ in range(m):
    question = sys.stdin.readline().rstrip()
    if question.isdigit():
        print(pokemon.get(int(question)))
    else:
        print(pokemon_inv.get(question))
profile
JUST DO IT

0개의 댓글