結果

問題 No.2028 Even Choice
ユーザー tassei903
提出日時 2022-08-08 07:21:35
言語 PyPy3
(7.3.15)
結果
RE  
実行時間 -
コード長 898 bytes
コンパイル時間 133 ms
コンパイル使用メモリ 82,724 KB
実行使用メモリ 119,052 KB
最終ジャッジ日時 2024-09-18 18:38:56
合計ジャッジ時間 5,845 ms
ジャッジサーバーID
(参考情報)
judge3 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 24 RE * 4
権限があれば一括ダウンロードができます

ソースコード

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[i])

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