結果

問題 No.1331 Moving Penguin
コンテスト
ユーザー persimmon-persimmon
提出日時 2021-02-14 09:37:23
言語 PyPy3
(7.3.17)
コンパイル:
pypy3 -mpy_compile _filename_
実行:
pypy3 _filename_
結果
AC  
実行時間 243 ms / 1,500 ms
コード長 741 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 273 ms
コンパイル使用メモリ 85,152 KB
実行使用メモリ 95,488 KB
最終ジャッジ日時 2026-05-07 22:58:22
合計ジャッジ時間 7,819 ms
ジャッジサーバーID
（参考情報） 		judge3_0 / judge2_1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 49
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

def main1(n,a):
  dp=[0]*n
  mod=10**9+7
  dp[0]=1
  base=0
  c=int(n**0.5)+1
  c=51
  ary=[[0]*(c+1) for _ in range(c+1)]
  for i in range(n):
    # マスiに到達するケースを加算
    for j in range(1,min(i,c)+1):
      dp[i]+=ary[j][i%j]
      dp[i]%=mod
    dp[i]+=base
    dp[i]%=mod
    if i==n-1:break
    k=a[i]
    if k==1:
      base+=dp[i]
      base%=mod
    elif k<c:
      dp[i+1]+=dp[i]
      dp[i+1]%=mod
      ary[k][i%k]+=dp[i]
      ary[k][i%k]%=mod
    else:
      dp[i+1]+=dp[i]
      dp[i+1]%=mod
      ii=i+k
      while ii<n:
        dp[ii]+=dp[i]
        dp[ii]%=mod
        ii+=k
  return dp[-1]

if __name__=='__main__':
  n=int(input())
  a=list(map(int,input().split()))
  ret1=main1(n,a)
  print(ret1)
0