結果

問題 No.258 回転寿司(2)
ユーザー tktk_snsn
提出日時 2021-01-27 01:50:17
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 405 bytes
コンパイル時間 222 ms
コンパイル使用メモリ 82,404 KB
実行使用メモリ 55,652 KB
最終ジャッジ日時 2024-06-23 23:33:12
合計ジャッジ時間 7,163 ms
ジャッジサーバーID
(参考情報)
judge2 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample WA * 4
other WA * 67
権限があれば一括ダウンロードができます

ソースコード

diff #

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

take = [0] * (N + 1)
stay = [0] * (N + 1)
for i, a in enumerate(A, 1):
    # とるとき
    take[i] = stay[i - 1] + a
    # とらない
    stay[i] = max(stay[i-1], take[i-1])


ans = max(take[N], stay[N])

pos = []
i = N
while i > 0:
    if ans == take[i]:
        pos.append(i)
        ans -= A[i - 1]
        i -= 1
    i -= 1

print(*pos[::-1])
0