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

문제 링크

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

 

문제 풀이

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

pokemon_dict = {}

for i in range(n):
  pokemon = sys.stdin.readline().strip()
  pokemon_dict[i + 1] = pokemon
  pokemon_dict[pokemon] = i + 1

for i in range(m):
  test = sys.stdin.readline().strip()
  if test.isdigit() == True:
    print(pokemon_dict[int(test)])
  elif test.isalpha() == True:
    print(pokemon_dict[test])

1620 입출력 예시

*key point: 리스트를 이용해서도 구현할 수 있지만 시간 초과를 해결하기 위해 딕셔너리 자료형을 이용한다. 번호를 입력하면 이름이, 이름을 입력하면 번호가 나오게 해야하므로 key와 value가 정반대인 데이터도 포함하도록 딕셔너리를 만든다.