🟩 [백준] [Python] Class3_1620번_ 나는야 포켓몬 마스터 이다솜

문제

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

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

풀이

# 백준 1620번 문제 - 나는야 포켓몬 마스터 이다솜
import sys

n, m = map(int, input().split())

# 리스트에 포켓몬 저장
pokedic_int_key = {}  # Key값이 int인 dictionary
pokedic_name_key = {}  # Key값이 str인 dictionary
for i in range(n):
    name = sys.stdin.readline().strip()
    pokedic_int_key[i] = name
    pokedic_name_key[name] = i

# 포켓몬 탐색
for _ in range(m):
    item = sys.stdin.readline().strip()
    # 입력값이 int일 경우, key값이 int인 dictionary에서 value를 가져옴
    if item.isdigit() == True:  # isdigit -> O(n)
        print(pokedic_int_key[int(item)-1])
    # 입력값이 int가 아닐 경우 (문자열일경우), key값이 str인 dictionary에서 value를 가져옴
    else:
        print(pokedic_name_key[item]+1)

처음에 시간초과가 떠서 생각을 더 해본 결과 딕셔너리를 사용하여 구현하면 시간복잡도가 줄어들기 때문에 그렇게 구현하였다.