結果
問題 |
No.891 隣接3項間の漸化式
|
ユーザー |
|
提出日時 | 2021-01-03 21:27:31 |
言語 | PyPy3 (7.3.15) |
結果 |
TLE
|
実行時間 | - |
コード長 | 641 bytes |
コンパイル時間 | 135 ms |
コンパイル使用メモリ | 81,976 KB |
実行使用メモリ | 84,404 KB |
最終ジャッジ日時 | 2024-10-13 15:26:25 |
合計ジャッジ時間 | 4,344 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 10 TLE * 1 -- * 28 |
ソースコード
mod = 10 ** 9 + 7 def matrix_multiplication(a, b): n = len(a) res = [[0] * n for i in range(n)] for i in range(n): for k in range(n): for j in range(n): res[i][j] += a[i][k] * b[k][j] res[i][j] %= mod return res def matrix_pow(a, k): n = len(a) res = [[0] * n for i in range(n)] for i in range(n): res[i][i] = 1 while k: if k & 1: res = matrix_multiplication(res, a) a = matrix_multiplication(a, a) k >>= 1 return res a, b, n = map(int, input().split()) M = [[a, b], [1, 0]] print(matrix_pow(M, n - 1)[0][0])