본문으로 이동
코드패스 by Devin.KR

문제 5653 · 기가 I

밭 구역 나누기

인증 제출 통계 · 5분마다 갱신 · 정답 비율 = 정답 제출 / 전체 제출
시간 제한메모리 제한제출정답맞힌 사람정답 비율
1000ms / 테스트 (언어별 보정은 아래 표)메모리: 측정하지 않음(브라우저 채점)0000.00%

문제

한 줄로 늘어선 N개의 밭이 있다. i번째 밭에는 토양 수분 편차 지수 a_i 가 측정되어 있으며, 기준보다 건조하면 양수, 습하면 음수인 정수이다.\n\n농부는 밭을 앞에서부터 차례로 끊어 연속한 구역 여러 개로 나누어 관리하려 한다. 모든 밭은 정확히 하나의 구역에 속해야 하고, 구역은 적어도 밭 하나를 포함한다. 한 구역을 관리하는 비용은 C와, 그 구역에 속한 밭들의 a 값을 모두 더한 합의 제곱을 더한 값이다. 즉 구역이 l번부터 r번 밭까지라면 비용은 C + (a_l + a_{l+1} + ... + a_r)^2 이다.\n\n밭을 구역으로 나누는 방법은 자유롭다. 모든 구역의 관리 비용을 합한 값이 최소가 되도록 나눌 때, 그 최솟값을 구하여라.

입력

첫째 줄에 N과 C가 공백으로 구분되어 주어진다. (1 ≤ N ≤ 200,000, 1 ≤ C ≤ 10^9)\n둘째 줄에 a_1, a_2, ..., a_N 이 공백으로 구분되어 주어진다. (-1000 ≤ a_i ≤ 1000, 정수)

출력

비용 합의 최솟값을 한 줄에 출력한다.

예제 입력 1

3 1
1 -1 1

예제 출력 1

2

예제 입력 2

3 3
2 2 2

예제 출력 2

21

예제 입력 3

4 1
3 -1 -1 3

예제 출력 3

10
힌트 펼치기
  • S_i 를 앞에서부터의 누적합이라 하면 dp[i] = min over j<i of dp[j] + (S_i - S_j)^2 + C 이다.
  • 식을 전개하면 dp[j] + S_j^2 - 2 S_j S_i 는 S_i 에 대한 일차식이므로 직선들의 최솟값 질의가 된다.
  • a_i 가 음수일 수 있어 기울기와 질의 좌표가 단조롭지 않으므로, 질의 좌표를 압축한 리 차오 트리를 쓴다. 답은 64비트를 넘지 않지만 2^53 은 넘을 수 있다.

관련 서재 장

관련 장

출처: Devin.KR 자체 제작

알고리즘 분류동적 계획법리 차오 트리볼록 껍질 트릭

풀이 작성

연습 채점은 공개 테스트로 실행합니다. 인증 제출은 코드를 봉인한 뒤 새 입력을 실행하고 서버에서 출력을 대조하며 코드와 판정을 저장합니다.

Python 기본 시간 제한 = 문제 시간 제한 × 3 + 2초. 문제별 예외는 아래 표에 반영됩니다. 표준 라이브러리만 지원합니다.

Ctrl+Enter: 연습 채점 · Tab: 들여쓰기 · Esc 후 Tab: 다음 요소로 이동. 언어별 초안은 이 브라우저에 저장됩니다. JavaScript 입력: require('fs').readFileSync(0, 'utf8'), readline 또는 input().

로그인하면 인증 제출을 사용할 수 있습니다.

풀이를 제출하면 여기에서 결과를 확인할 수 있습니다.

    연습 기록은 이 브라우저에 최근 50개를 저장합니다. 인증 기록은 내 제출에서 확인하세요.

    이 브라우저의 연습 기록

      언어별 제한

      테스트 케이스마다 새 전역에서 실행하며 준비 시간은 제외합니다. 시간 초과된 출력은 검사하지 않습니다. 브라우저·기기 성능에 따라 실행 시간이 달라집니다.

      언어엔진(버전)시간 제한메모리
      javascript사용자 브라우저 JavaScript 엔진 (버전은 브라우저에 따라 다름)1000ms / 테스트측정하지 않음(브라우저 채점)
      pythonPyodide 314.0.7 (CPython 3.14 WASM)5000ms / 테스트측정하지 않음(브라우저 채점)