結果
問題 | No.3006 ベイカーの問題 |
ユーザー |
![]() |
提出日時 | 2025-01-17 23:04:14 |
言語 | PyPy3 (7.3.15) |
結果 |
WA
|
実行時間 | - |
コード長 | 748 bytes |
コンパイル時間 | 332 ms |
コンパイル使用メモリ | 82,048 KB |
実行使用メモリ | 60,160 KB |
最終ジャッジ日時 | 2025-01-17 23:04:41 |
合計ジャッジ時間 | 2,803 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 23 WA * 1 |
ソースコード
x,y,N=map(int, input().split())if N==1:print(x,y)exit()mod=998244353def matmul(A, B):Ah, Bh, Bw = len(A), len(B), len(B[0])C = [[0 for _ in range(Bw)] for _ in range(Ah)]for i in range(Ah):for j in range(Bw):for k in range(Bh):C[i][j] += A[i][k] * B[k][j] % modC[i][j] %= modreturn C# Mのk乗を効率的に計算するdef doubling(M, k):k -= 1Mc = M.copy()while k > 0:if k & 1 == 1:Mc = matmul(Mc, M)M = matmul(M, M) # Mの(2のi乗)の乗 を計算するk >>= 1return McM = [[x,-5*y,x], [y, x,y],[0,0,1]]F = [[x], [y],[1]]Mn = doubling(M, N-1)G = matmul(Mn,F)print(G[0][0]%mod,G[1][0]%mod)