結果
問題 | No.2582 Random Average^K |
ユーザー |
|
提出日時 | 2023-12-10 00:29:15 |
言語 | C++23 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 455 ms / 2,000 ms |
コード長 | 940 bytes |
コンパイル時間 | 777 ms |
コンパイル使用メモリ | 79,028 KB |
実行使用メモリ | 50,176 KB |
最終ジャッジ日時 | 2024-09-27 03:54:55 |
合計ジャッジ時間 | 3,816 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 15 |
ソースコード
#include <iostream> using namespace std; constexpr int Z = 998244353; int bexp(int a, int n) { if (n == 0) { return 1; } if (n % 2 == 1) { return 1ll * a * bexp(a, n - 1) % Z; } int t = bexp(a, n / 2); return 1ll * t * t % Z; } int main() { int n, k; cin >> n >> k; int inv[n+k+1]; for (int i = 0; i <= n + k; i++) { inv[i] = i <= 1 ? i : 1ll * (Z - Z / i) * inv[Z%i] % Z; } int ifac[n+1]; for (int i = 0; i <= n; i++) { ifac[i] = i == 0 ? 1 : 1ll * inv[i] * ifac[i-1] % Z; } int ans = 0; for (int i = 0; i <= n; i++) { int t = 1ll * ifac[i] * ifac[n-i] % Z * bexp(n - i, n + k) % Z; ans = (0ll + ans + (i % 2 == 0 ? 1 : -1) * t + Z) % Z; } ans = 1ll * ans * bexp(inv[n], k) % Z; for (int i = 1; i <= n; i++) { ans = 1ll * ans * inv[k+i] % Z * i % Z; } cout << ans << endl; return 0; }