結果

問題 No.1331 Moving Penguin
ユーザー shotoyooshotoyoo
提出日時 2021-01-08 22:03:45
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 742 ms / 1,500 ms
コード長 791 bytes
コンパイル時間 556 ms
コンパイル使用メモリ 82,560 KB
実行使用メモリ 93,440 KB
最終ジャッジ日時 2024-11-07 14:42:43
合計ジャッジ時間 19,687 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 49
権限があれば一括ダウンロードができます

ソースコード

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(300, max(a)+1)
dp = [0]*(n)
dp[0] = 1
M = 10**9+7
vals = [[0]*B for _ in range(B)]
def sub(i):
    res = 0
    for v in range(1,B):
        res += vals[v][i%v]
        res %= M
    return res
#     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