結果

問題 No.1331 Moving Penguin
ユーザー Navier_Boltzmann
提出日時 2023-09-13 11:28:12
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 910 ms / 1,500 ms
コード長 757 bytes
コンパイル時間 279 ms
コンパイル使用メモリ 82,356 KB
実行使用メモリ 92,672 KB
最終ジャッジ日時 2024-06-30 20:47:59
合計ジャッジ時間 36,805 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 49
権限があれば一括ダウンロードができます

ソースコード

diff # 

from collections import *
from itertools import *
from functools import *
from heapq import *
import sys,math
input = sys.stdin.readline

N = int(input())
A = [0]+list(map(int,input().split()))

mod = 10**9 + 7

B = math.ceil(math.sqrt(N))
S = [[0]*(B+1) for _ in range(B)]
dp = [0]*(N+1)

dp[1] = 1
a = A[1]
if a>B:
    for i in range(1+a,N+1,a):
        dp[i] = (dp[i]+dp[1])%mod
else:
    S[1%a][a] = (S[1%a][a] + dp[1])%mod
    
for i in range(2,N+1):
    
    if A[i-1]!=1:
        dp[i] = (dp[i]+dp[i-1])%mod
    
    for j in range(1,B+1):
        dp[i] = (dp[i] + S[i%j][j])%mod
    
    a = A[i]
    if a>B:
        for j in range(i+a,N+1,a):
            dp[j] = (dp[j]+dp[i])%mod
    else:
        S[i%a][a] = (S[i%a][a] + dp[i])%mod
print(dp[-1])
0