結果

問題 No.3050 Prefix Removal
ユーザー tassei903
提出日時 2025-03-07 22:12:44
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 478 ms / 2,000 ms
コード長 727 bytes
コンパイル時間 807 ms
コンパイル使用メモリ 81,984 KB
実行使用メモリ 205,176 KB
最終ジャッジ日時 2025-03-07 22:13:03
合計ジャッジ時間 18,606 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 55
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
input = lambda :sys.stdin.readline()[:-1]
ni = lambda :int(input())
na = lambda :list(map(int,input().split()))
yes = lambda :print("yes");Yes = lambda :print("Yes");YES = lambda : print("YES")
no = lambda :print("no");No = lambda :print("No");NO = lambda : print("NO")
#######################################################################
inf = 10 ** 20

n, k = na()
a = na()

b = [0] + a
for i in range(n):
    b[i+1] += b[i]

ans = -inf

from heapq import *
hq = []
z = b[0]
for i in range(1, k):
    heappush(hq, -b[i])
    z += b[i]
# print(b)
for i in range(k, n + 1):
    # print(b[i], z, hq)
    ans = max(ans, b[i] * k - z)
    heappush(hq, -b[i])
    z += b[i]
    x = -heappop(hq)
    z -= x

print(ans)
0