結果

問題 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)
0