結果

問題 No.2028 Even Choice
ユーザー ntuda
提出日時 2022-10-08 12:41:15
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 222 ms / 2,000 ms
コード長 691 bytes
コンパイル時間 288 ms
コンパイル使用メモリ 82,112 KB
実行使用メモリ 104,376 KB
最終ジャッジ日時 2024-06-22 14:21:31
合計ジャッジ時間 5,397 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 28
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

if K == 1:
    print(max(A[1:N:2]))
    exit()

# ある偶数より後の枚数(N-i(odd))でK-1より大きい数
if N & 1 :
    K2 = (K - 1) // 2 * 2 + 1
else:
    K2 = K // 2 * 2

X= A[N - K2:N]
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