[Python] sys.stdin.readline() 사용하기

알고리즘 문제를 풀다보면 시간 제한이라는 벽을 마주하게 된다. 
Python은 입력을 받기 위해 input()과 sys.stdin.readline()을 사용하는데 sys.stdin.readline()을 사용하면 더 빠르게 입력을 처리할 수 있다. 

 

Input( ) vs sys.stdin.readline( )

• input( )
  • input( ) 은 자동으로 개행 문자를 제거한다. (입력받은 문자열에 개행 문자만을 제거하여 줄 바꿈 없이 출력함)
  • 상대적으로 느리다. 
  • 더 이상 입력이 없는 경우에 실행되면 EOF 에러를 출력한다. 
• sys.stdin.readline( )
  • sys.stdin.readline( )은 개행 문자를 포함하여 반환한다.
  • 상대적으로 빠르다.
  • EOF를 만나도 에러를 반환하지 않고 빈 문자열을 반환한다. 

sys.stdin.readline( ) 사용법

sys.stdin.readline( )은 개행문자를 함께 입력 받기 때문에 처리를 해주어야 한다. 

1. 문자열을 입력받을 때

import sys

a = sys.stdin.readline() # 개행 문자 포함
b = sys.stdin.readline().rstrip("\n") # 개행 문자 미포함

 

sys.stdin.readline( ) 은 기본적으로 문자열을 반환하므로 문자열을 입력 받을 수 있다. 

개행 문자를 포함하기 때문에 개행 문자를 제외하고 받고 싶은 경우에는 strip( )을 통해 개행 문자를 제거할 수 있다.

2. N개의 문자열을 받아 리스트에 저장

import sys

n = int(sys.stdin.readline())
a = [sys.stdin.readline().strip() for _ in range(n)]

 

3. 정수를 입력 받을 때

import sys

a = int(sys.stdin.readline())

sys.stdin.readline( ) 은 문자열을 반환하기 때문에 명시적으로 형변환을 해주어야 한다. 

문자열을 int( )를 통해 형변환을 해주면 자동으로 개행 문자는 사라진다. 

 

4. 여러 개의 정수를 입력받기

import sys
a,b = map(int,sys.stdin.readline().split())