본문 바로가기
🔅코딩테스트 공부🔅/❗백준

[백준] 11659 구간 합 구하기 4(python)(누적합)

by 윤무무 2023. 3. 1.

https://www.acmicpc.net/problem/11659

 

11659번: 구간 합 구하기 4

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

www.acmicpc.net

 

1. 내가 작성한 코드
import sys

input = sys.stdin.readline

n,m = map(int, input().rstrip().split())

a = list(map(int, input().rstrip().split())) + [0]

for i in range(n):
  a[i] += a[i-1]

for i in range(m):
  left,right = map(int,input().rstrip().split())
  print(a[right-1] - a[left-2])

 

누적합 개념을 이용했다.

 

처음에는 누적합을 저장하기 위한 새로운 list를 만들었으나,

 

이 과정을 피하기 위해 입력받은 수열 끝에 [0]을 하나 더 붙임으로써 첫 번째 항의 인덱스 범위 초과 에러를 해결했다.

댓글