URL : https://www.acmicpc.net/problem/11659
11659번: 구간 합 구하기 4
첫째 줄에 수의 개수 N과 합을 구해야 하는 횟수 M이 주어진다. 둘째 줄에는 N개의 수가 주어진다. 수는 1,000보다 작거나 같은 자연수이다. 셋째 줄부터 M개의 줄에는 합을 구해야 하는 구간 i와 j
www.acmicpc.net
[ RESULT ]
import java.io.BufferedReader
import java.io.BufferedWriter
import java.io.InputStreamReader
import java.io.OutputStreamWriter
fun main() {
val br = BufferedReader(InputStreamReader(System.`in`))
val bw = BufferedWriter(OutputStreamWriter(System.out))
val (N, M) = br.readLine().split(" ").map { it.toInt() }
val arr = br.readLine().split(" ").map { it.toInt() }
val prefixSum = IntArray(N + 1)
for (i in 1..N) {
prefixSum[i] = prefixSum[i - 1] + arr[i - 1]
}
repeat(M) {
val (i, j) = br.readLine().split(" ").map { it.toInt() }
val sum = prefixSum[j] - prefixSum[i - 1]
bw.write("$sum\n")
}
br.close()
bw.flush()
bw.close()
}
시간 초과가 지속적으로 발생하여서, 입출력을 최적화하고, 구간합을 미리 구함으로써 해결했다.
Baekjoon(Kotlin) - Num. ProblemName (0) | 2023.09.11 |
---|---|
Baekjoon(Kotlin) - 10986. 나머지 합 (0) | 2023.09.10 |
Baekjoon(Kotlin) - 11054. 가장 긴 바이토닉 부분 수열 (0) | 2023.09.08 |
Baekjoon(Kotlin) - 24416. 알고리즘 수업 - 피보나치 수 1 (0) | 2023.09.07 |
Baekjoon(Kotlin) - 11053. 가장 긴 증가하는 부분 수열 (3) | 2023.09.07 |