結果
問題 | No.718 行列のできるフィボナッチ数列道場 (1) |
ユーザー |
|
提出日時 | 2022-01-09 20:37:51 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 46 ms / 2,000 ms |
コード長 | 480 bytes |
コンパイル時間 | 271 ms |
コンパイル使用メモリ | 82,104 KB |
実行使用メモリ | 61,320 KB |
最終ジャッジ日時 | 2024-11-14 10:30:21 |
合計ジャッジ時間 | 2,265 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 20 |
ソースコード
N = int(input()) P = 10 ** 9 + 7 A = [[1,1,1,0],[1,0,0,1],[2,0,1,0],[0,0,0,1]] def seki(X,Y): ans = [[0] * 4 for _ in range(4)] for i in range(4): for j in range(4): for k in range(4): ans[i][j] = (ans[i][j] + X[i][k]*Y[k][j]) % P return ans tmp = [[1,0,0,0],[0,1,0,0],[0,0,1,0],[0,0,0,1]] u = N-1 while u: if u & 1: tmp = seki(tmp,A) A = seki(A,A) u >>= 1 gn = tmp[0][0] + tmp[0][1] + tmp[0][3] print(gn%P)