結果

問題 No.1331 Moving Penguin
ユーザー persimmon-persimmon
提出日時 2021-02-14 09:02:02
言語 PyPy3
(7.3.15)
結果
TLE  
(最新)
AC  
(最初)
実行時間 -
コード長 614 bytes
コンパイル時間 309 ms
コンパイル使用メモリ 82,176 KB
実行使用メモリ 499,836 KB
最終ジャッジ日時 2024-07-21 14:36:24
合計ジャッジ時間 20,923 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 44 TLE * 5
権限があれば一括ダウンロードができます

ソースコード

diff # 

n=int(input())
a=list(map(int,input().split()))
dp=[0]*n
mod=10**9+7
dp[0]=1
ary=[{} for i in range(n)]
base=0
for i in range(n):
  # マスiに到達するケースを加算
  if ary[i]:
    for k,v in ary[i].items():
      dp[i]=(dp[i]+v)%mod
      if i+k<n:
        if k in ary[i+k]:
          ary[i+k][k]+=v
        else:
          ary[i+k][k]=v
  dp[i]+=base
  dp[i]%=mod
  if i==n-1:break
  k=a[i]
  if k==1:
    base+=dp[i]
    base%=mod
  else:
    dp[i+1]+=dp[i]
    if i+k<n:
      if k in ary[i+k]:
        ary[i+k][k]+=dp[i]
        ary[i+k][k]%=mod
      else:
        ary[i+k][k]=dp[i]
print(dp[-1])
0