結果
問題 |
No.658 テトラナッチ数列 Hard
|
ユーザー |
![]() |
提出日時 | 2023-01-30 12:33:23 |
言語 | PyPy3 (7.3.15) |
結果 |
TLE
|
実行時間 | - |
コード長 | 837 bytes |
コンパイル時間 | 286 ms |
コンパイル使用メモリ | 82,432 KB |
実行使用メモリ | 78,228 KB |
最終ジャッジ日時 | 2024-06-30 01:45:15 |
合計ジャッジ時間 | 15,142 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 6 TLE * 2 |
ソースコード
def Matprod(A, B, mod): N = len(A) temp = [[0] * N for i in range(N)] for i in range(N): for j in range(N): for k in range(N): temp[i][j] += A[i][k] * B[k][j] temp[i][j] %= mod return temp def Matpow(A, M, mod): N = len(A) Mat = [[0] * N for i in range(N)] for i in range(N): Mat[i][i] = 1 for i in range(64): if (M >> i) & 1: Mat = Matprod(Mat, A, mod) A = Matprod(A, A, mod) return Mat ans = [0, 0, 0, 1] Q = int(input()) for _ in range(Q): n = int(input()) n -= 1 if n <= 3: print(ans[n]) continue M = [[0] * 4 for i in range(4)] for i in range(4): M[0][i] = 1 if i < 3: M[i + 1][i] = 1 M = Matpow(M, n - 3, 17) print(M[0][0])