結果

問題 No.2381 Gift Exchange Party
コンテスト
ユーザー t98slider
提出日時 2023-07-15 17:50:00
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 6 ms / 2,000 ms
コード長 571 bytes
コンパイル時間 3,959 ms
コンパイル使用メモリ 230,684 KB
実行使用メモリ 6,944 KB
最終ジャッジ日時 2024-09-17 02:50:58
合計ジャッジ時間 4,750 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 22
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#include <atcoder/all>
using namespace std;
using ll = long long;
using mint = atcoder::modint998244353;

int main(){
    int n, p, id = 1;
    cin >> n >> p;
    vector<mint> fact(n + 1), inv(n + 1);
    fact[0] = 1;
    for(int i = 1; i <= n; i++) fact[i] = i * fact[i - 1];
    inv[n] = 1 / fact[n];
    for(int i = n; i >= 1; i--) inv[i - 1] = i * inv[i];
    mint ans = fact[n] - 1, coef = 1, div = mint(1) / p;
    while(n >= p){
    	coef *= inv[n -= p] * fact[n] * div;
    	ans -= coef * inv[id++];
    }
    cout << ans.val() << '\n';
}
0