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

---

Python 풀이 (빠른 입출력 사용)

Python에서 많은 양의 데이터를 처리할 때는 sys.stdin.read와 sys.stdout.write를 사용하는 것이 좋습니다.

import sys

# 입력받기 (빠른 입력)
input = sys.stdin.read

# 입력 전체를 한 번에 읽은 후 줄 단위로 분리
data = input().splitlines()

# 첫 번째 줄에 테스트 케이스 개수 T
T = int(data[0])

# 결과를 담을 리스트
results = []

# 각 테스트 케이스 처리
for i in range(1, T + 1):
    A, B = map(int, data[i].split())
    results.append(A + B)

# 빠른 출력
sys.stdout.write("\n".join(map(str, results)) + "\n")

 

Java 풀이 (BufferedReader와 BufferedWriter 사용)

Java에서는 많은 데이터를 처리할 때 Scanner 대신 BufferedReader와 BufferedWriter를 사용하는 것이 더 빠릅니다.

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.io.IOException;

public class Main {
    public static void main(String[] args) throws IOException {
        // 빠른 입출력을 위한 BufferedReader와 BufferedWriter
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));

        // 테스트 케이스 개수 입력받기
        int T = Integer.parseInt(br.readLine());

        // 각 테스트 케이스 처리
        for (int i = 0; i < T; i++) {
            String[] numbers = br.readLine().split(" ");
            int A = Integer.parseInt(numbers[0]);
            int B = Integer.parseInt(numbers[1]);
            bw.write((A + B) + "\n");
        }

        // 출력
        bw.flush();
        bw.close();
    }
}

 

C++ 풀이 (cin/cout을 sync 해제하여 속도 개선)

C++에서는 cin과 cout의 동기화를 해제하여 속도를 빠르게 할 수 있습니다.

#include <iostream>
using namespace std;

int main() {
    ios::sync_with_stdio(false);
    cin.tie(NULL);
    
    // 테스트 케이스 개수 입력받기
    int T;
    cin >> T;
    
    // 각 테스트 케이스 처리
    for (int i = 0; i < T; i++) {
        int A, B;
        cin >> A >> B;
        cout << A + B << "\n";
    }

    return 0;
}

 

JavaScript 풀이 (readline 사용)

JavaScript에서는 readline을 사용하여 입력을 빠르게 처리할 수 있습니다.

const readline = require('readline');
const rl = readline.createInterface({
    input: process.stdin,
    output: process.stdout
});

const input = [];
rl.on('line', line => {
    input.push(line);
}).on('close', () => {
    const T = parseInt(input[0]);
    let result = '';
    
    // 각 테스트 케이스 처리
    for (let i = 1; i <= T; i++) {
        const [A, B] = input[i].split(' ').map(Number);
        result += (A + B) + '\n';
    }
    
    // 출력
    process.stdout.write(result);
});

 

C 풀이 (scanf와 printf 사용)

C언어에서는 scanf와 printf를 사용하여 빠르게 입출력할 수 있습니다.

#include <stdio.h>

int main() {
    int T, A, B;

    // 테스트 케이스 개수 입력받기
    scanf("%d", &T);

    // 각 테스트 케이스 처리
    for (int i = 0; i < T; i++) {
        scanf("%d %d", &A, &B);
        printf("%d\n", A + B);
    }

    return 0;
}