結果
問題 |
No.718 行列のできるフィボナッチ数列道場 (1)
|
ユーザー |
![]() |
提出日時 | 2024-09-26 20:40:14 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 48 ms / 2,000 ms |
コード長 | 678 bytes |
コンパイル時間 | 219 ms |
コンパイル使用メモリ | 82,432 KB |
実行使用メモリ | 52,992 KB |
最終ジャッジ日時 | 2024-09-26 20:40:27 |
合計ジャッジ時間 | 2,110 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 20 |
ソースコード
def mat_mul(a, b): res = [[0] * len(b[0]) for _ in range(len(a))] for i in range(len(a)): for j in range(len(b[0])): for k in range(len(b)): res[i][j] = (res[i][j] + a[i][k] * b[k][j]) % MOD return res def mat_pow(m, k): res = [[0] * len(m) for _ in range(len(m))] for i in range(len(m)): res[i][i] = 1 while k: if k & 1: res = mat_mul(res, m) m = mat_mul(m, m) k >>= 1 return res MOD=10**9+7 n=int(input()) def fib(n): X=[[1], [0]] T=[[1,1], [1,0]] res=mat_pow(T,n-1) res=mat_mul(res,X) return res[0][0] print(fib(n)*fib(n+1)%MOD)