본문 바로가기
Develop/백준 (python)

[백준] 11659번: 구간 합 구하기 4 (python)

by Tarra 2022. 2. 8.

11659번: 구간 합 구하기 4


문제 )
수 N개가 주어졌을 때, i번째 수부터 j번째 수까지 합을 구하는 프로그램을 작성하시오.

 

입력 :

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

 

출력 :

총 M개의 줄에 입력으로 주어진 i번째 수부터 j번째 수까지 합을 출력한다.

 

제한:

-  1 ≤ N ≤ 100,000
-  1 ≤ M ≤ 100,000
-  1 ≤ i ≤ j ≤ N

 

 

풀이)

1
2
3
4
5
6
7
8
9
10
11
12
13
14
import sys
 
n, m = map(int, sys.stdin.readline().strip().split())
li = list(map(int, sys.stdin.readline().strip().split()))
 
li_2 = [0#누적합 리스트
total = 0
for i in li:
    total += i
    li_2.append(total)
 
for i in range(m): # li_2[p]-li_2[o-1]를 하면 계산량은 줄이고 결과는 똑같이 낼 수 있다.
    o, p = map(int, sys.stdin.readline().strip().split())
    print(li_2[p]-li_2[o-1])
cs

 

 


+ 22-03-23 2차 풀이

 

1
2
3
4
5
6
7
8
9
10
11
import sys
 
n, m = map(int, sys.stdin.readline().strip().split())
li = [0+ list(map(int, sys.stdin.readline().strip().split()))
prefix = [0* (n + 1)
for i in range(n + 1):
    prefix[i] = prefix[i - 1+ li[i]
 
for i in range(m):
    a, b = map(int, sys.stdin.readline().strip().split())
    print(prefix[b] - prefix[a - 1])
cs

 


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

 

11659번: 구간 합 구하기 4

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

www.acmicpc.net