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

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

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

문제가 어마무시하게 길다.

안읽어도 된다.

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

문제 접근

문제가 길지만 풀기쉬운 문제. dictionary로 풀면 쉽게 풀린다.

마음같아선 양쪽에 다 key를 사용하고싶지만 dict는 key-value형태로 되어있다.

그래서 번호로 입력하던 포켓몬 이름으로 입력하던 둘 다 출력할 수 있게 하기위해서 dict를 2개 사용했다.

하나는 이름을 저장해두고 하나는 번호를 저장해서 만들었다.

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

코드

import sys
input = sys.stdin.readline

"""

"""

N, M = map(int,input().split())
dict_name = {}
dict_num = {}
for i in range(1,N+1):
  pokemon = input().rstrip()
  dict_name[i] = pokemon
  dict_num[pokemon] = i

for _ in range(M):
  x = input().rstrip()
  if x.isdigit():
    print(dict_name[int(x)])
  else:
    print(dict_num[x])

256ms