結果

問題 No.2028 Even Choice
ユーザー ntudantuda
提出日時 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
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 222 ms
101,844 KB
testcase_01 AC 185 ms
101,960 KB
testcase_02 AC 214 ms
102,204 KB
testcase_03 AC 115 ms
100,356 KB
testcase_04 AC 125 ms
100,740 KB
testcase_05 AC 116 ms
80,904 KB
testcase_06 AC 196 ms
104,376 KB
testcase_07 AC 117 ms
83,500 KB
testcase_08 AC 166 ms
95,044 KB
testcase_09 AC 167 ms
98,504 KB
testcase_10 AC 84 ms
82,800 KB
testcase_11 AC 120 ms
80,204 KB
testcase_12 AC 98 ms
76,684 KB
testcase_13 AC 136 ms
101,836 KB
testcase_14 AC 131 ms
83,116 KB
testcase_15 AC 33 ms
53,628 KB
testcase_16 AC 35 ms
53,540 KB
testcase_17 AC 35 ms
52,884 KB
testcase_18 AC 37 ms
53,744 KB
testcase_19 AC 36 ms
53,872 KB
testcase_20 AC 48 ms
62,780 KB
testcase_21 AC 42 ms
61,292 KB
testcase_22 AC 40 ms
59,036 KB
testcase_23 AC 33 ms
53,912 KB
testcase_24 AC 34 ms
53,248 KB
testcase_25 AC 35 ms
53,252 KB
testcase_26 AC 37 ms
52,828 KB
testcase_27 AC 37 ms
53,928 KB
testcase_28 AC 66 ms
91,296 KB
testcase_29 AC 58 ms
81,192 KB
testcase_30 AC 52 ms
72,488 KB
権限があれば一括ダウンロードができます

ソースコード

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