結果

問題 No.2457 Stampaholic (Easy)
ユーザー Carpenters-CatCarpenters-Cat
提出日時 2023-09-02 00:06:50
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 1,082 ms / 4,000 ms
コード長 2,334 bytes
コンパイル時間 4,365 ms
コンパイル使用メモリ 265,488 KB
実行使用メモリ 4,384 KB
最終ジャッジ日時 2023-09-02 00:06:59
合計ジャッジ時間 8,629 ms
ジャッジサーバーID
（参考情報） 		judge13 / judge16
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
4,376 KB
testcase_01 AC 286 ms
4,376 KB
testcase_02 AC 32 ms
4,380 KB
testcase_03 AC 2 ms
4,376 KB
testcase_04 AC 2 ms
4,380 KB
testcase_05 AC 319 ms
4,376 KB
testcase_06 AC 116 ms
4,380 KB
testcase_07 AC 1 ms
4,384 KB
testcase_08 AC 34 ms
4,380 KB
testcase_09 AC 21 ms
4,380 KB
testcase_10 AC 54 ms
4,384 KB
testcase_11 AC 73 ms
4,380 KB
testcase_12 AC 112 ms
4,376 KB
testcase_13 AC 151 ms
4,376 KB
testcase_14 AC 238 ms
4,380 KB
testcase_15 AC 295 ms
4,380 KB
testcase_16 AC 294 ms
4,376 KB
testcase_17 AC 1,082 ms
4,376 KB
testcase_18 AC 88 ms
4,376 KB
testcase_19 AC 3 ms
4,376 KB
testcase_20 AC 2 ms
4,376 KB
testcase_21 AC 55 ms
4,376 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#include <atcoder/all>
using namespace std;
using namespace atcoder;
using mint = modint998244353;
// using mint = double;
using ll = long long;
mint mpow(mint a, ll n) {
	mint ret = 1;
	while (n) {
		if (n & 1) ret *= a;
		n >>= 1;
		a = a * a;
	}
	return ret;
}
int main () {
	ll H, W;
	ll N, K;
	cin >> H >> W >> N >> K;
	if (H > W) swap(H, W);
	mint h = H - K + 1, w = W - K + 1;
	if (H < 2 * K - 1) {
		if (W < 2 * K - 1) {
            // puts("A");
			mint ans = 0;
			for (ll i = 0; i < H; i ++) {
				for (ll j = 0; j < W; j ++) {
					ll dh = (min(i, H - K) - max(0ll, i - K + 1)) + 1;
					ll dw = (min(j, W - K) - max(0ll, j - K + 1)) + 1;
					mint p = dh * dw;
					p /= h * w;
					ans += 1 - mpow(1 - p, N);
				}
			}
			cout << ans.val() << endl;
            // cout << setprecision(5) << ans << endl;
		} else {
            // puts("B");
			mint ans = 0;
			for (ll i = 0; i < H; i ++) {
				ll dh = min(i, H - K) - max(0ll, i - K + 1) + 1;
				for (ll j = 0; j < K - 1; j ++) {
					ll dw = min(j, W - K) - max(0ll, j - K + 1) + 1;
					mint p = dh * dw;
					p /= h * w;
					ans += 1 - mpow(1 - p, N);
				}
				ll wn = W - 2 * (K - 1);
				mint p_ = dh * (ll)(K);
				p_ /= h * w;
				ans += (1 - mpow(1 - p_, N)) * wn;
				for (ll j = W - K + 1; j < W; j ++) {
					ll dw = min(j, W - K) - max(0ll, j - K + 1) + 1;
					mint p = dh * dw;
					p /= h * w;
					ans += 1 - mpow(1 - p, N);
				}
			}
			cout << ans.val() << endl;
            // cout << setprecision(5) << ans << endl;
		}
	} else {
        // puts("C");
		mint ans = 0;
		for (ll i = 0; i < K - 1; i ++) {
			for (ll j = 0; j < K - 1; j ++) {
				ll dh = i + 1, dw = j + 1;
				mint p = dh * dw;
				p /= h * w;
				mint p2 = 1 - mpow(1 - p, N);
				ans += 4 * p2;
			}
		}
		for (ll i = 0; i < K - 1; i ++) {
			ll dh = i + 1, dw = K;
			mint p = dh * dw;
			p /= h * w;
			mint p2 = 1 - mpow(1 - p, N);
			ans += p2 * 2 * (W - 2 * (K - 1));
		}
		for (ll j = 0; j < K - 1; j ++) {
			ll dh = K, dw = j + 1;
			mint p = dh * dw;
			p /= h * w;
			mint p2 = 1 - mpow(1 - p, N);
			ans += p2 * 2 * (H - 2 * (K - 1));
		}
		mint p = K * K;
		p /= h * w;
		mint p2 = 1 - mpow(1 - p, N);
		ans += p2 * (H - 2 * (K - 1)) * (W - 2 * (K - 1));
		cout << ans.val() << endl;
        // cout << setprecision(5) << ans << endl;
	}
}
0