結果
問題 |
No.526 フィボナッチ数列の第N項をMで割った余りを求める
|
ユーザー |
|
提出日時 | 2025-01-01 20:06:03 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 32 ms / 2,000 ms |
コード長 | 867 bytes |
コンパイル時間 | 183 ms |
コンパイル使用メモリ | 82,556 KB |
実行使用メモリ | 54,072 KB |
最終ジャッジ日時 | 2025-01-01 20:06:16 |
合計ジャッジ時間 | 1,373 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 12 |
ソースコード
def matrix_multiply(A, B, mod): rows_A = len(A) cols_A = len(A[0]) cols_B = len(B[0]) C = [[0] * cols_B for _ in range(rows_A)] for i in range(rows_A): for j in range(cols_B): for k in range(cols_A): C[i][j] = (C[i][j] + A[i][k] * B[k][j]) % mod return C def matrix_power(A, n, mod): rows = len(A) result = [[1 if i == j else 0 for j in range(rows)] for i in range(rows)] while n > 0: if n % 2 == 1: result = matrix_multiply(result, A, mod) A = matrix_multiply(A, A, mod) n //= 2 return result def fibonacci(n, mod): if n <= 1: return n base_matrix = [[1, 1], [1, 0]] powered_matrix = matrix_power(base_matrix, n - 1, mod) return powered_matrix[0][0] N, MOD = map(int, input().split()) result = fibonacci(N-1, MOD) print(result)