結果

問題 No.978 Fibonacci Convolution Easy
ユーザー NatsubiSogan
提出日時 2020-09-30 22:07:08
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
WA  
実行時間 -
コード長 248 bytes
コンパイル時間 92 ms
コンパイル使用メモリ 12,672 KB
実行使用メモリ 89,344 KB
最終ジャッジ日時 2024-07-06 12:52:31
合計ジャッジ時間 11,215 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 1 WA * 20
権限があれば一括ダウンロードができます

ソースコード

diff # 

n, p = map(int, input().split())
fib = [0, 1]
for i in range(n - 2):
    fib.append((fib[-1] * p + fib[-2]) % (10 ** 9 + 7))
cnt = 0
for i in fib:
    cnt += i ** 2
print(((sum(fib) ** 2 + cnt) // 2) % (10 ** 9 + 7) if not n == 1 and p == 1 else 0)
0