結果
問題 |
No.2631 Rectangle Grid Game
|
ユーザー |
|
提出日時 | 2024-02-17 01:15:48 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 33 ms / 2,000 ms |
コード長 | 1,742 bytes |
コンパイル時間 | 292 ms |
コンパイル使用メモリ | 82,588 KB |
実行使用メモリ | 54,176 KB |
最終ジャッジ日時 | 2024-09-28 23:22:31 |
合計ジャッジ時間 | 2,100 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 30 |
ソースコード
def solve(): N = H//2 M = W//2 A = N+2 B = M+2 ans = H*W%MOD*N%MOD*M%MOD ans -= A*B%MOD*N%MOD*M%MOD ans %= MOD ans += M*(M+1)//2*A%MOD*N%MOD ans %= MOD ans += N*(N+1)//2*B%MOD*M%MOD ans %= MOD ans -= (N*(N+1)//2)*(M*(M+1)//2)%MOD ans %= MOD ans *= 4 ans %= MOD return ans def solve2(): N = H//2 M = W//2 A = N+2 B = M+2 ans = H*W%MOD*N%MOD*M%MOD ans -= A*B%MOD*N%MOD*M%MOD ans %= MOD ans += M*(M+1)//2*A%MOD*N%MOD ans %= MOD ans += N*(N+1)//2*B%MOD*M%MOD ans %= MOD ans -= (N*(N+1)//2)*(M*(M+1)//2)%MOD ans %= MOD ans *= 4 ans %= MOD tmp = H*W%MOD*N%MOD tmp %= MOD tmp -= A*N%MOD tmp %= MOD tmp += N*(N+1)//2 tmp %= MOD tmp *= 2 tmp %= MOD return (ans+tmp)%MOD def solve3(): N = H//2 M = W//2 A = N+2 B = M+2 ans = H*W%MOD*N%MOD*M%MOD ans -= A*B%MOD*N%MOD*M%MOD ans %= MOD ans += M*(M+1)//2*A%MOD*N%MOD ans %= MOD ans += N*(N+1)//2*B%MOD*M%MOD ans %= MOD ans -= (N*(N+1)//2)*(M*(M+1)//2)%MOD ans %= MOD ans *= 4 ans %= MOD tmp = H*W%MOD*N%MOD tmp %= MOD tmp -= A*N%MOD tmp %= MOD tmp += N*(N+1)//2 tmp %= MOD tmp *= 2 tmp %= MOD tmp2 = H*W%MOD*M%MOD tmp2 %= MOD tmp2 -= B*M%MOD tmp2 %= MOD tmp2 += M*(M+1)//2 tmp2 %= MOD tmp2 *= 2 tmp2 %= MOD tmp3 = (H*W-1)%MOD return (ans+tmp+tmp2+tmp3)%MOD import sys input = sys.stdin.readline MOD = 998244353 H, W = map(int, input().split()) if H%2==0 and W%2==0: ans = solve() elif H%2==0: ans = solve2() elif W%2==0: H, W = W, H ans = solve2() else: ans = solve3() print(ans)