백준 15552번 빠른: A+B
by Jm Park
문제
본격적으로 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를 한 줄에 하나씩 순서대로 출력한다.
예제 입력/출력
입력 | 출력 |
---|---|
5 1 1 12 34 5 500 40 60 1000 1000 |
2 46 505 100 2000 |
코드
#include <cstdlib>
#include <iostream>
#include <vector>
using namespace std;
int main() {
int n, a, b;
vector<int> ans;
scanf("%d", &n);
ans.reserve(n + 5); // 메모리 할당
for (int i = 0; i < n; i++) {
scanf("%d %d", &a, &b);
ans.push_back(a + b);
}
// 답 출력
for (vector<int>::iterator i = ans.begin(); i < ans.end(); i++)
cout << *i << '\n';
return 0;
}
풀이
문제는 어렵지 않다. 단순히 입력 받은 줄의 a,b 값을 더해서 배열 혹은 백터에 저장한 후 저장한 모든 값을 한번에 출력시키면 된다.
Subscribe via RSS