結果

問題 No.2683 Two Sheets
ユーザー FromBooska
提出日時 2024-03-21 12:30:02
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 385 ms / 2,000 ms
コード長 829 bytes
コンパイル時間 200 ms
コンパイル使用メモリ 82,452 KB
実行使用メモリ 75,648 KB
最終ジャッジ日時 2024-09-30 10:06:16
合計ジャッジ時間 4,095 ms
ジャッジサーバーID
(参考情報)
judge4 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 13
権限があれば一括ダウンロードができます

ソースコード

diff #

# ダブりを引く、1次元ずつ計算する、はよかったがTLEした
# 縦のダブります数期待値、横のダブります数期待値を求める、あるマスをダブりとするための枠の2乗でO(1)で可能

H, W, A, B = map(int, input().split())
mod = 998244353

tate_EV = 0
HA_inv = pow(H+1-A, mod-2, mod)
for i in range(H):
    # left, rightはiを含む枠の左端の位置
    left = max(0, i-A+1)
    right = min(H-A, i)
    calc = ((right+1-left)*HA_inv)**2
    tate_EV += calc
    tate_EV %= mod
    
yoko_EV = 0
WB_inv = pow(W+1-B, mod-2, mod)
for j in range(W):
    # left, rightはiを含む枠の左端の位置
    left = max(0, j-B+1)
    right = min(W-B, j)
    calc = ((right+1-left)*WB_inv)**2
    yoko_EV += calc
    yoko_EV %= mod
    
ans = (2*A*B - tate_EV*yoko_EV)%mod
print(ans)
0