結果

問題 No.2028 Even Choice
ユーザー tassei903
提出日時 2022-08-08 07:23:18
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 265 ms / 2,000 ms
コード長 899 bytes
コンパイル時間 276 ms
コンパイル使用メモリ 81,588 KB
実行使用メモリ 118,396 KB
最終ジャッジ日時 2024-09-18 18:39:53
合計ジャッジ時間 5,510 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 28
権限があれば一括ダウンロードができます

ソースコード

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")
#######################################################################
from heapq import *
ans = 0
n,k = na()
k-=1
a = na()
b = [-float("inf")]
for i in range(1,n,2):
    b.append(max(b[-1],a[i]))
s = 0
hq = []
if n % 2:
    hq.append(a[-1])
    s += a[-1]
    while len(hq) > k:
        x = heappop(hq)
        s-=x
    if len(hq) == k:
        ans = max(ans, s + b[-1])

for i in range(n//2-1,-1,-1):
    heappush(hq, a[i*2])
    s += a[i*2]
    heappush(hq, a[i*2+1])
    s += a[i*2 + 1]
    while len(hq) > k:
        x = heappop(hq)
        s-=x
    if len(hq) == k:
        ans = max(ans, s + b[i])

print(ans)
0