結果

問題 No.1331 Moving Penguin
ユーザー shotoyoo
提出日時 2021-01-08 21:46:51
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 693 bytes
コンパイル時間 320 ms
コンパイル使用メモリ 82,688 KB
実行使用メモリ 189,440 KB
最終ジャッジ日時 2024-11-16 11:03:08
合計ジャッジ時間 41,367 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 40 TLE * 9
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
input = lambda : sys.stdin.readline().rstrip()

sys.setrecursionlimit(2*10**5+10)
write = lambda x: sys.stdout.write(x+"\n")
_print = lambda *x: print(*x, file=sys.stderr)

n = int(input())
a = list(map(int, input().split()))
B = min(1000, max(a)+1)
dp = [0]*(n)
dp[0] = 1
M = 10**9+7
vals = [[0]*B for _ in range(B)]
def sub(i):
    return sum(vals[v][i%v] for v in range(1,B))%M
for i,v in enumerate(a):
    dp[i] += sub(i)
    if i==n-1:
        break
    if v>=2:
        dp[i+1] += dp[i]
    if v>=B:
        for j in range(i+v,n,v):
            dp[j] += dp[i]
            dp[j] %= M
    else:
        vals[v][i%v] += dp[i]
        vals[v][i%v] %= M
ans = dp[-1]%M
print(ans%M)
0