結果
問題 | No.1331 Moving Penguin |
ユーザー | aaaaaaaaaa2230 |
提出日時 | 2021-01-08 22:23:02 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 792 ms / 1,500 ms |
コード長 | 595 bytes |
コンパイル時間 | 426 ms |
コンパイル使用メモリ | 81,844 KB |
実行使用メモリ | 91,200 KB |
最終ジャッジ日時 | 2024-11-07 14:50:25 |
合計ジャッジ時間 | 31,728 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 49 |
ソースコード
n = int(input()) A = list(map(int,input().split())) mod = 10**9+7 x = int(n**0.5) count = [[0]*(i+1) for i in range(x)] dp = [0]*(n+1) dp[0] = 1 for i,a in enumerate(A): for j in range(x): dp[i] += count[j][(i+1)%(j+1)] dp[i] %= mod if a == 1: count[0][0] += dp[i] count[0][0] %= mod continue dp[i+1] += dp[i] dp[i+1] %= mod if a <= x: count[a-1][(i+1)%a] += dp[i] count[a-1][(i+1)%a] %= mod else: for j in range(i+a,n,a): dp[j] += dp[i] dp[j] %= mod print(dp[-2])