#910435 (C++23) No.2457 Stampaholic (Easy)

• 問題: No.2457 Stampaholic (Easy)
• ユーザー: hiro1729
• 提出日時: 2023-09-02 06:27:14
• 言語: C++23 (gcc 12.3.0 + boost 1.83.0)
• 結果: AC
• 実行時間: 293 ms / 4,000 ms
• コード長: 757 bytes
• コンパイル時間: 2,887 ms
• コンパイル使用メモリ: 249,064 KB
• 実行使用メモリ: 5,376 KB
• 最終ジャッジ日時: 2024-06-11 13:21:50
• 合計ジャッジ時間: 5,889 ms
• ジャッジサーバーID （参考情報）: judge1 / judge4

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	1 ms; 5,248 KB
testcase_01	AC	259 ms; 5,376 KB
testcase_02	AC	30 ms; 5,376 KB
testcase_03	AC	1 ms; 5,376 KB
testcase_04	AC	2 ms; 5,376 KB
testcase_05	AC	253 ms; 5,376 KB
testcase_06	AC	114 ms; 5,376 KB
testcase_07	AC	2 ms; 5,376 KB
testcase_08	AC	32 ms; 5,376 KB
testcase_09	AC	18 ms; 5,376 KB
testcase_10	AC	47 ms; 5,376 KB
testcase_11	AC	68 ms; 5,376 KB
testcase_12	AC	108 ms; 5,376 KB
testcase_13	AC	130 ms; 5,376 KB
testcase_14	AC	219 ms; 5,376 KB
testcase_15	AC	258 ms; 5,376 KB
testcase_16	AC	289 ms; 5,376 KB
testcase_17	AC	293 ms; 5,376 KB
testcase_18	AC	24 ms; 5,376 KB
testcase_19	AC	1 ms; 5,376 KB
testcase_20	AC	2 ms; 5,376 KB
testcase_21	AC	47 ms; 5,376 KB

ソースコード

#include <bits/stdc++.h>
#include <atcoder/modint>
using namespace std;
using ll = long long;
using mint = atcoder::modint998244353;

int main() {
	ll h, w, n, k;
	cin >> h >> w >> n >> k;
	vector<mint> r(k + 1), c(k + 1);
	if (k * 2 <= h) {
		for (int i = 1; i <= k; i++) r[i] += 2;
		r[k] += h - k * 2;
	} else {
		for (int i = 1; i <= h - k; i++) r[i] += 2;
		r[h - k + 1] += 2 * k - h;
	}
	if (k * 2 <= w) {
		for (int i = 1; i <= k; i++) c[i] += 2;
		c[k] += w - k * 2;
	} else {
		for (int i = 1; i <= w - k; i++) c[i] += 2;
		c[w - k + 1] += 2 * k - w;
	}
	mint ans = 0, m = (h - k + 1) * (w - k + 1);
	for (int i = 1; i <= k; i++) {
		for (int j = 1; j <= k; j++) ans += r[i] * c[j] * (1 - ((m - i * j) / m).pow(n));
	}
	cout << ans.val() << endl;
}