[Python] 백준 #1620. 나는야 포켓몬 마스터 이다솜

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

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

 

# 1620 - Python 3

import sys

input = sys.stdin.readline

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

dic = {input().strip(): i for i in range(1, n+1)}   # 시간 복잡도를 줄이기 위해 dictionary 사용
reverse_dic = dict(map(reversed, dic.items()))      # 숫자로 입력되는 경우를 위해 reverse된 새로운 
                                                    # dictionary도 선언
for i in range(m):
    x = input().strip()

    if x in dic.keys():                             # input 값이 dic의 key값과 같은 경우
        print(dic[x])
    else:                                           # input 값이 dic의 value값과 같은 경우
        print(reverse_dic[int(x)])                  # input 값이 str로 입력 된 정수이므로 int로 변환

< line 7 ~ 8 >

- 시간 복잡도를 줄이기 위해 dictionary 사용

- input이 정수인 경우를 대비하기 위해 key값이 정수인 reversed된 새로운 dictionary도 선언

 

< line 16 >

- x(input된 값)이 str로 입력된 정수인 경우, int(x)를 통해 정수로 변환