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

📌 문제

💯 정답

import sys
input = sys.stdin.readline

n, m = map(int, input().split())
mon = list(input().strip() for _ in range(n))	# 포켓몬 도감 입력
dic = {mon[i]: i+1 for i in range(len(mon))}	# 딕셔너리로 번호 주기
for _ in range(m):
    tmp = input().strip()
    if tmp.isalpha():			# 문자가 들어왔으면
        print(dic[tmp])			# 키값으로 value값(번호) 출력
    else:						# 숫자가 들어왔으면
        print(mon[int(tmp)-1])	# 도감 리스트에서 인덱스이용해 출력

📝 설명

• 입력받은 포켓몬 이름들로 도감 리스트 mon을 생성한다.
• 리스트에 들어있는 포켓몬이름을 key값으로, 번호를 value값으로 하는 딕셔너리 dic을 생성한다.
• 문자가 들어오면 키 값을 이용해 출력하고 숫자가 들어오면 인덱스를 이용해 출력한다.