結果

問題 No.3006 ベイカーの問題
ユーザー PNJ
提出日時 2025-01-17 23:09:16
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 62 ms / 2,000 ms
コード長 845 bytes
コンパイル時間 176 ms
コンパイル使用メモリ 81,792 KB
実行使用メモリ 62,592 KB
最終ジャッジ日時 2025-01-17 23:09:29
合計ジャッジ時間 2,351 ms
ジャッジサーバーID
(参考情報)
judge2 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 24
権限があれば一括ダウンロードができます

ソースコード

diff #
プレゼンテーションモードにする

def matrix_prod(A, B, mod = 998244353):
l, m, n = len(A), len(B), len(B[0])
C = [[0 for _ in range(n)] for _ in range(l)]
for i in range(l):
for j in range(m):
a = A[i][j]
for k in range(n):
C[i][k] = (C[i][k] + a * B[j][k]) % mod
return C
def matrix_pow(A, K, mod = 998244353):
n = len(A)
if K == 0:
return [[int(i == j) for j in range(n)] for i in range(n)]
D = K.bit_length()
res = [[A[i][j] for j in range(n)] for i in range(n)]
for d in range(D - 2, -1, -1):
res = matrix_prod(res, res, mod)
if (K >> d) & 1:
res = matrix_prod(A, res, mod)
return res
mod = 998244353
x, y, N = map(int, input().split())
b = [[x], [y], [x], [y]]
A = [[x, -5 * y, 0, 0], [y, x, 0, 0], [x, -5 * y, 1, 0], [y, x, 0, 1]]
b = matrix_prod(matrix_pow(A, N - 1), b)
print(b[2][0] % mod, b[3][0] % mod)
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0