반응형
[문제 출처]
https://www.acmicpc.net/problem/11659
11659번: 구간 합 구하기 4
첫째 줄에 수의 개수 N과 합을 구해야 하는 횟수 M이 주어진다. 둘째 줄에는 N개의 수가 주어진다. 수는 1,000보다 작거나 같은 자연수이다. 셋째 줄부터 M개의 줄에는 합을 구해야 하는 구간 i와 j
www.acmicpc.net
[소스 코드]
누적 합으로 푸는 문제이다.
arr[i]에는 i까지의 원소의 합이 저장되도록 하고,
만일 i부터 j까지의 합을 구해야 한다면, arr[j]에서 arr[i-1]을 빼서 계산한다.
ex)
5 4 3 2 1
i = 2, j = 4
5 5+4 5+4+3 5+4+3+2 5+4+3+2+1
2번째부터 4번째까지의 합은
arr[4] - arr[1]
(5+4+3+2) - (5) = 4+3+2 = 9
#11659
import sys
input = sys.stdin.readline
n,m = map(int, input().split())
arr = list(map(int, input().split()))
for i in range(1,n):
arr[i] += arr[i-1]
arr = [0] + arr
for i in range(m):
a,b = map(int, input().split())
print(arr[b]-arr[a-1])
반응형
'알고리즘 > python' 카테고리의 다른 글
[python/파이썬] 백준 25304 영수증 (0) | 2022.12.29 |
---|---|
[python/파이썬] 백준 2588 곱셈 (0) | 2022.12.29 |
[python/파이썬] 백준 3273 sumX (0) | 2022.12.27 |
[python/파이썬] 백준 11022 A+B - 8 (0) | 2022.12.26 |
[python/파이썬] 백준 1157 단어 공부 (0) | 2022.12.23 |