結果

問題 No.3160 Party Game
コンテスト
ユーザー pia019
提出日時 2025-05-24 12:14:52
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 33 ms / 2,000 ms
コード長 1,702 bytes
コンパイル時間 5,662 ms
コンパイル使用メモリ 332,536 KB
実行使用メモリ 22,772 KB
最終ジャッジ日時 2025-05-27 22:07:02
合計ジャッジ時間 7,942 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 38
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#include <atcoder/all>
using namespace std;
typedef long long ll;
const int INF = 1<<30;
const ll INFLL = 1LL<<60;
const ll MOD = 998244353;
const double INFD = 1.0E18;
const int dx[4] = {1, 0, -1, 0};
const int dy[4] = {0, -1, 0, 1};
//const int dx[8] = {1, 1, 0, -1, -1, -1, 0, 1};
//const int dy[8] = {0, 1, 1, 1, 0, -1, -1, -1};
using Pair = pair<ll, ll>;
using Graph = vector<vector<pair<int, int>>>;
using mint = atcoder::modint998244353;

struct combination{
    vector<mint> fact, ifact;
    combination(int n):fact(n+1), ifact(n+1){
        assert(n < MOD);
        fact[0] = 1;
        for (int i=1; i <= n; i++) fact[i] = fact[i-1] * i;
        ifact[n] = fact[n].inv();
        for (int i=n; i >= 1; i--) ifact[i-1] = ifact[i] * i;
    }
    
    mint operator()(int n, int k){
        if (k < 0 || n < k) return 0;
        return fact[n] * ifact[k] * ifact[n-k];
    }
} comb(2000005);


int main(){
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    cout << fixed << setprecision(15);
    ll n, m; cin >> n >> m;
    
    if (n == 1){
        mint ans = (m - 1) * mint(2).inv();
        cout << ans.val() << endl;
        return 0;
    }

    vector<mint> vec(m);
    vec[0] = comb(m + n, n) - n; //独り占めだけだめなことに注意
    mint ans = 0;
    for (ll x = 0; x <= m; x++){
        if (m - (x + 1) * n < 0) vec[x + 1] = 0;
        else vec[x + 1] = comb(m - (x + 1) * n + n, n);
        //total += vec[x];
        ans += x * (vec[x] - vec[x + 1]);
        //cerr << x << ' ' << ans.val () << endl;
    }
    
    cerr << vec[0].val() << " " << ans.val() << endl;
    
    cout << (ans * vec[0].inv()).val() << endl;
    
    return 0;
}
0