백준|1620번|나는야포켓몬마스터

README·2022년 7월 31일
0

파이썬 PS풀이

목록 보기
57/136

문제설명
포켓몬의 이름을 입력받고 난 뒤 그 포켓몬이 몇번째로 입력되었는지 또는 몇번째로 입력 된 포켓몬이 무엇인지 출력하는 문제입니다.

작동 순서
1. 두 개의 딕셔너리를 생성해서 하나는 번호를 키로, 하나는 이름을 키로 지정합니다.
2. 찾을 검색어를 입력받고 검색어가 숫자인경우 번호를 키로 지정한 딕셔너리에서 검색하고 검색어가 문자열인 경우 이름을 키로 지정한 딕셔너리에서 검색을 합니다.
3. 검색결과를 출력합니다.

소스코드

import sys
N, M = map(int, sys.stdin.readline().strip().split())
PocketMonster1 = {}
PocketMonster2 = {}
for i in range(1, N+1):
    name = sys.stdin.readline().strip()
    PocketMonster1[i] = name
    PocketMonster2[name] = i
for i in range(M):
    find = sys.stdin.readline().strip()
    if find.isnumeric():
        print(PocketMonster1[int(find)])
    else:
        print(PocketMonster2[find])

후기
굉장히 간단한 문제여서 딱히 할 말이 없는 것 같습니다. 딕셔너리에 키와 밸류 모두 검색하는 기능이 있으면 좋을텐데 없는게 아쉬웠습니다.

profile
INTP 개발자 지망생

0개의 댓글