結果
問題 | No.1403 調和の魔法陣 |
ユーザー |
![]() |
提出日時 | 2025-03-26 15:57:18 |
言語 | PyPy3 (7.3.15) |
結果 |
WA
|
実行時間 | - |
コード長 | 1,638 bytes |
コンパイル時間 | 400 ms |
コンパイル使用メモリ | 82,636 KB |
実行使用メモリ | 77,908 KB |
最終ジャッジ日時 | 2025-03-26 15:58:06 |
合計ジャッジ時間 | 3,431 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | WA * 19 |
ソースコード
MOD = 998244353# Precompute four variables sum (for H=2, W=2 case)max_four = 4 * 9four_dp = [0] * (max_four + 1)four_dp[0] = 1for _ in range(4):new_dp = [0] * (len(four_dp) + 9)for s in range(len(four_dp)):if four_dp[s] == 0:continuefor k in range(10):new_dp[s + k] = (new_dp[s + k] + four_dp[s]) % MODfour_dp = new_dp[:max_four + 1]T = int(input())for _ in range(T):W, H, X = map(int, input().split())# Determine the caseif H >= 3 and W >= 3:print(1 if X == 0 else 0)elif H == 1 or W == 1:if (H == 1 and W == 1) or (W == 1 and H == 1):print(1 if 0 <= X <= 9 else 0)else:if H == 1:K = Welse:K = Hif K == 2:if X < 0 or X > 18:print(0)else:if X <= 9:print((X + 1) % MOD)else:print((19 - X) % MOD)else:print(1 if 0 <= X <=9 else 0)elif H == 2 and W == 2:if X < 0 or X > 36:print(0)else:print(four_dp[X] % MOD)elif (H == 2 and W >= 2) or (W == 2 and H >= 2):max_X = 6 * 9 # 54if X < 0 or X > max_X:print(0)else:total = 0for s in range(0, X + 1):term = pow(s + 1, 2, MOD) * pow(X - s + 1, 2, MOD)term %= MODtotal = (total + term) % MODprint(total % MOD)else:print(0)