Python 입력 빠르게 받기

Python 입력 빠르게 받기

Python에서 입력을 받을 때는 input 함수를 사용한다.

a = input()

그런데 백준 사이트에서 문제를 풀 때 이 input 함수를 쓰면 시간 초과가 발생하는 경우가 있다. 15552번: 빠른 A+B 문제에 이에 대한 설명이 되어있는데 (링크 참고)

15552번: 빠른 A+B

이 경우에는 input 대신 sys.stdin.readline 함수를 사용해야한다.

import sys
a = sys.stdin.readline()

단, 이 때 맨 끝의 줄바꿈 문자(\n)까지 입력이 된다.

import sys
a = sys.stdin.readline()
print(a)

print로 출력해보면 줄바꿈 문자가 포함된 것을 볼 수 있다.
줄바꿈 문자는 제외하고 입력받고 싶으면 뒤에 .rstrip()을 붙이면 된다.

import sys
a = sys.stdin.readline().rstrip()
print(a)

이번에는 줄바꿈 문자가 포함되지 않았다.