結果

問題 No.2105 Avoid MeX
コンテスト
ユーザー baLoon8128
提出日時 2022-10-21 23:28:42
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 51 ms / 2,000 ms
コード長 984 bytes
コンパイル時間 4,050 ms
コンパイル使用メモリ 232,652 KB
実行使用メモリ 18,944 KB
最終ジャッジ日時 2024-07-01 07:40:26
合計ジャッジ時間 5,117 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 6
other AC * 19
権限があれば一括ダウンロードができます

ソースコード

diff # 

// #define _GLIBCXX_DEBUG
#include <bits/stdc++.h>
using namespace std;
#include <atcoder/all>
using namespace atcoder;
using ll = long long;
using vi = vector<int>;
using vvi = vector<vector<int>>;
using pii = pair<int, int>;
#define rep(i, n) for (int i = 0; i < (int)(n); ++i)
#define repr(i, n) for (int i = (int)(n - 1); i >= 0; --i)

using mint = modint998244353;
int main() {
    int c, x;
    cin >> c >> x;
    if (x == 0) {
        cout << mint(c + 1).inv().val() << endl;
    } else {
        c = min(c, x);
        vector<vector<mint>> dp(x + 1, vector<mint>(x + 1, 0));
        dp[c][0] = 1;
        for (int k = c + 1; k <= x; k++) {
            mint kinv = mint(k).inv();
            repr(i, x) {
                dp[k][i] += dp[k - 1][i] * i * kinv;
                dp[k][i + 1] += dp[k - 1][i] * (k - i) * kinv;
            }
        }
        mint ret = 1;
        rep(i, x + 1) ret -= dp[x][i] / (x + 1 - i);
        cout << ret.val() << endl;
    }
    return 0;
}
0