結果

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

ソースコード

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 = -float("inf")
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:
      sm -= heappop(Q)
    if len(Q)==k-1:
      ans = max(ans,sm-M[i+1]*k)
print(ans)
0