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

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

 

 

#Solution

딕셔너리를 2개 만들어서 풀 수 있다. (매.우.간.단)

'출력형식이 잘못되었습니다'를 5번 정도 받았는데... 입력 문자 양 옆에 공백이 있다. strip으로 벗겨준 뒤 입력을 받고 출력하자.

import sys

N, M = map(int, input().split())
number_pokemon = 1
pokemon_dict1 = {}
pokemon_dict2 = {}

for _ in range(N):
    name = str(sys.stdin.readline()).strip()
    pokemon_dict1[number_pokemon] = name
    pokemon_dict2[name] = number_pokemon
    number_pokemon += 1

answer = []
for _ in range(M):
    pokemon = str(sys.stdin.readline()).strip()
    try:
        print(pokemon_dict1[int(pokemon)])
    except:
        print(pokemon_dict2[pokemon])