알고리즘/백준

[파이썬] BOJ_11659(구간 합 구하기 4)

딱따구르리 2023. 6. 26. 19:09
728x90
반응형

 

 

 

문제

 

https://www.acmicpc.net/problem/11659

 

11659번: 구간 합 구하기 4

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

www.acmicpc.net


 

해설

 

반응형

 

누적합을 저장할 리스트를 만들고 0 ~ i까지의 누적합을 계산해 저장한다.

a~b까지의 구간 합은 b까지의 누적합과 a까지의 누적합을 뺀 것이기 때문에 

누적합만 계산해서 넣어두면 쉽게 풀 수 있다.

리스트에 0을 미리 넣어두면 인덱스를 헷갈리지 않고 풀 수 있다.


코드

 

-파이썬

#백준 11659(구간 합 구하기 4)

import sys
input = sys.stdin.readline
n, m = map(int, input().split())
num = [*map(int, input().split())]
prefix_sum = [0]
temp = 0
for i in num:
    temp += i
    prefix_sum.append(temp)

for j in range(m):
    a, b = map(int, input().split())
    print(prefix_sum[b] - prefix_sum[a-1])

 

 

 

 

728x90
반응형