結果

問題 No.1403 調和の魔法陣
ユーザー lam6er
提出日時 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
権限があれば一括ダウンロードができます

ソースコード

diff #
プレゼンテーションモードにする

MOD = 998244353
# Precompute four variables sum (for H=2, W=2 case)
max_four = 4 * 9
four_dp = [0] * (max_four + 1)
four_dp[0] = 1
for _ in range(4):
new_dp = [0] * (len(four_dp) + 9)
for s in range(len(four_dp)):
if four_dp[s] == 0:
continue
for k in range(10):
new_dp[s + k] = (new_dp[s + k] + four_dp[s]) % MOD
four_dp = new_dp[:max_four + 1]
T = int(input())
for _ in range(T):
W, H, X = map(int, input().split())
# Determine the case
if 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 = W
else:
K = H
if 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 # 54
if X < 0 or X > max_X:
print(0)
else:
total = 0
for s in range(0, X + 1):
term = pow(s + 1, 2, MOD) * pow(X - s + 1, 2, MOD)
term %= MOD
total = (total + term) % MOD
print(total % MOD)
else:
print(0)
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0