結果
問題 |
No.2381 Gift Exchange Party
|
ユーザー |
![]() |
提出日時 | 2023-07-14 23:51:16 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 79 ms / 2,000 ms |
コード長 | 1,595 bytes |
コンパイル時間 | 5,646 ms |
コンパイル使用メモリ | 304,584 KB |
最終ジャッジ日時 | 2025-02-15 14:49:30 |
ジャッジサーバーID (参考情報) |
judge1 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 22 |
ソースコード
#include <bits/extc++.h> #include <atcoder/all> using namespace std; using ll = long long; #define REP(i,n) for(int i=0;i<int(n);i++) #define FOR(i,a,b) for(int i=a;i<=int(b);i++) #define ALL(x) x.begin(),x.end() #define INF INT_MAX #define INFL LLONG_MAX / 4 using namespace atcoder; using mint = modint998244353; template<typename T> void chmin(T& a, T b) { a = min(a, b); } template<typename T> void chmax(T& a, T b) { a = max(a, b); } #define PR(x) cerr << #x << "=" << x << endl using i128 = __int128_t; struct nCkmod { vector<long long> fanc; vector<long long> fancinv; long long mod; nCkmod(long long n, long long mod_ = 998244353): fanc(n + 1), fancinv(n + 1), mod(mod_) { fanc[0] = 1; fancinv[0] = 1; for(long long i = 1; i <= n; i++) { fanc[i] = (fanc[i - 1] * i) % mod; fancinv[i] = (fancinv[i - 1] * modinv(i)) % mod; } } long long modinv(long long a) { long long b = mod, u = 1, v = 0; while (b) { long long t = a / b; a -= t * b; swap(a, b); u -= t * v; swap(u, v); } u %= mod; if (u < 0) u += mod; return u; } long long nck(int n, int k) { if(n < k) return 0; return fanc[n] * fancinv[k] % mod * fancinv[n - k] % mod; } }; int main() { int n; ll p; cin >> n >> p; nCkmod nck(200005); mint ans = nck.fanc[n]; REP(i, n / p + 1) { mint tmp = nck.fanc[n]; tmp *= nck.fancinv[n-p*i]; mint p_ = 1; p_ /= p; tmp *= p_.pow(i); tmp *= nck.fancinv[i]; ans -= tmp; // cout << tmp.val() << endl; } cout << ans.val(); return 0; }