코테 노트/백준

백준 11659 구간 합 구하기 4 Python3

화요밍 2021. 8. 18. 18:56
728x90
반응형

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

 

11659번: 구간 합 구하기 4

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

www.acmicpc.net

 

최종 코드

 

GitHub - hwayeon351/BEAKJOON-Algorithms: 백준 알고리즘 소스 코드 모음

백준 알고리즘 소스 코드 모음. Contribute to hwayeon351/BEAKJOON-Algorithms development by creating an account on GitHub.

github.com

import sys
N, M = map(int, sys.stdin.readline().split())
nums = list(map(int, sys.stdin.readline().split()))
nums = [0] + nums
dp = [0]*(N+1)
for i in range(1, N+1):
    dp[i] = dp[i-1] + nums[i]

for _ in range(M):
    i, j = map(int, sys.stdin.readline().split())
    print(dp[j]-dp[i-1])

 

GitHub - hwayeon351/BEAKJOON-Algorithms: 백준 알고리즘 소스 코드 모음

백준 알고리즘 소스 코드 모음. Contribute to hwayeon351/BEAKJOON-Algorithms development by creating an account on GitHub.

github.com

import sys
sys.setrecursionlimit(200000)
N, M = map(int, sys.stdin.readline().split())
nums = list(map(int, sys.stdin.readline().split()))
nums = [0] + nums
def cal_dp(i):
    if i == 0: return 0
    if dp[i] != 0:
        return dp[i]
    dp[i] = cal_dp(i-1) + nums[i]
    return dp[i]

dp = [0]*(N+1)

for _ in range(M):
    i, j = map(int, sys.stdin.readline().split())
    print(cal_dp(j)-cal_dp(i-1))
728x90
반응형