結果
問題 | No.2457 Stampaholic (Easy) |
ユーザー | minimum |
提出日時 | 2023-09-01 22:10:05 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 843 ms / 4,000 ms |
コード長 | 797 bytes |
コンパイル時間 | 487 ms |
コンパイル使用メモリ | 82,176 KB |
実行使用メモリ | 65,536 KB |
最終ジャッジ日時 | 2025-01-03 08:54:03 |
合計ジャッジ時間 | 7,891 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 19 |
ソースコード
from collections import defaultdict MOD = 998244353 def f(x, K): ans = defaultdict(int) if x <= 2 * K: for i in range(x): ans[(min(i + K - 1, x - 1) - max(i - K + 1, 0) + 2 - K) % MOD] += 1 else: for i in range(K): ans[(min(i + K - 1, x - 1) - max(i - K + 1, 0) + 2 - K) % MOD] += 1 for i in range(x - 1, x - 1 - K, -1): ans[(min(i + K - 1, x - 1) - max(i - K + 1, 0) + 2 - K) % MOD] += 1 ans[K] += x - 2 * K return ans H, W, N, K = map(int, input().split()) s0 = f(H, K) s1 = f(W, K) C = pow(((H - K + 1) * (W - K + 1)) % MOD, MOD - 2, MOD) ans = 0 for i in s0: for j in s1: res = (1 - pow((1 - C * i * j) % MOD, N, MOD)) % MOD ans += res * s0[i] * s1[j] ans %= MOD print(ans)