2023.01.18 나는야 포켓몬 마스터 이다솜[파이썬]

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

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

 

import sys
from collections import defaultdict

input = sys.stdin.readline

n,m = map(int,input().split())

ency = defaultdict(str)

for i in range(1,n+1):
    names = input().strip()
    ency[names] = str(i)
    ency[str(i)] = names
for _ in range(m):
    finds = input().strip()
    print(ency[finds])

문제부터가 엄청 눈을 끌어서 다 받고 index 하고 찾고 하면 무조건 시간 초과 나게 나올거 같아서 처음부터 그냥 dict에 숫자랑 name을 둘 넣어서 바로 값을 넣어서 나오게 만들었다.