[BOJ/백준-Python]15552번 - 빠른 A+B

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

예제 출력 1

해결할 방법을 생각해보자.💡

10950번 문제와 똑같다. 다만, 설명에 적혀져있는 것처럼 일반적인 input()으로 입력을 받고 출력하면 시간이 오래걸린다는 것이다. 따라서, sys 함수를 호출하고 sys.stdin.readline을 input에 저장해준다. 그리고 10950번과 똑같이 풀면 된다!

앞으로 시간과 메모리를 적게 소비하고 싶을 때 sys.stdin.readline을 사용해야한다는 것을 꼭! 기억하자. 또한, 저걸로 받은 문자열은 개행문자를 포함한다. 개행문자는 우리가 알고 있는 다음 줄로 넘어가게 하는 \n이다. 하지만 정수형으로 바꿔주면 자연스럽게 개행문자는 사라지게 된다!

코드💻

import sys
input = sys.stdin.readline
num = int(input())
for i in range(num):
    x,y = map(int,input().split())
    print(x+y)

해결 로그

문제 푼 날짜	체크
2022-02-12	✔

2022 196
2021 22

맨 위로 이동 ↑