結果
問題 | No.2381 Gift Exchange Party |
ユーザー |
![]() |
提出日時 | 2023-07-14 23:47:58 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
WA
|
実行時間 | - |
コード長 | 1,627 bytes |
コンパイル時間 | 5,690 ms |
コンパイル使用メモリ | 304,380 KB |
最終ジャッジ日時 | 2025-02-15 14:48:53 |
ジャッジサーバーID (参考情報) |
judge5 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 14 WA * 8 |
ソースコード
#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 / 4using 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 << endlusing 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];if(n >= p) {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;}