파이썬) 백준 알고리즘 | 1620번 : 나는야 포켓몬 마스터 이다솜

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

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

백준 1620 '나는야 포켓몬 마스터 이다솜' 문제입니다.

dictionary 자료형을 이용하면 쉽게 풀 수 있습니다.

dictionary에 저장해두고 원하는 포켓몬을 찾아주면 됩니다.

import sys
input = sys.stdin.readline
n, m = map(int, input().split())
s1 = dict()
s2 = dict()
for i in range(n):
    name = input().strip()
    s1[name] = i+1
    s2[i+1] = name
for i in range(m):
    question = input().strip()
    try:
        print(s2[int(question)])
    except:
        print(s1[question])