#879779 (C++23) No.2582 Random Average^K

• 問題: No.2582 Random Average^K
• ユーザー: 遭難者
• 提出日時: 2023-06-11 01:14:37
• 言語: C++23 (gcc 12.3.0 + boost 1.83.0)
• 結果: WA
• 実行時間: -
• コード長: 995 bytes
• コンパイル時間: 7,327 ms
• コンパイル使用メモリ: 348,108 KB
• 実行使用メモリ: 6,676 KB
• 最終ジャッジ日時: 2023-12-09 23:30:32
• 合計ジャッジ時間: 9,043 ms
• ジャッジサーバーID （参考情報）: judge14 / judge15

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	2 ms; 6,676 KB
testcase_01	AC	2 ms; 6,676 KB
testcase_02	AC	2 ms; 6,676 KB
testcase_03	AC	2 ms; 6,676 KB
testcase_04	AC	2 ms; 6,676 KB
testcase_05	AC	2 ms; 6,676 KB
testcase_06	WA	-
testcase_07	AC	2 ms; 6,676 KB
testcase_08	WA	-
testcase_09	RE	-
testcase_10	RE	-
testcase_11	RE	-
testcase_12	RE	-
testcase_13	RE	-
testcase_14	AC	2 ms; 6,676 KB
testcase_15	RE	-
testcase_16	RE	-
testcase_17	RE	-

ソースコード

#pragma GCC target("avx2")
#pragma GCC optimize("O3")
#pragma GCC optimize("Ofast,unroll-loops")
#include <bits/stdc++.h>
#include <atcoder/all>
#define rep(i, n) for (int i = 0; i < n; i++)
#define ALL(a) a.begin(), a.end()
#define ll long long
using namespace std;
using namespace atcoder;
constexpr int mod = 998244353;
using mint = modint998244353;
mint fac[5050];
mint inv[5050];
void solve()
{
	fac[0] = 1;
	for (int i = 1; i < 5050; i++)
		fac[i] = i * fac[i - 1];
	inv[5049] = 1 / fac[5049];
	for (int i = 5049; i > 0; i--)
		inv[i - 1] = i * inv[i];
	int n, m;
	cin >> n >> m;
	mint ans = 0;
	for (int k = 1; k <= n; k++)
	{
		mint mul = fac[n] * inv[k] * inv[n - k];
		if ((n - k) & 1)
			mul = -mul;
		mul *= mint(k).pow(n + m);
		ans += mul;
	}
	mint ff = 1;
	for (int i = 1; i <= n; i++)
		ff *= i + m;
	ans /= ff * mint(n).pow(m);
	cout << ans.val() << '\n';
}
int main()
{
	cin.tie(nullptr);
	ios::sync_with_stdio(false);
	cout << fixed << setprecision(13);
	solve();
	return 0;
}