結果
問題 | No.2529 Treasure Hunter |
ユーザー |
![]() |
提出日時 | 2023-11-03 22:15:11 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 137 ms / 2,000 ms |
コード長 | 821 bytes |
コンパイル時間 | 609 ms |
コンパイル使用メモリ | 82,324 KB |
実行使用メモリ | 76,800 KB |
最終ジャッジ日時 | 2024-09-25 20:39:00 |
合計ジャッジ時間 | 2,528 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 1 |
other | AC * 22 |
ソースコード
import sysinput = sys.stdin.readlinedef Matprod(A, B, mod, N):temp = [0] * N*Nfor i in range(N):for j in range(N):ij = i * N + jfor k in range(N):temp[ij] += A[i*N+k] * B[k*N+j]temp[ij] %= modreturn tempdef Matpow_Linear(A, M, mod, N):Mat = [0] * N*Nfor i in range(N):Mat[i*N+i] = 1while M:if M & 1:Mat = Matprod(Mat, A, mod, N)A = Matprod(A, A, mod, N)M >>= 1return MatT = int(input())mod = 998244353for _ in range(T):W, H = map(int, input().split())M = [1, 1, 1, W, W - 1, W - 2, max(0, W * (W - 3)//2), (W - 3) * (W - 2)//2, max(0, (W - 4) * (W - 3)//2 + 1)]M = Matpow_Linear(M, H, mod, 3)print( (M[0] + M[3] + M[6]) % mod )