結果

問題 No.258 回転寿司(2)
ユーザー chocorusk
提出日時 2020-09-18 22:59:43
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 31 ms / 2,000 ms
コード長 554 bytes
コンパイル時間 371 ms
コンパイル使用メモリ 12,800 KB
実行使用メモリ 11,008 KB
最終ジャッジ日時 2024-06-22 10:16:22
合計ジャッジ時間 6,721 ms
ジャッジサーバーID
(参考情報)
judge3 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 4
other AC * 67
権限があれば一括ダウンロードができます

ソースコード

diff #

import sys
read=sys.stdin.buffer.read
readline=sys.stdin.buffer.readline
readlines=sys.stdin.buffer.readlines

n=int(readline())
v=list(map(int, readline().split()))
dp=v[:]
for i in range(1, n):
    if dp[i]<dp[i-1]:
        dp[i]=dp[i-1]
    if i-2>=0 and dp[i]<dp[i-2]+v[i]:
        dp[i]=dp[i-2]+v[i]
ans=[]
i=n-1
while i>=0:
    if dp[i]==v[i]:
        ans.append(i+1)
        break
    if i-2>=0 and dp[i]==dp[i-2]+v[i]:
        ans.append(i+1)
        i-=2
    else:
        i-=1
ans=ans[::-1]
print(dp[n-1])
print(' '.join([str(i) for i in ans]))
0