#1024731 (C++17) No.2303 Frog on Grid

• 問題: No.2303 Frog on Grid
• ユーザー: eve__fuyuki
• 提出日時: 2024-11-08 01:20:34
• 言語: C++17 (gcc 12.3.0 + boost 1.83.0)
• 結果: AC
• 実行時間: 49 ms / 2,000 ms
• コード長: 845 bytes
• コンパイル時間: 3,870 ms
• コンパイル使用メモリ: 242,092 KB
• 実行使用メモリ: 12,872 KB
• 最終ジャッジ日時: 2024-11-08 01:20:43
• 合計ジャッジ時間: 5,399 ms
• ジャッジサーバーID （参考情報）: judge3 / judge1

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	2 ms; 5,248 KB
testcase_01	AC	2 ms; 5,248 KB
testcase_02	AC	46 ms; 11,696 KB
testcase_03	AC	23 ms; 7,040 KB
testcase_04	AC	45 ms; 11,136 KB
testcase_05	AC	47 ms; 12,024 KB
testcase_06	AC	23 ms; 7,440 KB
testcase_07	AC	45 ms; 10,932 KB
testcase_08	AC	44 ms; 11,080 KB
testcase_09	AC	12 ms; 5,248 KB
testcase_10	AC	22 ms; 6,852 KB
testcase_11	AC	12 ms; 5,252 KB
testcase_12	AC	24 ms; 8,012 KB
testcase_13	AC	2 ms; 5,248 KB
testcase_14	AC	2 ms; 5,248 KB
testcase_15	AC	2 ms; 5,248 KB
testcase_16	AC	2 ms; 5,248 KB
testcase_17	AC	2 ms; 5,248 KB
testcase_18	AC	48 ms; 12,744 KB
testcase_19	AC	48 ms; 12,744 KB
testcase_20	AC	48 ms; 12,748 KB
testcase_21	AC	48 ms; 12,872 KB
testcase_22	AC	49 ms; 12,744 KB

ソースコード

#include <bits/stdc++.h>
using namespace std;

void fast_io() {
	ios_base::sync_with_stdio(false);
	cin.tie(nullptr);
}

#include <atcoder/convolution>
#include <atcoder/modint>
using namespace atcoder;
using mint = modint998244353;
int main() {
	fast_io();
	int h, w;
	cin >> h >> w;
	int N = h + w;
	vector<mint> fact(N + 1), inv(N + 1);
	fact[0] = 1;
	for (int i = 1; i <= N; i++) fact[i] = fact[i - 1] * i;
	inv[N] = fact[N].inv();
	for (int i = N; i >= 1; i--) inv[i - 1] = inv[i] * i;
	vector<mint> f(h + 1), g(w + 1);
	for (int i = 0; i <= h / 2; i++) {
		int j = h - 2 * i;
		f[i + j] = inv[i] * inv[j];
	}
	for (int i = 0; i <= w / 2; i++) {
		int j = w - 2 * i;
		g[i + j] = inv[i] * inv[j];
	}
	auto fg = convolution(f, g);
	mint ans = 0;
	for (int i = 0; i <= h + w; i++) {
		ans += fg[i] * fact[i];
	}
	cout << ans.val() << endl;
}