동캄의 코딩도장

백준 11659 구간 합 구하기 본문

코테/BOJ

백준 11659 구간 합 구하기

동 캄 2023. 3. 20. 10:22

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

 

11659번: 구간 합 구하기 4

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

www.acmicpc.net

#백준 11659 구간 합 구하기
import sys
input=sys.stdin.readline
N,M=map(int,input().split())
l=list(map(int,input().split()))
s={0:0}
for i in range(N):
    s[i+1]=(s[i]+l[i])
for _ in range(M):
    start,end=map(int,input().split())
    ans=s[end]-s[start-1]
    print(ans)

딕셔너리 s를생성하고, 첫 항부터 i항까지의 합을 s[i]에 저장한다.

start항~end항 까지의 부분 합은 = s[end]-s[start-1] 이므로 계산해준다.

 

#백준 11659 구간 합 구하기
import sys
input=sys.stdin.readline
N,M=map(int,input().split())
l=list(map(int,input().split()))
s=[0]
for i in range(N):
    s.append(s[i]+l[i])
for _ in range(M):
    start,end=map(int,input().split())
    ans=s[end]-s[start-1]
    print(ans)

s를 리스트로 생성하고, 첫항부터 i번째항 까지 합이라고 하면,

s[i+1] = s[i]+l[i] 가 성립한다. 따라서 start항~end항 까지의 부분 합은 = s[end]-s[start-1]로 계산해준다.