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

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

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

📕 설명 📕

1. N개의 포켓몬 도감을 set에 넣는다.

 

2. 문제들을 받는다.

 

3. 문제를 순회하며 포켓몬 도감에서의 값을 가져온다.

 

=> 계속된 시간 초과를 해결하기 위하여 dictionary를 하나 더 만들어서 해결하였다.

🧑🏻‍💻 나의 풀이 🧑🏻‍💻

N, M = map(int ,input().split())
pokemon_encyclopedia_id = {}
pokemon_encyclopedia_name = {}
for i in range(N):
    pokemon_encyclopedia_id[i] = input()
    pokemon_encyclopedia_name[pokemon_encyclopedia_id[i]] = i
for i in range(M):
    problem = input()
    if problem.isdigit():
        print(pokemon_encyclopedia_id[int(problem)-1])
    else:
        print(pokemon_encyclopedia_name[problem]+1)