문제 링크

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

문제 설명

• 1번 포켓몬부터 N번 포켓몬까지 이름이 주어진다.
• 이후에 번호가 들어오면 이름을 출력
• 이름이 들어오면 번호가 출력

풀이

• 자료 구조를 아는지에 대한 문제같다
• 리스트와 딕셔너리를 각각 만든다
• 리스트에서는 숫자를 인덱스로 사용하고
• 딕셔너리에서는 문자열을 인덱스로 사용한다

코드

import sys


ipt = sys.stdin.readline
opt = sys.stdout.write
n, m = map(int, ipt().split())
pocketmon_list = ['']
pocketmon_dict = dict()
for i in range(1, n+1):
    name = ipt().rstrip()
    pocketmon_list.append(name)
    pocketmon_dict[name] = i
for i in range(m):
    temp = ipt().rstrip()
    if temp.isdigit():
        opt(f'{pocketmon_list[int(temp)]}\n')
    else:
        opt(f'{pocketmon_dict[temp]}\n')