백준 1620) 나는야 포켓몬 마스터 이다솜 [ Python, dictionary, sys.stdin.readline().strip() ]

---

분류 : 딕셔너리

표면적으로 딕셔너리를 사용해서 푸는 문제이지만, 두가지 해결해야 하는 이슈들이 더 있다.

1. Value 로 Key 값을 찾기.

2. input() 에서 발생하는 시간초과 문제 해결하기

---

keypoint : python input / value로 key 찾기 

---

code

import sys 
n,m = map(int,input().split(' '))

pocketmon_list = dict()
rev_poecketmon_list = dict()
cnt=1
for i in range(0,n):
    name = sys.stdin.readline().strip()
    pocketmon_list[str(cnt)] = name
    rev_poecketmon_list[name] = str(cnt)
    cnt+=1



for i in range(0,m):
    tmp_input = sys.stdin.readline().strip()
    if tmp_input.isdigit():
            print(pocketmon_list[tmp_input])
    else:
         print(rev_poecketmon_list[tmp_input])

---

중요한 내용 

1. Value 로 Key 찾기 

 

[Python] 파이썬 딕셔너리 value로 key 찾는 방법

 

 

위 글을 참고하기 바란다. 결론은 value로 key 를 직접 찾는 것은 for 문을 사용한 완전탐색밖에 없다. 

 

2. 왜 input() 이 sys.stdin.readline().stirp() 보다 느릴까?

input() 함수는 Python 에서 기본적으로 제공하는 사용자 입력 함수인데 다음과 같은 특징을 갖고 있다.

1. 입력 된 값을 '문자열로 반환' 하고 '자동으로 개행 문자 제거' 를 한다.

2. 프롬프트 메시지를 인자로 받을 수 있다.

 

여기서 이 문자열로 변환하여 반환하고 자동으로 개행 문자를 제거하는게 물리적으로 시간이 대단히 오래걸린다.

 

그에 반해 readline() 함수는 개행 문자를 포함하여 문자열을 반환한다. 그렇기 때문에 그 시간 차이가 발생하는데 readline 함수에서는 strip() 을 사용하여 개행문자를 지울 수 있다. 

 

두 함수의 시간차이를 보여주는 함수이다.

 

import sys
import time

# sys.stdin.readline() 사용
start = time.time()
for _ in range(100000):
    line = sys.stdin.readline().strip()
end = time.time()
print(f'sys.stdin.readline() 사용 시간: {end - start}초')

# input() 사용
start = time.time()
for _ in range(100000):
    line = input()
end = time.time()
print(f'input() 사용 시간: {end - start}초')

 

 

100000줄의 입력을 처리하는 데 걸리는 시간:
input() 함수: 12.3456초
sys.stdin.readline() 함수: 0.4567초

 

결과값은 어마어마하게 차이가 난다. 따라서 python 에서 시간초과문제를 겪을 때 input 을 sys.stdin.readline().strip() 으로 변경해보자.

 

---

import sys
input = sys.stdin.readline().strip

 

이렇게 하면 코드변경 없이도 기존 input 함수에 적용하여 사용 할 수 있다.