### 알아두면 유용한 함수 ###
#############
# > 구간 합 계산
# 연속적으로 나열된 N 개의 수가 있을 때, 특정 구간의 모든 수를 합한 값을 구한다.
# 1. N개의 수에 대하여 접두사 합(Prefix Sum)을 계산하여 배열 P에 저장한다.
# 2. 매 M갸의 쿼리 정보[L, R]을 확인할 때 구간 합은 P[R] - P[L - 1]이다.
# 접두사 합을 활용한 구간 합 계산
# 데이터의 개수 N과 전체 데이터 선언
n = 5
data = [10, 20, 30, 40, 50]
# 접두사 합(Prefix Sum) 배열 계산
sum_value = 0
prefix_sum = [0]
for i in data:
sum_value += i
prefix_sum.append(sum_value)
# 구간 합 계산( 예시 : 3번째 부터 네번째 수까지)
left = 3 # 시작 구간
right = 4 # 종료 구간
print(prefix_sum[right] - prefix_sum[left - 1])
'이것이 취업을 위한 코딩테스트다 with 파이썬' 부록 학습내용
'공부 > 파이썬' 카테고리의 다른 글
파이썬 기타 알고리즘 - 순열과 조합 (0) | 2022.12.09 |
---|---|
파이썬 기타 알고리즘 - 투 포인터 (1) | 2022.12.09 |
파이썬 기타 알고리즘 - 소수의 판별 (0) | 2022.12.08 |
파이썬 주요 라이브러리 문법 (0) | 2022.12.07 |
파이썬 입출력 (0) | 2022.12.06 |