結果

問題 No.2582 Random Average^K
ユーザー fdironiafdironia
提出日時 2023-12-10 00:29:15
言語 C++23
(gcc 12.3.0 + boost 1.83.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
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
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 1 ms
5,376 KB
testcase_04 AC 2 ms
5,376 KB
testcase_05 AC 1 ms
5,376 KB
testcase_06 AC 3 ms
5,376 KB
testcase_07 AC 2 ms
5,376 KB
testcase_08 AC 3 ms
5,376 KB
testcase_09 AC 102 ms
17,340 KB
testcase_10 AC 240 ms
37,500 KB
testcase_11 AC 212 ms
29,568 KB
testcase_12 AC 455 ms
50,176 KB
testcase_13 AC 211 ms
11,264 KB
testcase_14 AC 175 ms
42,624 KB
testcase_15 AC 438 ms
50,176 KB
testcase_16 AC 311 ms
46,592 KB
testcase_17 AC 325 ms
46,464 KB
権限があれば一括ダウンロードができます

ソースコード

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