누적합·연습 문제

합이 K인 부분 배열 개수

문제 설명PROBLEM

합이 K인 부분 배열 개수

왜 이 문제를 푸나요?

‘합이 K인 부분 배열 개수’은 누적합의 핵심 연산을 다른 입력 조건에서도 스스로 선택할 수 있게 만드는 반복 문제입니다. 결과뿐 아니라 경계값과 목표 복잡도까지 확인하세요.

익혀야 할 개념

  • 전처리
  • 구간 합
  • 인덱스 경계
  • 쿼리 최적화

과제

합이 target인 연속 부분 배열의 개수를 누적합 빈도로 구하세요.

함수

count_subarrays_sum(numbers, target)

목표 복잡도

평균 시간 O(n), 공간 O(n)

테스트 케이스

실행하면 아래 순서대로 채점합니다. print() 출력도 이 순서대로 쌓입니다.

#이름numberstarget기대값
1두 구간[1, 1, 1]22
2음수 포함[1, -1, 0]03
3없음[1, 2]90
4겹치는 0 구간[0, 0, 0]06
5여러 음수 경로[3, -1, -2, 5, -2]22

Python 표준 라이브러리는 사용할 수 있습니다. 함수 이름과 매개변수는 제시된 형태를 유지하세요.

개념 노트와 실수 포인트

누적합

  • 앞에서부터 합을 한 번 저장하면 구간 합을 O(1)에 답할 수 있습니다.
  • prefix[0] = 0으로 두면 [left, right] 합은 prefix[right+1] - prefix[left]입니다.