結果

問題 No.1331 Moving Penguin
コンテスト
ユーザー 👑 SPD_9X2
提出日時 2021-01-08 22:05:41
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 761 ms / 1,500 ms
コード長 706 bytes
コンパイル時間 207 ms
コンパイル使用メモリ 82,432 KB
実行使用メモリ 90,416 KB
最終ジャッジ日時 2024-11-07 14:43:47
合計ジャッジ時間 30,476 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 49
権限があれば一括ダウンロードができます

ソースコード

diff # 

"""

無理やりやれば解けそうではある

"""

from sys import stdin
mod = 10**9+7
LIM = 317

N = int(stdin.readline())
A = list(map(int,stdin.readline().split()))

lis = [[0] * i for i in range(LIM)]

ans = [0] * N
ans[0] = 1

for i in range(N):
    
    #受け取るパート
    for j in range(1,LIM):
        ans[i] += lis[j][i % j]
    ans[i] %= mod

    if i == N-1:
        print (ans[i])
        break
    
    #あげるパート
    if A[i] != 1:
        ans[i+1] += ans[i]

    if A[i] < LIM:
        lis[A[i]][i % A[i]] += ans[i]
        lis[A[i]][i % A[i]] %= mod
    else:
        for j in range(i+A[i],N,A[i]):
            ans[j] += ans[i]
            ans[j] %= mod

#print (ans)
0