結果

問題 No.1331 Moving Penguin
コンテスト
ユーザー chineristAC
提出日時 2020-10-30 17:59:49
言語 PyPy3
(7.3.17)
コンパイル:
pypy3 -mpy_compile _filename_
実行:
pypy3 _filename_
結果
AC  
実行時間 431 ms / 1,500 ms
コード長 660 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 292 ms
コンパイル使用メモリ 85,120 KB
実行使用メモリ 97,152 KB
最終ジャッジ日時 2026-05-07 22:40:04
合計ジャッジ時間 16,310 ms
ジャッジサーバーID
（参考情報） 		judge1_0 / judge2_0
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 49
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

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(320)]
dp[-1] = 1
for j in range(1,min(N,320)):
    data[j][N%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[A[i]][i%A[i]]

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

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

print(dp[1])
0