結果

問題 No.3006 ベイカーの問題
コンテスト
ユーザー sasa8uyauya
提出日時 2025-01-17 22:53:35
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 60 ms / 2,000 ms
コード長 524 bytes
コンパイル時間 2,165 ms
コンパイル使用メモリ 81,920 KB
実行使用メモリ 60,544 KB
最終ジャッジ日時 2025-01-17 22:54:25
合計ジャッジ時間 2,713 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 24
権限があれば一括ダウンロードができます

ソースコード

diff # 

a,b,n=map(int,input().split())
n-=1
l=4
M=998244353

def seki(a,b):
  c=[[0]*l for i in range(l)]
  for i in range(l):
    for j in range(l):
      for k in range(l):
        c[i][j]+=a[i][k]*b[k][j]
        c[i][j]%=M
  return c

A=[
  [a,-5*b,0,0],
  [b,a,0,0],
  [a,-5*b,1,0],
  [b,a,0,1]
]
B=[
  [1,0,0,0],
  [0,1,0,0],
  [0,0,1,0],
  [0,0,0,1]
]

for i in range(61):
  if (n>>i)&1:
    B=seki(A,B)
  A=seki(A,A)

sx=(B[2][0]*a+B[2][1]*b+B[2][2]*a+B[2][3]*b)%M
sy=(B[3][0]*a+B[3][1]*b+B[3][2]*a+B[3][3]*b)%M
print(sx,sy)
0