코딩 테스트/python

파이썬 코딩테스트 구간합 구하는 방법

remarkable-book 2023. 8. 28. 08:00
반응형

파이썬 코딩테스트 구간합 구하는 방법

 

prefix_sum = [0]
temp = 0
for i in range(1, 11):
    temp += i
    prefix_sum.append(temp)
    
prefix_sum
# [0, 1, 3, 6, 10, 15, 21, 28, 36, 45, 55]

'''
+1을 하는 이유는 파이썬이 0부터 시작하기 때문. -1을 하는 이유는 마지막 인덱스까지 계산하기 위해서
'''
n = 3
prefix_sum[n + 1] - prefix_sum[1-1]
# 10

prefix_sum[5-1] 
# 5 - 1 = 4 번째 인덱스. 즉, 4번째 인덱스에 있는 값 10 반환.

 

반응형