結果

問題 No.2028 Even Choice
ユーザー FromBooska
提出日時 2023-08-09 18:58:35
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 246 ms / 2,000 ms
コード長 834 bytes
コンパイル時間 356 ms
コンパイル使用メモリ 82,464 KB
実行使用メモリ 112,228 KB
最終ジャッジ日時 2024-11-15 15:53:32
合計ジャッジ時間 5,400 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 28
権限があれば一括ダウンロードができます

ソースコード

diff # 

# 一番手前の数字は偶数番目である必要ある
# それ以後で元々偶数番なら最初にとればいいので取れる
# それ以後で奇数番なら最初の偶数番を取ってから取ればいいので取れる
# つまり一番手前が偶数番ならば自由にとれる
# 後ろからheapで高いものを管理
# 常に最大値K-1個をheapに維持する
# 偶数番をとったときにans更新

N, K = map(int, input().split())
A = list(map(int, input().split()))

from heapq import *
H = []
heapify(H)
s = 0
ans = 0
for i in range(N-1, -1, -1):
    a = A[i]
    heappush(H, a)
    s += a
    
    if len(H) == K:
        if i%2 == 1: 
            ans = max(ans, s)        
        smallest = heappop(H)
        s -= smallest
        # 常に最大値K-1個をheapに維持する
    
print(ans)
0