結果

問題 No.3050 Prefix Removal
コンテスト
ユーザー suisen
提出日時 2023-01-05 22:30:22
言語 PyPy3
(7.3.17)
コンパイル:
pypy3 -mpy_compile _filename_
実行:
pypy3 _filename_
結果
AC  
実行時間 309 ms / 2,000 ms
コード長 406 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 237 ms
コンパイル使用メモリ 95,724 KB
実行使用メモリ 213,892 KB
最終ジャッジ日時 2026-07-06 22:25:26
合計ジャッジ時間 14,416 ms
ジャッジサーバーID
（参考情報） 		judge2_1 / judge1_1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 55
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

import heapq

N, K = map(int, input().split())
A = list(map(int, input().split()))
B = [0] * (N + 1)
for i, v in enumerate(A):
    B[i + 1] = B[i] + v
ans = -(1 << 62)

pq = []
pq_sum = 0
for X in range(1, N + 1):
    if X >= K:
        ans = max(ans, K * B[X] - pq_sum)
    heapq.heappush(pq, -B[X]) # reversed order
    pq_sum += B[X]
    if len(pq) >= K:
        pq_sum -= -heapq.heappop(pq)
print(ans)
0