[JAVA] 백준 - 15552번 빠른 A+B

안녕하세요. pitang입니다.

백준 문제를 단계별로 풀고 있는데 기억해야 할 것들이 담긴 문제들을 포스팅해보려 합니다.

 

15552번 빠른 A+B

본격적으로 for문 문제를 풀기 전 주의해야 할 점이 있다.

입출력 방식이 느리면 여러 줄을 입력받거나 출력할 때 시간 초과가 날 수 있다는 점이다. (시간제한 1.5초)

Java를 사용하고 있다면, Scanner와 System.out.println 대신 BufferedReader 와 BufferedWriter를 사용할 수 있다.

BufferedWriter.flush 는 맨 마지막에 한 번만 하면 된다.

또한 입력과 출력 스트림은 별개이므로, 테스트 케이스를 전부 입력받아서 저장한 뒤 전부 출력할 필요는 없다.

테스트케이스를 하나 받은 뒤 하나 출력해도 된다.

 

첫 줄에 테스트케이스의 개수 T가 주어진다.

T는 최대 1,000,000 이다. 다음 T 줄에는 각각 두 정수 A와 B가 주어진다. A와 B는 1 이상 1,000 이하이다.

 

<예제 입력>	<예제 출력>
5 1 1 12 34 5 500 40 60 1000 1000	2 46 505 100 2000

 

-> 처음 이 문제를 보고 이전 단계에서 BufferedReader와 BufferedWriter를 사용해본 적이 없어 어려움을 겪었다.

---

풀이

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 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");
		}
		
		br.close();
		
		bw.flush();
		bw.close();
		
	}
	
}

-> 입력은 BufferedReader.readLine()으로 받고,

     문자열 분리를 위해 StringTokenizer을 생성과 동시에 입력받고

     반환되는 타입이 String 이므로 Integer.parseInt를 통해 int형으로 바꿔준 뒤, 더해준 값을 BufferedWriter.write()에 넣어주었다.

 

    (참고 - flush를 해준 후, close를 해줘야 한다.)

 

-> 888ms의 시간이 걸리고 코드 길이는 740B이다.

 

 

 

감사합니다. 

 

*m1 맥북을 사용 중입니다.*