#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
• 実行時間: 288 ms / 4,000 ms
• コード長: 757 bytes
• コンパイル時間: 2,962 ms
• コンパイル使用メモリ: 248,100 KB
• 実行使用メモリ: 4,380 KB
• 最終ジャッジ日時: 2023-09-02 06:27:21
• 合計ジャッジ時間: 6,672 ms
• ジャッジサーバーID （参考情報）: judge15 / judge11

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	2 ms; 4,376 KB
testcase_01	AC	287 ms; 4,376 KB
testcase_02	AC	32 ms; 4,376 KB
testcase_03	AC	1 ms; 4,376 KB
testcase_04	AC	1 ms; 4,376 KB
testcase_05	AC	287 ms; 4,376 KB
testcase_06	AC	122 ms; 4,376 KB
testcase_07	AC	1 ms; 4,376 KB
testcase_08	AC	35 ms; 4,380 KB
testcase_09	AC	20 ms; 4,380 KB
testcase_10	AC	52 ms; 4,376 KB
testcase_11	AC	75 ms; 4,376 KB
testcase_12	AC	118 ms; 4,380 KB
testcase_13	AC	148 ms; 4,380 KB
testcase_14	AC	250 ms; 4,376 KB
testcase_15	AC	287 ms; 4,380 KB
testcase_16	AC	288 ms; 4,376 KB
testcase_17	AC	288 ms; 4,380 KB
testcase_18	AC	26 ms; 4,380 KB
testcase_19	AC	1 ms; 4,376 KB
testcase_20	AC	1 ms; 4,380 KB
testcase_21	AC	52 ms; 4,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;
}