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

1. 문제
   

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

2. 풀이
   포켓몬 도감 번호에 해당하는 경우는 딕셔너리를 통해 저장을 하고 포컷몬리스트 같은 경우 따로 배열로 저장한 후 숫자라면 리스트의 인덱스를 이용해 조회하고 아니라면 키값을 조회하도록 구현했다.
3. 코드

최종코드

import sys

n,m = map(int,sys.stdin.readline().split())
pocketmon = {}
pocketmonlist = []

for i in range(n):
    name = sys.stdin.readline().strip()
    pocketmonlist.append(name)
    pocketmon[name] = i + 1

for i in range(m):
    question = sys.stdin.readline().strip()
    if question.isdigit():
        print(pocketmonlist[int(question) - 1])
    else:
        print(pocketmon[question])

틀린코드

//	시간초과가 났다.
import sys

n,m = map(int,sys.stdin.readline().split())
pocketmonlist = []

for i in range(n):
    name = sys.stdin.readline().strip()
    pocketmonlist.append(name)

for i in range(m):
    question = sys.stdin.readline().strip()
    if question.isdigit():
        print(pocketmonlist[int(question) - 1])
    else:
        print(pocketmonlist.index(question) + 1)