結果

問題 No.978 Fibonacci Convolution Easy
ユーザー qib
提出日時 2022-11-18 20:21:29
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 96 ms / 2,000 ms
コード長 356 bytes
コンパイル時間 177 ms
コンパイル使用メモリ 81,620 KB
実行使用メモリ 91,136 KB
最終ジャッジ日時 2024-09-20 01:09:38
合計ジャッジ時間 2,218 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 21
権限があれば一括ダウンロードができます

ソースコード

diff # 

MOD = 10 ** 9 + 7

n, p = map(int, input().split())

if n == 1:
  print(0)
  exit()

a = [0 for _ in range(n + 1)]
s = [0 for _ in range(n + 1)]
a[2] = 1
s[2] = 1
for i in range(3, n + 1):
  a[i] = (p * a[i - 1] + a[i - 2]) % MOD
  s[i] = (s[i - 1] + a[i]) % MOD

ans = 0
for i in range(1, n + 1):
  ans = (ans + a[i] * (s[n] - s[i - 1])) % MOD

print(ans)
0