백준) 15552번 빠른 A + B 문제 (파이썬)

문제) 본격적으로 for문 문제를 풀기 전에 주의해야 할 점이 있다.

 

입출력 방식이 느리면 여러 줄을 입력받거나 출력할 때 시간초과가 날 수 있다는 점이다.

 

Python을 사용하고 있다면, input 대신 sys.stdin.readline을 사용할 수 있다.

 

단, 맨 끝의 개행문자까지 같이 입력받기 때문에 문자열을 저장하고 싶을 경우 .rstrip()을 추가로 해 주는 것이 좋다.

 

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

 

테스트케이스를 하나 받은 뒤 하나 출력해도 된다.

 

import sys
X = int(input())

for i in range(X):
  a, b = map(int,sys.stdin.readline().split())
  print(a + b)

이번 문제는 파이썬을 입출력할 때 시간초과가 나는 것을 방지하기 위한 함수를 사용하는 것이다.

 

일단 기존 더하기 문제와 달리 import sys를 선언해야하고, map(int,input().split())에서

 

input자리에 sys.stdin.readline() 넣어 map(int,sys.stdin.readline().split())으로 사용한다.

 

앞으로 문제를 풀 때 많은 함수가 들어감에 따라 시간 초과가 나는 경우가 생길 것이다. 이번 기능은 잘 숙지하고,

 

잘 이용한다면 좀 더 빠른 입출력을 얻을 수 있다.

 

이해가 안되시거나 틀린 정보는 댓글로 알려주시면 감사하겠습니다!