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

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

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

 

---

풀이 과정

---

문제가 너무 길어서 다 문제인가 했는데 맨 마지막 부분만 문제다

1. 포켓몬의 개수와 번호를 리스트로 두는 경우에는 N, M이 100000까지 가능하므로 시간 초과가 나타날 수 있으므로 해시 방식을 사용해야 한다. 해시는 딕셔너리로 구현할 수 있다.
2. 두개의 딕셔너리를 사용한다.
   1. [포켓몬 이름] => 번호 
   2. [번호] => 포켓몬 이름
3. 먼저, 포켓몬 정보에 맞추어서 딕셔너리를 채워준다.
   • 인덱스를 하나씩 늘려가면서 포켓몬 이름에 대응되는 인덱스, 인덱스에 대응되는 포켓몬 이름을 채워준다.
4. 문제가 주어지면 우선 문제가 숫자인지 검사하고, 숫자라면 [인덱스] => 포켓몬 이름 형식의 딕셔너리에서 값을 가져오고, 아니라면 [포켓몬 이름] => 인덱스 형식의 딕셔너리에서 값을 가져와서 출력해주면 된다.

---

소스 코드

---

import sys

input = lambda : sys.stdin.readline().rstrip()

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

num_to_pokemon = {}
pokemon_to_num = {}
idx = 1

for _ in range(n):
    pokemon = input()
    num_to_pokemon[idx] = pokemon
    pokemon_to_num[pokemon] = idx
    idx += 1

for _ in range(m):
    question = input()
    if question.isdigit():
        print(num_to_pokemon[int(question)])
    else:
        print(pokemon_to_num[question])