結果
問題 | No.2441 行列累乗 |
ユーザー | るこーそー |
提出日時 | 2024-09-26 20:24:28 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 40 ms / 2,000 ms |
コード長 | 590 bytes |
コンパイル時間 | 331 ms |
コンパイル使用メモリ | 82,040 KB |
実行使用メモリ | 53,804 KB |
最終ジャッジ日時 | 2024-09-26 20:24:31 |
合計ジャッジ時間 | 1,878 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
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])return resdef mat_pow(m, k):res = [[0] * len(m) for _ in range(len(m))]for i in range(len(m)):res[i][i] = 1while k:if k & 1:res = mat_mul(res, m)m = mat_mul(m, m)k >>= 1return resM=[list(map(int,input().split())) for _ in range(2)]res=mat_pow(M,3)for row in res:print(*row)