[백준/Python3] 15552번: 빠른 A+B

문제

본격적으로 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()을 추가로 해 주는 것이 좋다.

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

자세한 설명 및 다른 언어의 경우는 이 글에 설명되어 있다.

이 블로그 글에서 BOJ의 기타 여러 가지 팁을 볼 수 있다.

입력

첫 줄에 테스트케이스의 개수 T가 주어진다. T는 최대 1,000,000이다. 다음 T줄에는 각각 두 정수 A와 B가 주어진다. A와 B는 1 이상, 1,000 이하이다.

출력

각 테스트케이스마다 A+B를 한 줄에 하나씩 순서대로 출력한다.

예제 입력 1 복사

5
1 1
12 34
5 500
40 60
1000 1000

예제 출력 1 복사

2
46
505
100
2000

---

 

Python
rstrip을 하라는 건 문자열 자체를 변수에 저장하고 싶을 때 얘기지, 개행문자가 맨 끝에 들어와도 int 변환이나 split()을 그대로 할 수 있습니다. 즉 int(sys.stdin.readline()), sys.stdin.readline().split() 이렇게 해도 아무 문제 없습니다. 참고로 이름이 꽤 길기 때문에 저는 input = sys.stdin.readline을 맨 처음에 함으로써 쓰는 편입니다.

jh05013 님께서 작성하신 글 중 일부입니다.

---

저는 백준 문제를 풀때 jupyter Notebook 을 사용하는데... 이게 sys.stdin.readline 을 사용하면 계속 오류가 생깁니다. 

그래서 BOJ 에 질문해서 물어봤는데...

lcr7324   15시간 전
Anaconda 환경에서 테스트하고 계시다면, Anaconda IPython 콘솔에서 sys.stdin.readline이 제대로 동작하지 않기 때문에 발생하는 오류일 겁니다.로컬에서 테스트하실 때에는 input으로 하시고 백준에 제출할 때만 sys.stdin.readline으로 바꾸시거나, I/O redirection을 통해 파일 입출력으로 바꿔주어 테스트하는 방법이 있습니다..

그렇다네요

import sys
n = int(input(""))
input = sys.stdin.readline
for i in range(n):
    A,B = map(int, input().rstrip().split())
    print(A+B)