結果

問題 No.1331 Moving Penguin
ユーザー chineristACchineristAC
提出日時 2020-10-30 17:45:26
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 954 ms / 1,500 ms
コード長 669 bytes
コンパイル時間 408 ms
コンパイル使用メモリ 82,452 KB
実行使用メモリ 339,312 KB
最終ジャッジ日時 2024-11-07 14:35:16
合計ジャッジ時間 37,955 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 49
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

assert 1<=N<=10**5
assert len(A)==N
assert all(1<=A[i]<=N for i in range(N))

A = [0] + A
mod = 10**9 + 7

dp = [0 for i in range(N+1)]
data = [[0 for i in range(320)] for j in range(N)]
dp[-1] = 1
for j in range(1,min(N,320)):
    data[N-j][j] = 1

for i in range(N-1,0,-1):
    if A[i]**2 > N:
        for j in range(A[i],N+1-i,A[i]):
            dp[i] += dp[i+j]
            dp[i] %= mod
    else:
        dp[i] = data[i][A[i]]

    if A[i]!=1:
        dp[i] += dp[i+1]
        dp[i] %= mod

    for j in range(1,min(320,i)):
        data[i - j][j] = data[i][j] + dp[i]
        data[i - j][j] %= mod

print(dp[1])
0