누적합·연습 문제

구간의 1 개수

문제 설명PROBLEM

구간의 1 개수

왜 이 문제를 푸나요?

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

익혀야 할 개념

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

과제

0과 1 배열에서 각 포함 구간의 1 개수를 누적합으로 반환하세요.

함수

count_ones_ranges(bits, queries)

목표 복잡도

시간 O(n) 이내

테스트 케이스

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

#이름bitsqueries기대값
1여러 구간[1, 0, 1, 1][[0, 1], [1, 3]][1, 2]
2한 칸[0, 1][[1, 1]][1]
3쿼리 없음[1][][]
4전체가 0[0, 0, 0][[0, 2], [1, 1]][0, 0]
5양 끝 경계[1, 1, 0, 1][[0, 3], [0, 0], [3, 3]][3, 1, 1]

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

개념 노트와 실수 포인트

누적합

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