백준 알고리즘

[백준/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)