結果
問題 |
No.2028 Even Choice
|
ユーザー |
|
提出日時 | 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 |
ソースコード
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)