結果
問題 | No.2457 Stampaholic (Easy) |
ユーザー |
|
提出日時 | 2023-09-01 22:34:05 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 518 ms / 4,000 ms |
コード長 | 2,108 bytes |
コンパイル時間 | 4,205 ms |
コンパイル使用メモリ | 256,340 KB |
最終ジャッジ日時 | 2025-02-16 16:57:23 |
ジャッジサーバーID (参考情報) |
judge4 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 19 |
ソースコード
#include <bits/stdc++.h> #include <atcoder/all> using namespace std; using namespace atcoder; struct Fast { Fast() { std::cin.tie(nullptr); ios::sync_with_stdio(false); cout << setprecision(10); } } fast; #define all(a) (a).begin(), (a).end() #define contains(a, x) ((a).find(x) != (a).end()) #define rep(i, a, b) for (int i = (a); i < (int)(b); i++) #define rrep(i, a, b) for (int i = (int)(b)-1; i >= (a); i--) #define writejoin(s, a) rep(_i, 0, (a).size()) cout << (a)[_i] << (_i + 1 < (int)(a).size() ? s : "\n"); #define YN(b) cout << ((b) ? "YES" : "NO") << "\n"; #define Yn(b) cout << ((b) ? "Yes" : "No") << "\n"; #define yn(b) cout << ((b) ? "yes" : "no") << "\n"; using ll = long long; using mint = modint998244353; int main() { ll h, w, n, k; cin >> h >> w >> n >> k; mint prob = 1; prob *= h - k + 1; prob *= w - k + 1; prob = prob.inv(); mint ans = 0; if (h <= 2 * k) { if (w <= 2 * k) { rep(i, 0, h) rep(j, 0, w) { ll x = min((ll)i, h - k) - max(0ll, i - k + 1) + 1; ll y = min((ll)j, w - k) - max(0ll, j - k + 1) + 1; ans += 1 - (1 - prob * x * y).pow(n); } } else { rep(i, 0, h) { ll x = min((ll)i, h - k) - max(0ll, i - k + 1) + 1; rep(y, 1, k) ans += (1 - (1 - prob * x * y).pow(n)) * 2; ans += (1 - (1 - prob * x * k).pow(n)) * (w - 2 * (k - 1)); } } } else { if (w <= 2 * k) { swap(h, w); rep(i, 0, h) { ll x = min((ll)i, h - k) - max(0ll, i - k + 1) + 1; rep(y, 1, k) ans += (1 - (1 - prob * x * y).pow(n)) * 2; ans += (1 - (1 - prob * x * k).pow(n)) * (w - 2 * (k - 1)); } } else { rep(x, 1, k) { rep(y, 1, k) ans += (1 - (1 - prob * x * y).pow(n)) * 2; ans += (1 - (1 - prob * x * k).pow(n)) * (w - 2 * (k - 1)); } ans *= 2; { mint tmp = 0; rep(y, 1, k) tmp += (1 - (1 - prob * k * y).pow(n)) * 2; tmp += (1 - (1 - prob * k * k).pow(n)) * (w - 2 * (k - 1)); ans += tmp * (h - 2 * (k - 1)); } } } cout << ans.val() << "\n"; }