結果
問題 | No.978 Fibonacci Convolution Easy |
ユーザー | rlangevin |
提出日時 | 2023-01-28 16:58:02 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 165 ms / 2,000 ms |
コード長 | 455 bytes |
コンパイル時間 | 2,222 ms |
コンパイル使用メモリ | 86,396 KB |
実行使用メモリ | 122,756 KB |
最終ジャッジ日時 | 2023-09-11 05:42:55 |
合計ジャッジ時間 | 5,049 ms |
ジャッジサーバーID (参考情報) |
judge13 / judge12 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 86 ms
76,452 KB |
testcase_01 | AC | 120 ms
94,072 KB |
testcase_02 | AC | 103 ms
85,132 KB |
testcase_03 | AC | 161 ms
120,696 KB |
testcase_04 | AC | 103 ms
87,240 KB |
testcase_05 | AC | 89 ms
78,284 KB |
testcase_06 | AC | 115 ms
91,792 KB |
testcase_07 | AC | 135 ms
104,880 KB |
testcase_08 | AC | 120 ms
95,784 KB |
testcase_09 | AC | 141 ms
109,332 KB |
testcase_10 | AC | 165 ms
122,756 KB |
testcase_11 | AC | 109 ms
89,300 KB |
testcase_12 | AC | 87 ms
77,840 KB |
testcase_13 | AC | 111 ms
91,404 KB |
testcase_14 | AC | 90 ms
80,332 KB |
testcase_15 | AC | 116 ms
93,448 KB |
testcase_16 | AC | 161 ms
122,668 KB |
testcase_17 | AC | 160 ms
122,468 KB |
testcase_18 | AC | 73 ms
71,196 KB |
testcase_19 | AC | 73 ms
71,188 KB |
testcase_20 | AC | 74 ms
71,204 KB |
ソースコード
N, p = map(int, input().split()) mod = 10 ** 9 + 7 P = [1] * (N + 1) for i in range(N): P[i + 1] = P[i] * p P[i + 1] %= mod invp = pow(p-1, mod-2, mod) ans = 0 A = [0] * (N + 1) A[1] = 1 for i in range(2, N + 1): A[i] = p * A[i - 1] + A[i - 2] A[i] %= mod Ac = [0] * (N + 1) Ac[0] = A[0] for i in range(1, N + 1): Ac[i] = Ac[i - 1] + A[i] Ac[i] %= mod for i in range(1, N): ans += A[i] * Ac[i] ans %= mod print(ans)