結果
問題 | No.2381 Gift Exchange Party |
ユーザー | umezo |
提出日時 | 2023-07-14 23:11:24 |
言語 | C++17 (gcc 12.3.0 + boost 1.83.0) |
結果 |
WA
|
実行時間 | - |
コード長 | 1,128 bytes |
コンパイル時間 | 1,980 ms |
コンパイル使用メモリ | 209,332 KB |
実行使用メモリ | 13,324 KB |
最終ジャッジ日時 | 2023-10-14 13:38:25 |
合計ジャッジ時間 | 6,524 ms |
ジャッジサーバーID (参考情報) |
judge13 / judge11 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 9 ms
13,152 KB |
testcase_01 | AC | 8 ms
12,924 KB |
testcase_02 | AC | 9 ms
13,116 KB |
testcase_03 | AC | 9 ms
13,020 KB |
testcase_04 | AC | 9 ms
12,996 KB |
testcase_05 | AC | 8 ms
12,972 KB |
testcase_06 | AC | 8 ms
12,988 KB |
testcase_07 | AC | 9 ms
13,064 KB |
testcase_08 | AC | 9 ms
13,040 KB |
testcase_09 | AC | 8 ms
12,920 KB |
testcase_10 | AC | 9 ms
12,908 KB |
testcase_11 | AC | 9 ms
12,956 KB |
testcase_12 | AC | 8 ms
12,928 KB |
testcase_13 | AC | 9 ms
12,996 KB |
testcase_14 | WA | - |
testcase_15 | WA | - |
testcase_16 | AC | 8 ms
12,956 KB |
testcase_17 | AC | 9 ms
12,984 KB |
testcase_18 | AC | 8 ms
13,112 KB |
testcase_19 | AC | 8 ms
13,188 KB |
testcase_20 | TLE | - |
testcase_21 | -- | - |
testcase_22 | -- | - |
testcase_23 | -- | - |
testcase_24 | -- | - |
ソースコード
#define rep(i,n) for(int i=0;i<(int)(n);i++) #define ALL(v) v.begin(),v.end() typedef long long ll; #include<bits/stdc++.h> using namespace std; const int MAX=410000; const int MOD=998244353; ll fac[MAX],finv[MAX],inv[MAX]; void init(){ fac[0]=fac[1]=1; finv[0]=finv[1]=1; inv[1]=1; for(int i=2;i<MAX;i++){ fac[i]=fac[i-1]*i%MOD; inv[i]=MOD-inv[MOD%i]*(MOD/i)%MOD; finv[i]=finv[i-1]*inv[i]%MOD; } } ll nCr(int n,int k){ if(n<k) return 0; if(n<0 || k<0) return 0; return fac[n]*(finv[k]*finv[n-k]%MOD)%MOD; } vector<ll> A; map<pair<ll,ll>,ll> m; ll f(ll x,ll c){ if(x<A[c]) return m[{x,c}]=1; if(x==0) return m[{x,c}]=1; if(c==0) return m[{x,c}]=1; ll sum=0,tmp=1; for(ll i=0;i*A[c]<=x;i++){ sum=(sum+f(x-i*A[c],c-1)*tmp%MOD)%MOD; tmp=tmp*nCr(x-i*A[c],A[c])%MOD*fac[A[c]-1]%MOD*inv[i+1]%MOD; } return m[{x,c}]=sum; } int main(){ ios::sync_with_stdio(false); std::cin.tie(nullptr); init(); ll n,p; cin>>n>>p; ll t=1,c=0; A.push_back(1); while(t*p<=n){ t*=p; A.push_back(t); c++; } cout<<(fac[n]-f(n,c)+MOD)%MOD<<endl; return 0; }