[백준/Java] 15552번 빠른 A+B (반복문)

Jun 13, 2024

Contents

문제

본격적으로 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;
import java.util.StringTokenizer;

public class Main {
    public static void main(String[] args) throws IOException {
        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++) {
            StringTokenizer st = new StringTokenizer(br.readLine());
            int A = Integer.parseInt(st.nextToken());
            int B = Integer.parseInt(st.nextToken());

            int sum = A + B;
            bw.write(Integer.toString(sum));
            bw.newLine();
        }

        bw.flush();
        bw.close();

        br.close();
    }
}

풀이

◽ 공백 처리를 해줘야 하는 이유

readLine은 1줄을 읽기 때문에 혹시 입력되는 문자열에 공백이 발생할 경우, Integer.pareseInt가 실패할 수 있다. 공백은 정수로 파싱할 수 없는 문자로 간주하기 때문이다.

StringTokenizer는 공백을 구분자로 문자열을 분리한다. StringTokenizer를 호출하면, 공백이 아닌 값만을 올바르게 추출하여 Integer.pareseInt를 해도 성공할 것이다.

❔ StringTokenizer

공백을 구분자로 문자열을 분리한다.


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

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
        
        // 테스트케이스의 개수 T를 입력받는다.
        int T = Integer.parseInt(br.readLine());

        for (int i = 0; i < T; i++) {
		        // A와 B를 입력 받는다.
            StringTokenizer st = new StringTokenizer(br.readLine());
            int A = Integer.parseInt(st.nextToken());
            int B = Integer.parseInt(st.nextToken());
						
						// A와 B를 더한 결과를 출력한다.
            int sum = A + B;
            bw.write(Integer.toString(sum));
            bw.newLine();
        }

				// BufferedWriter를 flush하고 닫는다.
        bw.flush();
        bw.close();
        
				// BufferedReader를 flush하고 닫는다.
        br.close();
    }
}

See more posts

[백준/Java] 15552번 빠른 A+B (반복문)

문제

입력

출력

답

풀이

◽ 공백 처리를 해줘야 하는 이유

❔ StringTokenizer

More articles

[백준/Java] 11022번 A+B - 8 (반복문)

[백준/Java] 11021번 A+B - 7 (반복문)

[백준/Java] 25314번 코딩은 체육과목 입니다 (반복문)

[백준/Java] 25304번 영수증 (반복문)