結果
問題 | No.2381 Gift Exchange Party |
ユーザー |
|
提出日時 | 2023-07-14 21:32:02 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 27 ms / 2,000 ms |
コード長 | 1,161 bytes |
コンパイル時間 | 688 ms |
コンパイル使用メモリ | 73,472 KB |
実行使用メモリ | 5,376 KB |
最終ジャッジ日時 | 2024-09-16 06:21:29 |
合計ジャッジ時間 | 1,409 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 22 |
ソースコード
#include<iostream>#include<cassert>#include<atcoder/modint>using namespace std;#include<vector>template<typename T>struct combination{vector<T>fac,ifac;combination(size_t N=0):fac(1,1),ifac(1,1){make_table(N);}void make_table(size_t N){if(fac.size()>N)return;size_t now=fac.size();N=max(N,now*2);fac.resize(N+1);ifac.resize(N+1);for(size_t i=now;i<=N;i++)fac[i]=fac[i-1]*i;ifac[N]=1/fac[N];for(size_t i=N;i-->now;)ifac[i]=ifac[i+1]*(i+1);}T factorial(size_t n){make_table(n);return fac[n];}T invfac(size_t n){make_table(n);return ifac[n];}T P(size_t n,size_t k){if(n<k)return 0;make_table(n);return fac[n]*ifac[n-k];}T C(size_t n,size_t k){if(n<k)return 0;make_table(n);return fac[n]*ifac[n-k]*ifac[k];}T H(size_t n,size_t k){if(n==0)return k==0?1:0;return C(n-1+k,k);}};using mint=atcoder::modint998244353;combination<mint>C;int main(){ios::sync_with_stdio(false);cin.tie(nullptr);int N,P;cin>>N>>P;mint ans=C.factorial(N);for(int c=0;c*P<=N;c++){ans-=C.P(N,c*P)/mint::raw(P).pow(c)*C.invfac(c);}cout<<ans.val()<<endl;}