[백준 1620] 나는야 포켓몬 마스터 이다솜(실버 4) 문제 풀이- 파이썬 python

0. 자세한 설명은 YouTube 영상으로

 

1. 해시를 활용한 풀이

# 0. 입력 및 초기화
import sys
input = sys.stdin.readline
N, M = map(int, input().split())

# 1. N개의 정보를 hash에 반영
idx_to_name = {}
name_to_idx = {}
for i in range(1, N + 1):
    name = input().rstrip()
    idx_to_name[i] = name
    name_to_idx[name] = i

# 2. M개의 쿼리를 해시를 통해 출력
for _ in range(M):
    query = input().rstrip()
    if query.isdigit():
        print(idx_to_name[query])
    else:
        print(name_to_idx[query])