[🥉2 / 백준 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를 한 줄에 하나씩 순서대로 출력한다.

 

예제 입력  5 1 1 12 34 5 500 40 60 1000 1000

예제 출력  2 46 505 100 2000

 

풀이

 

들어오는 입력이 아주 많기 때문에, 일반적인 파이썬의 input이나 자바의 scanner로는 처리가 불가능하다.

따라서 입력을 한 줄로 받아 빠르게 처리할 수 있는 방법을 사용해야 한다.

 

파이썬의 경우는 sys 모듈의 sys.stdin.readline 함수를 통해 개행 문자를 포함하여 한 줄씩 입력을 받을 수 있다.

문제 설명에도 나와있지만, map이나 int 함수로 형변환을 할 경우 외에 문자열 그대로 입력받고 싶은 경우엔 개행 문자가 그대로 남아있으므로 rstrip() 함수로 개행 문자를 지워주도록 하자.

 

import sys
input = sys.stdin.readline
for _ in range(int(input())):
    print(sum(map(int,input().split())))

 

자바의 경우는 BufferedWriter와 BufferedReader를 이용하여 문자열을 처리한다.

BufferedWriter 대신 StringBuilder를 사용하기도 한다.

 

import java.io.*;
import java.util.*;
public class Main {

	public static void main(String args[]) throws Exception{
		BufferedReader br = new BufferedReader (new InputStreamReader(System.in));
		BufferedWriter bw = new BufferedWriter (new OutputStreamWriter(System.out));
		
		int n = Integer.parseInt(br.readLine());
		
		StringTokenizer st;
		for (int i=0;i<n;i++) {
			st = new StringTokenizer(br.readLine());
			bw.write(Integer.parseInt(st.nextToken()) + Integer.parseInt(st.nextToken()) + "\n");
		}
		
		bw.flush();
		
	}
}