백준알고리즘 15552번 빠른 A + B C#

세부 정보

• 사이트: 백준알고리즘
• 번호: 15552
• 문제명: 빠른 A + B
• 언어: C#
• 분류: 수학, 구현, 사칙연산
• 비고: 문제풀기 팁 전수

 

문제

본격적으로 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()을 추가로 해 주는 것이 좋다.

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

 

입력

첫 줄에 테스트케이스의 개수 T가 주어진다. T는 최대 1,000,000이다. 다음 T줄에는 각각 두 정수 A와 B가 주어진다. A와 B는 1 이상, 1,000 이하이다.

 

출력

각 테스트케이스마다 A+B를 한 줄에 하나씩 순서대로 출력한다.

 

풀이

일반적인 Console.ReadLine()과 Console.WriteLine으로 입출력을 주고받으면 시간이 오래 걸리니 StreamReader와 StreamWriter를 사용하면 시간 초과를 피할 수 있다.

// 스트림 입력을 받을 준비를 한다.
StreamReader sr = new StreamReader(Console.OpenStandardInput());
// 스트림 출력을 할 준비를 한다.
StreamWriter sw = new StreamWriter(Console.OpenStandardOutput());

// 테스트 케이스의 개수를 입력받는다.
int testcase = Convert.ToInt32(sr.ReadLine());
// 테스트 케이스 입력값만큼의 크기를 가지는 배열을 생성한다.
int[] array = new int[testcase];

for (int i = 0; i < testcase; i++)
{
    // 공백으로 구분 후 문자열로 바꿔 배열에 값을 입력받는다.
    array = sr.ReadLine().Split().Select(int.Parse).ToArray();
    // 배열 내 값을 더해서 출력한다.
    sw.WriteLine(array[0] + array[1]);
}

// 스트림 입력을 종료한다.
sr.Close();
// 스트림 출력을 종료한다.
sw.Close();

 

후기

이러한 문제를 처음 접하는 사람들은 늘 하던 대로 Console로 입출력을 주고받는 스크립트를 작성할 것이다. 그러다가 시간 초과가 뜨면 당황을 할 것이다.

int testcase = Convert.ToInt32(Console.ReadLine());

int[] array = new int[testcase];

for (int i = 0; i < testcase; i++)
{
    array = Console.ReadLine().Split().Select(int.Parse).ToArray();
    Console.WriteLine(array[0] + array[1]);
}

 

나 또한 마찬가지였다. 위처럼 작성하고 시간 초과를 보았다.

하지만 Stream으로 입출력을 주고받는 방식은 더 빨랐다. 이 문제의 시간 제한이 1초인데 Stream으로 입출력을 주고받았더니 0.74초밖에 걸리지 않았다. 한마디로 빠르게 풀고 싶다면 Stream을 사용해야 한다. 만약에 Stream보다 더 빠른 방법이 있다면 그 방법대로 해도 된다.