結果

問題 No.1331 Moving Penguin
ユーザー titia
提出日時 2021-01-08 22:23:55
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
TLE  
実行時間 -
コード長 578 bytes
コンパイル時間 399 ms
コンパイル使用メモリ 12,800 KB
実行使用メモリ 309,408 KB
最終ジャッジ日時 2024-11-16 13:07:02
合計ジャッジ時間 119,832 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 3 TLE * 46
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
input = sys.stdin.readline

N=int(input())
A=list(map(int,input().split()))
mod=10**9+7


DP=[[0]*(N+1) for i in range(351)]
DP[0][0]=1

for i in range(N):
    a=A[i]
    for j in range(1,351):
        if i-j>=0:
            DP[j][i]+=DP[j][i-j]
            DP[j][i]%=mod

            DP[0][i]+=DP[j][i]
            DP[0][i]%=mod
            
    if a>350:
        for j in range(i+a,N,a):
            DP[0][j]+=DP[0][i]

    else:
        DP[a][i]+=DP[0][i]
        DP[a][i]%=mod

    if a!=1:
        DP[0][i+1]+=DP[0][i]
        DP[0][i+1]%=mod

print(DP[0][N-1])
0