16-SeongHoonC

[SeongHoonC]

🔗 문제 링크

구간 합 구하기 4

✔️ 소요된 시간

40분

✨ 수도 코드

첫번째 시도

리스트를 잘라서 총합 구했더니 당연히 시간초과가 발생했습니다.
최대 100000(N) * 100000(M) 번 연산이 실행될 수 있기 때문이었죠

import java.io.BufferedReader
import java.io.InputStreamReader

fun main() {
    val br = BufferedReader(InputStreamReader(System.`in`))
    val (n, m) = br.readLine().split(" ").map { it.toInt() }
    val nums = br.readLine().split(" ").map { it.toInt() }
    for (i in 0 until m) {
        val (start, end) = br.readLine().split(" ").map { it.toInt() }
        println(nums.subList(start - 1, end).sum())
    }
}

못풀겠다 싶어서 유형을 봤는데 누적합이었습니다.

아.. 5 번째까지 누적합에 2 번째까지 누적합을 빼면 3~5 구간합이 나오는 것이었습니다.

1. 누적합 배열을 만들고 각 자리까지의 누적합을 구한 후 (연산 N번)
2. M 번 만큼 구간합을 구해 출력합니다.

이러면 N + M 번 연산을 수행할 것으로 예상됩니다.

print 도 시간이 오래걸린다길래 BufferedWriter 에 write 하고 한번에 flush 했습니다.

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 nums = br.readLine().split(" ").map { it.toInt() }
    val addedSum = Array(n + 1) { 0 }
    addedSum[1] = nums[0]

    // i 까지 누적합 구하기
    for (i in 2..n) {
        addedSum[i] = addedSum[i - 1] + nums[i - 1]
    }

    // 누적합 end 에서 누적합 start - 1 를 빼서 start ~ end 구간 합 구하기
    for (i in 0 until m) {
        val (start, end) = br.readLine().split(" ").map { it.toInt() }
        bw.write((addedSum[end] - addedSum[start - 1]).toString())
        bw.newLine()
    }
    // BufferedWriter 로 한번에 출력
    bw.flush()
    bw.close()
}

📚 새롭게 알게된 내용

누적합을 이해하게 되었네요.
알면 쉬운데 모르면 떠올리기 어려운 문제같아서 공유했습니다.

[fnzksxl]

아 이 문제 오랜만이네요 ㅋㅋㅋㅋㅋ

똑같은 로직으로 풀었던 문제입니다!

import sys

input = sys.stdin.readline

N, M = map(int, input().split())

lst = list((map(int, input().split())))
sums = [0] * (N+1)

for i in range(N):
    sums[i+1] = sums[i] + lst[i]

for _ in range(M):
    i, j = map(int, input().split())
    print(sums[j] - sums[i-1])

[alstjr7437]

헉 저도 태원님이랑 비슷한 코드로 작성되는데 temp만 추가되서 했네요!!

import sys
input = sys.stdin.readline
 
m, n = map(int, input().split())
arr = list(map(int, input().split()))
answer = [0]    
 
temp = 0    
for i in arr:  
    temp += i
    answer.append(temp)
 
# print(answer)

for i in range(n):
    a, b = map(int, input().split())
    print(answer[b] - answer[a-1])

[wkdghdwns199]

처음에 문제 완전 쉽군 이랬다가 시간초과 뜨고 당황... 후에 다른 분들 코드보고 머리가 띵했네요 ㅋㅋㅋ
누적합 개념으로 한 수 배웁니다!

import sys
input = sys.stdin.readline
N,M = map(int, input().split())
num_list = list(map(int, input().split()))
accumulate_list = [0] * (N+1)
for i in range(N):
    accumulate_list[i+1] = accumulate_list[i] + num_list[i]

for _ in range(M):
    start, end = map(int, input().split())
    print(accumulate_list[end] - accumulate_list[start-1])