結果

問題 No.1331 Moving Penguin
コンテスト
ユーザー chineristAC
提出日時 2020-10-30 17:59:49
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 904 ms / 1,500 ms
コード長 660 bytes
コンパイル時間 476 ms
コンパイル使用メモリ 82,560 KB
実行使用メモリ 92,920 KB
最終ジャッジ日時 2024-11-07 14:33:21
合計ジャッジ時間 36,065 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
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(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