結果
問題 |
No.2028 Even Choice
|
ユーザー |
![]() |
提出日時 | 2024-03-18 16:43:04 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 218 ms / 2,000 ms |
コード長 | 743 bytes |
コンパイル時間 | 249 ms |
コンパイル使用メモリ | 82,788 KB |
実行使用メモリ | 114,352 KB |
最終ジャッジ日時 | 2024-09-30 04:59:03 |
合計ジャッジ時間 | 5,351 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 28 |
ソースコード
# 取れるのは偶数番だけ、最初に偶数番を取れば、あとは奇数番が偶数番になるので結局好きなのを取れる # つまり最初だけ偶数番、あとは好きなのを取れる # 逆順heapでK-1個の最高値を事前計算するか from heapq import heapify, heappop, heappush N, K = map(int, input().split()) A = list(map(int, input().split())) largest_sum = [0]*N H = [] heapify(H) s = 0 for i in range(N-1, 0, -1): heappush(H, A[i]) s += A[i] if len(H)>K-1: smallest = heappop(H) s -= smallest largest_sum[i-1] = s #print(A) #print(largest_sum) ans = 0 for i in range(N): if i%2 == 1: calc = A[i]+largest_sum[i] ans = max(ans, calc) print(ans)