#900944 (C++17) No.2409 Strange Werewolves

• 問題: No.2409 Strange Werewolves
• ユーザー: Carpenters-Cat
• 提出日時: 2023-08-11 23:03:37
• 言語: C++17 (gcc 12.3.0 + boost 1.83.0)
• 結果: AC
• 実行時間: 18 ms / 2,000 ms
• コード長: 859 bytes
• コンパイル時間: 2,088 ms
• コンパイル使用メモリ: 203,456 KB
• 実行使用メモリ: 12,544 KB
• 最終ジャッジ日時: 2024-04-29 14:14:20
• 合計ジャッジ時間: 3,002 ms
• ジャッジサーバーID （参考情報）: judge5 / judge4

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	2 ms; 5,248 KB
testcase_01	AC	2 ms; 5,376 KB
testcase_02	AC	2 ms; 5,376 KB
testcase_03	AC	18 ms; 12,544 KB
testcase_04	AC	16 ms; 12,544 KB
testcase_05	AC	10 ms; 7,936 KB
testcase_06	AC	17 ms; 12,544 KB
testcase_07	AC	17 ms; 12,544 KB
testcase_08	AC	10 ms; 8,320 KB
testcase_09	AC	11 ms; 8,832 KB
testcase_10	AC	6 ms; 5,888 KB
testcase_11	AC	3 ms; 5,376 KB
testcase_12	AC	6 ms; 5,888 KB
testcase_13	AC	10 ms; 8,192 KB
testcase_14	AC	11 ms; 9,088 KB
testcase_15	AC	11 ms; 8,704 KB
testcase_16	AC	8 ms; 7,808 KB
testcase_17	AC	3 ms; 5,376 KB

ソースコード

#include <bits/stdc++.h>
using namespace std;
using ll = long long;
ll const m = 998244353;
ll mpow(ll a, ll n) {
	ll ret = 1;
	while (n) {
		if (n & 1) ret = (ret * a) % m;
		a = (a * a) % m;
		n >>= 1;
	}
	return ret;
}
int main () {
	int X, Y, Z, W;
	cin >> X >> Y >> Z >> W;
	if (W == 0) {
		swap(X, Y); swap(Z, W);
	}
	int dx = X - Z - 1, dy = Y - W;
	ll ans = dx + 1;
	for (int i = 1; i <= dx + dy; i ++) {
		ans = (ans * i) % m;
	}
	std::vector<ll> kai(X + Y, 1), fkai(X + Y);
	for (int i = 1; i < X + Y; i ++) {
		kai[i] = (kai[i - 1] * i) % m;
	}
	fkai[X + Y - 1] = mpow(kai[X + Y - 1], m - 2);
	for (int i = X + Y - 1; i > 0; i --) {
		fkai[i - 1] = (fkai[i] * i) % m;
	}
	ll cb1 = ((kai[X] * fkai[dx + 1]) % m * fkai[Z]) % m;
	ll cb2 = ((kai[Y] * fkai[dy]) % m * fkai[W]) % m;
	ans = (ans * cb1) % m;
	ans = (ans * cb2) % m;
	cout << ans << endl;
}