結果

問題 No.2457 Stampaholic (Easy)
ユーザー Carpenters-CatCarpenters-Cat
提出日時 2023-09-02 00:01:41
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
WA  
実行時間 -
コード長 2,052 bytes
コンパイル時間 5,683 ms
コンパイル使用メモリ 266,556 KB
実行使用メモリ 6,948 KB
最終ジャッジ日時 2024-06-11 06:21:27
合計ジャッジ時間 8,783 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 WA -
testcase_01 WA -
testcase_02 WA -
testcase_03 WA -
testcase_04 AC 2 ms
6,940 KB
testcase_05 WA -
testcase_06 WA -
testcase_07 AC 2 ms
6,940 KB
testcase_08 WA -
testcase_09 WA -
testcase_10 WA -
testcase_11 WA -
testcase_12 WA -
testcase_13 WA -
testcase_14 WA -
testcase_15 WA -
testcase_16 WA -
testcase_17 WA -
testcase_18 WA -
testcase_19 WA -
testcase_20 WA -
testcase_21 WA -
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#include <atcoder/all>
using namespace std;
using namespace atcoder;
using mint = modint998244353;
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, w = W;
	if (H < 2 * K - 1) {
		if (W < 2 * K - 1) {
			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;
		} else {
			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);
				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;
		}
	} else {
		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;
	}
}
0