[1620번] 나는야 포켓몬 마스터 이다솜 Python🐍

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

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

알고리즘 [접근 방법]

이 문제는 dict를 사용해서 Key, value로 빠르게 접근하는 게 포인트다.

풀이

1. 포켓몬의 수 n과 입력받을 명령어 개수 m을 map형태로 입력받는다.
2. sys 라이브러리를 이용하여 입력을 받아야한다. (안그러면 시간초과)
3. sys라이브러리로 입력받을 경우 개행(\n)까지 dict에 입력되므로  rstrip() 처리를 해야한다.
4. dict에 키와 밸류를 입력할 때, 번갈아 2번 넣어줌으로써 포켓몬이든 숫자든 키밸류로 접근 가능하게 한다.
5. m번대로 문자열을 입력받을 때마다 숫자인지 아닌지 확인하여 각각 value값을 불러온다.

 

코드

import sys

input = sys.stdin.readline

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

dict = {}

for i in range(1, n + 1):
    a = input().rstrip()
    dict[i] = a
    dict[a] = i

for i in range(m):
    quest = input().rstrip()
    if quest.isdigit():
        print(dict[int(quest)])
    else:
        print(dict[quest])