[백준 파이썬] 11659 구간 합 구하기 4

📌문제 출처

솔브닷 3+ 클래스 단계
https://www.acmicpc.net/problem/11659

11659번: 구간 합 구하기 4

❓ 문제

❗ 풀이

• 누적합 활용

📗 풀이 코드

import sys
input = sys.stdin.readline
n,m = map(int,input().split())
nums = list(map(int,input().split()))
# 누적합 구하기
prefix_sum = [0]
tmp = 0
for num in nums:
    tmp += num
    prefix_sum.append(tmp)
for _ in range(m):
    a,b = map(int,input().split())
    print(prefix_sum[b]-prefix_sum[a-1])

📗 코드 해설

• 해당 인덱스까지의 누적합을 담는 prefix_sum을 만들어두고 0번째까지의 누적합인 0을 넣어둔다.
• nums의 숫자를 하나씩 돌며 0에다가 더해가며 더해진 수를 prefix_sum에 담아준다.
• 완성된 prefix_sum에서 b까지의 누적합에 a-1까지의 누적합을 빼주면 a부터 b까지 수의 누적합이 추출된다.