結果

問題 No.1331 Moving Penguin
ユーザー chineristAC
提出日時 2020-10-30 16:25:06
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,078 ms / 1,500 ms
コード長 763 bytes
コンパイル時間 510 ms
コンパイル使用メモリ 82,432 KB
実行使用メモリ 340,004 KB
最終ジャッジ日時 2024-11-07 14:32:12
合計ジャッジ時間 38,762 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 49
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

dp = [0 for i in range(N+1)]
data = [[0 for i in range(320)] for j in range(N)]
dp[-1] = 1
for j in range(1,min(N,320)):
    data[N-j][j] = 1

for i in range(N-1,0,-1):
    if A[i]**2 > N:
        for j in range(1,N+1):
            tmp = i + j * A[i]
            if tmp<N+1:
                dp[i] += dp[tmp]
                dp[i] %= mod
            else:
                break
        if A[i]!=1:
            dp[i] += dp[i+1]
            dp[i] %= mod
    else:
        dp[i] = data[i][A[i]]
        if A[i]!=1:
            dp[i] += dp[i+1]
            dp[i] %= mod

    for j in range(1,min(320,i)):
        data[i - j][j] = data[i][j] + dp[i]
        data[i - j][j] %= mod

print(dp[1])
0