結果

問題 No.3656 Game Scores and Costs
コンテスト
ユーザー chofu777
提出日時 2026-08-30 15:59:00
言語 Python3
(3.14.7 + numpy 2.5.2 + scipy 1.18.0)
コンパイル:
python3 -mpy_compile _filename_
実行:
python3 _filename_
結果
AC  
実行時間 285 ms / 2,000 ms
+ 46µs
コード長 481 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 291 ms
コンパイル使用メモリ 21,408 KB
実行使用メモリ 37,040 KB
最終ジャッジ日時 2026-08-30 15:59:11
合計ジャッジ時間 7,738 ms
ジャッジサーバーID
（参考情報） 		judge2_0 / judge1_0
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 21
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

import heapq

N,K,X = map(int, input().split())
A_li = list(map(int, input().split()))
ans = float("-inf") # 负无穷

max_k = [] # 小根堆，用来存A_li中最大的K个元素
max_k_sum = 0 # max_k中元素的和
for i in range(N):
    if len(max_k) < K:
        heapq.heappush(max_k, A_li[i])
        max_k_sum += A_li[i]
    elif A_li[i] > max_k[0]:
        max_k_sum += A_li[i] - heapq.heapreplace(max_k, A_li[i])

    ans = max(ans, max_k_sum - X * (i+1))
    
print(ans)
0