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

문제

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

C++을 사용하고 있고 cin/cout을 사용하고자 한다면, cin.tie(NULL)과 sync_with_stdio(false)를 둘 다 적용해 주고, endl 대신 개행문자(\n)를 쓰자. 단, 이렇게 하면 더 이상 scanf/printf/puts/getchar/putchar 등 C의 입출력 방식을 사용하면 안 된다.

Java를 사용하고 있다면, Scanner와 System.out.println 대신 BufferedReader와 BufferedWriter를 사용할 수 있다. BufferedWriter.flush는 맨 마지막에 한 번만 하면 된다.

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

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

 

 

 

결과

 

 

문제 풀이

import sys

input = sys.stdin.readline()

num = int(input)

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

 

 

새롭게 알게 된 것

기존에 파이썬에서 input() 을 써서 입력문자를 받았었는데, 이 방식은 간단하지만 실행 시간이 다소 느린점이 있었다.하지만 반복문으로 여러줄 입력받는 상황에서는 반드시 sys.stdin.readline()을 사용해야 시간초과가 발생하지 않는다.

sys.stdin.readline() 를 쓰기 전에는 import sys를 선언해주는 것을 잊지 말자.

문자열을 저장하고 싶을 경우 .rstrip()을 추가로 해 주는 것이 좋다.