結果

問題 No.2582 Random Average^K
ユーザー fdironia
提出日時 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
権限があれば一括ダウンロードができます

ソースコード

diff #

#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;
}
0