結果

問題 No.2028 Even Choice
ユーザー ntuda
提出日時 2022-10-08 12:12:29
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 590 bytes
コンパイル時間 281 ms
コンパイル使用メモリ 82,440 KB
実行使用メモリ 104,224 KB
最終ジャッジ日時 2024-06-22 13:32:27
合計ジャッジ時間 5,993 ms
ジャッジサーバーID
(参考情報)
judge3 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 23 WA * 5
権限があれば一括ダウンロードができます

ソースコード

diff #

import heapq
N, K = map(int, input().split())
A = list(map(int, input().split()))
K2 = (K - 1) // 2 * 2 + 2 - (K & 1)
X = A[N - K2:N]
if K == 1:
    print(max(A[1:N:2]))
    exit()
heapq.heapify(X)
while len(X) > K - 1:
    heapq.heappop(X)
tmp = sum(X)
ans = A[1] + tmp
for i in reversed(range(1, N - K2, 2)):
    ans = max(ans, A[i] + tmp)
    if i == 1:
        break
    for j in range(2):
        tmp2 = heapq.heappop(X)
        a = A[i - j]
        if tmp2 > a:
            heapq.heappush(X, tmp2)
        else:
            tmp += a - tmp2
            heapq.heappush(X, a)
print(ans)
0