[Python] 빠르게 입력받기 readline( )

코딩 테스트>> 데이터의 개수가 1,000만개를 넘어가거나 탐색 범위의 크기가 1,000억 이상일 경우 등 입력 데이터가 많은 문제에 input() 함수를 사용하면 동작 속도가 느려서 시간 초과로 오답 판정을 받을 수 있다. 

 

이처럼 입력 데이터가 많은 문제는 sys 라이브러리의 readline() 함수를 이용하면 시간 초과를 피할 수 있다. 

 

readline()

import sys
# 하나의 문자열 데이터 입력받기
input_data = sys.stdin.readline().rstrip()

# 입력받은 문자열 그대로 출력
print(input_data)

 

rstrip() : 입력의 맨 마지막 공백 문자를 제거해준다. 

 

Ref.

이것이 취업을 위한 코딩 테스트다 with 파이썬