11659 - 구간 합 구하기 4
📌 문제
수 N개가 주어졌을 때, i번째 수부터 j번째 수까지 합을 구하는 프로그램을 작성하시오.
📋 코드
import sys
input = sys.stdin.readline
N, M = map(int, input().split())
arr = [0] + list(map(int, input().split()))
prefix = [list(map(int, input().split())) for _ in range(M)]
for i in range(1, len(arr)):
arr[i] += arr[i-1]
for i, j in prefix:
print(arr[j] - arr[i-1])
💡 한마디
반복문을 통해 arr 리스트의 누적 합을 구한다. 그러면 i번째부터 j번째까지의 구간 합을 arr[j] - arr[i-1] 으로 나타낼 수 있다.