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
'Develop > 백준 (python)' 카테고리의 다른 글
[백준] 2309번: 일곱 난쟁이 (python) (0) | 2022.02.09 |
---|---|
[백준] 1541번: 잃어버린 괄호 (python) (0) | 2022.02.09 |
[백준] 11399번: ATM (python) (0) | 2022.02.08 |
[백준] 11047번: 동전 0 (python) (0) | 2022.02.07 |
[백준] 1003번: 피보나치 함수 (python) (0) | 2022.02.06 |