結果
問題 | No.978 Fibonacci Convolution Easy |
ユーザー |
![]() |
提出日時 | 2020-01-31 21:46:16 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 326 ms / 2,000 ms |
コード長 | 458 bytes |
コンパイル時間 | 203 ms |
コンパイル使用メモリ | 82,176 KB |
実行使用メモリ | 105,984 KB |
最終ジャッジ日時 | 2024-09-18 20:54:39 |
合計ジャッジ時間 | 4,292 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 21 |
ソースコード
def main(): import sys input = sys.stdin.readline N, p = map(int, input().split()) mod = 1000000007 if N == 1: print(0) exit() A = [0] * N A[1] = 1 S = [0] * N S[1] = 1 for i in range(2, N): A[i] = ((p*A[i-1])%mod + A[i-2])%mod S[i] = S[i-1] + A[i] ans = 0 for i in range(N): ans = (ans + (A[i] * S[i])%mod)%mod print(ans) if __name__ == '__main__': main()