백준 알고리즘
[백준/Python]11659_구간 합 구하기 4
헝그리개발자
2021. 6. 4. 14:19
문제
https://www.acmicpc.net/problem/11659
11659번: 구간 합 구하기 4
첫째 줄에 수의 개수 N과 합을 구해야 하는 횟수 M이 주어진다. 둘째 줄에는 N개의 수가 주어진다. 수는 1,000보다 작거나 같은 자연수이다. 셋째 줄부터 M개의 줄에는 합을 구해야 하는 구간 i와 j
www.acmicpc.net
풀이
처음에 m 만큼 리스트를 슬라이싱해서 풀었을 때 시간초과가 났습니다.
i번째 항까지 누적 합을 나타낸 리스트를 미리 만들고, 합하는 연산을 줄여서 통과했습니다.
코드
import sys
n, m = map(int, sys.stdin.readline().rstrip().split())
numbers = list(map(int, sys.stdin.readline().rstrip().split()))
memoization = [0] * n # 누적 합의 값을 미리 구해놓는다
for i in range(n):
if i == 0:
memoization[i] = numbers[i]
continue
if i != 0:
memoization[i] = memoization[i - 1] + numbers[i]
for _ in range(m):
start, end = map(int, sys.stdin.readline().rstrip().split())
result = memoization[end - 1] - memoization[start - 1] + numbers[start - 1]
print(result)