Baekjoon(Kotlin) - 11659. 구간 합 구하기 4

KDLiam 2023. 9. 9. 14:36

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()
}

시간 초과가 지속적으로 발생하여서, 입출력을 최적화하고, 구간합을 미리 구함으로써 해결했다.

저작자표시 (새창열림)