알고리즘 단계별로 풀어보기 : BOJ-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

내 풀이

import java.io.BufferedReader;

import java.io.BufferedWriter;

import java.io.IOException;

import java.io.InputStreamReader;

import java.io.OutputStreamWriter;

public class Main {

    public static void main (String args[]) {

        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));

        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));

        

        try {

            int num = Integer.parseInt(br.readLine());

            for(int i = 0; i < num; i++) {

                String[] str = br.readLine().split(" ");

                int sum = Integer.parseInt(str[0]) + Integer.parseInt(str[1]);

                bw.write(String.valueOf(sum));

                bw.newLine();

            }

            bw.flush();

            bw.close();

        } catch (NumberFormatException e) {

            // TODO Auto-generated catch block

            e.printStackTrace();

        } catch (IOException e) {

            // TODO Auto-generated catch block

            e.printStackTrace();

        }

        

    }

}

내 풀이 해석

BufferedReader를 사용해야 해야 된다는 의미의 문제입니다.

Scanner는 용량과 수행 시간이 많이 걸리므로 Buffered에 쌓아서 출력하는 방식으로 풀어야 합니다.

첫번째 입력 받은 수 만큼 반복문을 수행합니다.

반복문에서 두 수를 합산하여 버퍼에 저장하고 반복문이 종료되면 출력합니다.