結果
問題 |
No.2381 Gift Exchange Party
|
ユーザー |
|
提出日時 | 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 |
ソースコード
#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'; }