結果

問題 No.1331 Moving Penguin
ユーザー Navier_Boltzmann
提出日時 2023-09-13 11:27:25
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 739 bytes
コンパイル時間 437 ms
コンパイル使用メモリ 82,404 KB
実行使用メモリ 101,616 KB
最終ジャッジ日時 2024-06-30 20:46:42
合計ジャッジ時間 5,580 ms
ジャッジサーバーID
(参考情報)
judge5 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 4 TLE * 1 -- * 44
権限があれば一括ダウンロードができます

ソースコード

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 = 10**3
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