11659번: 구간 합 구하기 4

문제 출처 : https://www.acmicpc.net/problem/11659

 

언어 : Kotlin

 

문제 설명 :

수 N개가 주어졌을 때, i번째 수부터 j번째 수까지 합을 구하는 프로그램을 작성하시오.

• 첫째 줄에 수의 개수 N과 합을 구해야 하는 횟수 M이 주어진다. 둘째 줄에는 N개의 수가 주어진다. 수는 1,000보다 작거나 같은 자연수이다. 셋째 줄부터 M개의 줄에는 합을 구해야 하는 구간 i와 j가 주어진다.
• 총 M개의 줄에 입력으로 주어진 i번째 수부터 j번째 수까지 합을 출력한다.

 

제한 사항 :

• 시간 제한 : 1초
• 메모리 제한 : 256MB

 

• 1 ≤ N ≤ 100,000
• 1 ≤ M ≤ 100,000
• 1 ≤ i ≤ j ≤ N

 

입출력 예 :

입력	출력
5 3 5 4 3 2 1 1 3 2 4 5 5	12 9 1

풀이 : 

import java.io.*
import java.util.StringTokenizer

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 = IntArray(n + 1) { 0 }
    StringTokenizer(br.readLine()).run { for (i in 1 .. n) { arr[i] = arr[i - 1] + nextToken().toInt() } }
    repeat(m) {
        val (i, j) = br.readLine().split(" ").map { it.toInt() }
        bw.appendLine("${arr[j] - arr[i - 1]}")
    }
    bw.flush()
    bw.close()
}