개발 연습장/백준 문제풀이

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

LooanCheong 2023. 2. 17. 10:17
반응형

문제

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

 

11659번: 구간 합 구하기 4

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

www.acmicpc.net

코드

import sys
input = sys.stdin.readline

n, m = map(int,input().split())
num = list(map(int, input().split()))

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

for _ in range(m):
    i, j = map(int,input().split())
    print(num[j] - num[i-1])

설명

누적 합을 처음으로 접해봤던 문제였다.

범위에 맞게 i+1을 i까지의 합으로 만들어준다.
이후 계산이 헷갈리지 않게 앞에 [0]을 추가해 주어 시작 인덱스를 1로 맞춰준다.

구간의 합을 구할 숫자를 받아주고 이를 num 리스트에 있는 값을 통해 출력해 준다.

반응형