[백준] 1620번: 나는야 포켓몬 마스터 이다솜 - python

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

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

 

실버 문제여서 무지성으로 1분만에 코드 작성했는데..

 

시간초과가 발생하였다.

 

정신차리고 해시맵으로 다시 작성 입력받을때 한번에 key, value 교차해서 주입시켰다.

n, m = map(int, input().split())
arr = []
for i in range(n):
    arr.append(input())

for j in range(m):
    answer = input()
    if answer in arr:
        for k in range(n):
            if arr[k] == answer:
                print(k+1)
    
    else:
        for k in range(n):
            if k == answer:
                print(arr[k])

 

완성