누적합·연습 문제

좌우 합이 같은 인덱스

문제 설명PROBLEM

좌우 합이 같은 인덱스

왜 이 문제를 푸나요?

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

익혀야 할 개념

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

과제

자기 자신을 제외한 왼쪽 합과 오른쪽 합이 같은 첫 인덱스를 반환하세요. 없으면 -1입니다.

함수

equilibrium_index(numbers)

목표 복잡도

시간 O(n) 이내

테스트 케이스

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

#이름numbers기대값
1가운데[1, 3, 5, 2, 2]2
2첫 위치[0, 0]0
3없음[1, 2]-1
4한 원소[7]0
5마지막 위치[1, -1, 0]2

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

개념 노트와 실수 포인트

누적합

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