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

👉 문제링크

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

---

---

🔸 문제 분석 🔸

• 포켓몬 이름을 입력받고, 번호 혹은 포켓몬 이름으로 검색, 출력한다.

🔸 코드 🔸

import sys

n, m = map(int, sys.stdin.readline().split())

book = {}
number = 1
for i in range(n):
    poket = sys.stdin.readline().rstrip()
    if poket not in book:
        book[poket] = number
        number += 1

book_reverse = dict(map(reversed, book.items()))
for i in range(m):
    find = sys.stdin.readline().rstrip()
    if find[0] < "0" or find[0] > "9":
        sys.stdout.write(str(book[find]) + "\n")
    else:
        sys.stdout.write(book_reverse[int(find)] + "\n")

🔸 코드 해석 🔸

• 포켓몬 이름으로 번호를 검색할 수 있도록 딕셔너리 자료구조를 사용했다.
• 검색할 내용이 포켓몬 이름(문자열)이면 번호를 출력하고, 번호가 입력되면 이름을 출력한다.
  • 번호로도 검색할 수 있도록 map()의 reversed를 이용해서 key와 value를 뒤집은 딕셔너리를 하나 더 만든다.
  • 입력된 값의 첫 문자가 숫자가 아니면, 전체를 문자로 보고 결과를 출력한다.

---

🔸 end 🔸

• 해시가 사용된 딕셔너리를 연습할 수 있는 문제였다.
• value로 key를 검색하는 방법은 반복문 보다, 뒤집힌 딕셔너리를 만들어 빠르게 처리할 수 있다는걸 배웠다.