結果

問題 No.1117 数列分割
コンテスト
ユーザー tktk_snsn
提出日時 2021-03-15 22:21:14
言語 PyPy3
(7.3.17)
結果
MLE  
実行時間 -
コード長 792 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 347 ms
コンパイル使用メモリ 82,560 KB
実行使用メモリ 342,492 KB
最終ジャッジ日時 2024-11-07 10:58:07
合計ジャッジ時間 9,566 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 8 MLE * 2 -- * 16
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

import heapq
from itertools import accumulate

N, K, M = map(int, input().split())
A = list(map(int, input().split()))
B = list(accumulate(A))

X = [[] for _ in range(K+1)]
Y = [[] for _ in range(K+1)]
for i in range(K):
    X[i].append((0, N+1))
    Y[i].append((0, N+1))

dp = [[0] * (K + 1) for _ in range(N + 1)]
for i, b in enumerate(B, 1):
    L = max(0, i - M)
    for j in reversed(range(1, K + 1)):
        while X[j-1] and X[j-1][0][1] < L:
            heapq.heappop(X[j-1])
        while Y[j-1] and Y[j-1][0][1] < L:
            heapq.heappop(Y[j-1])
        val = 0
        val = max(val, b - X[j-1][0][0])
        val = max(val, -Y[j-1][0][0] - b)
        dp[i][j] = val
        heapq.heappush(X[j], (-(val - b), i))
        heapq.heappush(Y[j], (-(val + b), i))

print(dp[N][K])
0