2024 상반기 취준 준비 코테

구간합 리스트 빨리 구하기

void_melody 2024. 4. 2. 11:23
n, m = map(int, input().split())
numbers = list(map(int, input().split()))

s = [0 for _ in range(n+1)]
for i in range(1, n+1):
    for j in range(0, i):
        s[i] += numbers[j]

 

위의 것을 아래로 변경한다.

n, m = map(int, input().split())
numbers = list(map(int, input().split()))

for i in range(n-1):
    numbers[i+1] += numbers[i]
s = [0] + numbers