코딩테스트 대비/BOJ

[Baekjoon/Python] 11659번: 구간 합 구하기 4 - 효과는 굉장했다!

bluetag_boy 2021. 11. 14. 03:55
반응형
 

11659번: 구간 합 구하기 4

첫째 줄에 수의 개수 N과 합을 구해야 하는 횟수 M이 주어진다. 둘째 줄에는 N개의 수가 주어진다. 수는 1,000보다 작거나 같은 자연수이다. 셋째 줄부터 M개의 줄에는 합을 구해야 하는 구간 i와 j

www.acmicpc.net

알고리즘 분류

  • 누적 합

 

SOLUTION

 

import sys

N, M = map(int, sys.stdin.readline().split())
nums = list(map(int, sys.stdin.readline().split()))
sum_list = [0]

for i in range(N):
    sum_list.append(sum_list[-1] + nums[i]) # 미리 누적합을 구해놓음
    
for _ in range(M):
    i, j = map(int, sys.stdin.readline().split())
    if i == 1:
        print(sum_list[j])

    else: 
        print(sum_list[j]-sum_list[i-1])