빠른 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 이하이다.
BufferedReader / BufferedWriter은 버퍼를 이용해서 읽고 쓰는 함수로, 모아뒀다가 한 번에 전송하므로 입출력 효율을 높일 수 있다.
readLine()을 사용해 데이터를 라인 단위로 읽을 수 있는데, 리턴 값이 String이라서 다른 타입으로 입력받을 땐 형변환을 해줘야 한다.
사용한 후엔 close를 이용해 자원해제 해주도록 하자.
import java.io.*;
public class Main {
public static void main(String[] args) throws NumberFormatException, IOException {
// BufferReader/BufferWriter 생성
BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
int cnt = Integer.parseInt(br.readLine());
String input;
int num1, num2;
for (int i = 0; i < cnt; i++) {
input = br.readLine();
num1 = Integer.parseInt(input.split(" ")[0]);
num2 = Integer.parseInt(input.split(" ")[1]);
bw.write(num1 + num2+"\n");
}
br.close();
bw.flush();
bw.close();
}
}
참조
'6. 자료구조 & 알고리즘 > 6-2. 백준' 카테고리의 다른 글
백준 - [11654번: 아스키 코드] (0) | 2022.06.15 |
---|---|
백준 - [4673번: 셀프 넘버] (0) | 2022.06.15 |
백준 - [10818번 : 최소, 최대] (0) | 2022.06.10 |
백준 - [10951번: A+B -4] (0) | 2022.06.09 |
백준 코드 깃허브에 자동 업로드 (0) | 2022.06.09 |