[Baekjoon] 15552. 빠른 A+B - 파이썬 / readline()

문제

본격적으로 for문 문제를 풀기 전에 주의해야 할 점이 있다. 입출력 방식이 느리면 여러 줄을 입력받거나 출력할 때 시간초과가 날 수 있다는 점이다.

Python을 사용하고 있다면, input 대신 sys.stdin.readline을 사용할 수 있다. 단, 이때는 맨 끝의 개행문자까지 같이 입력받기 때문에 문자열을 저장하고 싶을 경우 .rstrip()을 추가로 해 주는 것이 좋다.

또한 입력과 출력 스트림은 별개이므로, 테스트케이스를 전부 입력받아서 저장한 뒤 전부 출력할 필요는 없다. 테스트케이스를 하나 받은 뒤 하나 출력해도 된다.

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

 

해결

주의

- 입력데이터의 개수가 많을 때 input()을 사용하면 느려서 시간초과로 인한 오답판정을 받을 수 있다.

해결법

- sys.stdin.readline()함수를 사용하면 빠른 입력이 가능하다.

import sys

data = sys.stdin.readline().rstrip()

sys.stdin.readline()은 개행문자(\n)를 포함하여 입력받는다. 따라서 rstrip()함수로 줄바꿈문자를 제거해주어야 한다.

 

또는

n = sys.stdin.readline()[:-1]

 

 

 

import sys
input = sys.stdin.readline
data = input().rstrip()

sys.stdin.readline을 매번 입력하는게 귀찮을때는 변수에 함수를 저장하여 간편하게 사용한다.

 

 

해결 코드

import sys
input = sys.stdin.readline

num = int(input())
for i in range(num):
    a, b = map(int, input().rstrip().split())
    print(a+b)

 

참고

https://computer-science-student.tistory.com/749

[파이썬, Python] 빠른 입출력