結果

問題 No.3050 Prefix Removal
ユーザー Eki1009
提出日時 2025-03-07 22:03:16
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 393 bytes
コンパイル時間 460 ms
コンパイル使用メモリ 82,352 KB
実行使用メモリ 201,040 KB
最終ジャッジ日時 2025-03-07 22:03:37
合計ジャッジ時間 18,621 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 54 WA * 1
権限があれば一括ダウンロードができます

ソースコード

diff # 

from heapq import*
n,k,*A = map(int,open(0).read().split())
Q = []
S = [0]*(n+1)
for i in range(n)[::-1]:
  S[i] = S[i+1]+A[i]
M = S[:]
for i in range(n)[::-1]:
  M[i] = min(M[i],M[i+1])
ans = -1<<60
Q = []
sm = S[0]
if k==1:
  ans = sm-M[1]
else:
  for i in range(1,n):
    heappush(Q,S[i])
    sm += S[i]
    if len(Q)==k-1:
      ans = max(ans,sm-M[i+1]*k)
      sm -= heappop(Q)
print(ans)
0