[코딩테스트 연습(Python)] 백준 1620번_나는야 포켓몬 마스터 이다솜

import sys

input = sys.stdin.readline

n, m = map(int, input().split())
pocket = {}

for i in range(n):
    name = input().rstrip()
    pocket[i+1] = name

# reversed로 dictionary의 key와 value를 바꿔줄 수 있음
pocket_reversed = dict(map(reversed, pocket.items()))
    
for _ in range(m):
    x = input().rstrip()
    
    # isdigit()으로 숫자 형식인지 판별
    if x.isdigit():
        x = int(x)
        print(pocket[x])
    else:
        print(pocket_reversed[x])