結果
問題 | No.2381 Gift Exchange Party |
ユーザー |
|
提出日時 | 2023-07-15 00:03:19 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 43 ms / 2,000 ms |
コード長 | 1,628 bytes |
コンパイル時間 | 1,134 ms |
コンパイル使用メモリ | 100,400 KB |
最終ジャッジ日時 | 2025-02-15 14:50:19 |
ジャッジサーバーID (参考情報) |
judge4 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 22 |
ソースコード
#include<iostream> #include<set> #include<algorithm> #include<vector> #include<string> #include<set> #include<map> #include<numeric> #include<queue> #include<cmath> using namespace std; typedef long long ll; const ll INF=1LL<<60; typedef pair<int,int> P; typedef pair<int,P> PP; const ll MOD=998244353; const double PI=acos(-1); //xのy乗 ll mod_pow(ll x,ll y,ll mod){ ll power=x; ll ret = 1; while(y>0){ if(y&1){ ret = ret*power%mod; } power = power*power%mod; y = y>>1; } return ret; } struct combination{ typedef long long ll; std::vector<ll> fact,ifact; combination(ll n): fact(n+1),ifact(n+1){ fact[0]=1;//0!=1通り for(ll i=1;i<=n;i++) fact[i] = fact[i-1]*i%MOD; //ifact[n] = fact[n].inv(); // n!の逆元がifact[n] ifact[n] = mod_pow(fact[n],MOD-2,MOD);// n!を(mod-2)乗した後でmodであまりをとった for(ll i=n;i>=1;i--) ifact[i-1] = ifact[i]*i%MOD; } ll operator()(ll n,ll k){ if(k<0 || k>n) return 0; // n!/( k!*(n-k)! ) return (fact[n]*ifact[k])%MOD * ifact[n-k]%MOD; } }; combination comb(10*100000); int main(){ ll N,P; cin>>N>>P; ll res=0; for(int k=0;k*P<=N;k++){ ll tmp=comb.fact[N]; ll d=comb.fact[k]; d*=comb.fact[N-k*P]; d%=MOD; d*=mod_pow(P,k,MOD); d%=MOD; tmp*=mod_pow(d,MOD-2,MOD); tmp%=MOD; res+=tmp; res%=MOD; } ll ans=comb.fact[N]; ans-=res; ans+=MOD; ans%=MOD; cout<<ans<<endl; }