연습장/백준(BOJ) 문제풀이
[백준 11659] 구간 합 구하기 4 with Node.js
Tesseractjh
2021. 8. 9. 12:47
문제 링크
https://www.acmicpc.net/problem/11659
11659번: 구간 합 구하기 4
첫째 줄에 수의 개수 N과 합을 구해야 하는 횟수 M이 주어진다. 둘째 줄에는 N개의 수가 주어진다. 수는 1,000보다 작거나 같은 자연수이다. 셋째 줄부터 M개의 줄에는 합을 구해야 하는 구간 i와 j
www.acmicpc.net
풀이
const input = require('fs').readFileSync('/dev/stdin').toString().trim().split('\n');
const arr = input[1].split(' ').map(v => +v);
const cumsum = new Array(arr.length+1).fill(0);
const output = [];
arr.forEach((v, i) => {
cumsum[i+1] = cumsum[i] + v;
});
input.slice(2).forEach(ij => {
const [i, j] = ij.split(' ').map(v => +v);
output.push(cumsum[j]-cumsum[i-1]);
});
console.log(output.join('\n'));
처음에 무작정 arr의 i번째 부터 j번째 수를 더하여 출력하였더니 시간 초과가 났다.
cumsum 배열을 만들어서 arr의 누적합을 저장한다.
cumsum[j]에서 cumsum[i-1]을 뺀 값이 곧 arr의 i~j까지의 합과 같다.