結果

問題 No.2381 Gift Exchange Party
ユーザー umezoumezo
提出日時 2023-07-14 23:09:10
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
WA  
実行時間 -
コード長 1,096 bytes
コンパイル時間 1,877 ms
コンパイル使用メモリ 209,664 KB
実行使用メモリ 13,272 KB
最終ジャッジ日時 2023-10-14 13:34:40
合計ジャッジ時間 6,222 ms
ジャッジサーバーID
（参考情報） 		judge13 / judge11
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 9 ms
13,176 KB
testcase_01 AC 9 ms
13,048 KB
testcase_02 AC 10 ms
12,932 KB
testcase_03 AC 10 ms
13,180 KB
testcase_04 AC 9 ms
13,056 KB
testcase_05 AC 9 ms
12,924 KB
testcase_06 AC 9 ms
12,980 KB
testcase_07 AC 9 ms
12,964 KB
testcase_08 AC 9 ms
13,072 KB
testcase_09 AC 9 ms
13,016 KB
testcase_10 AC 9 ms
12,980 KB
testcase_11 AC 9 ms
13,212 KB
testcase_12 AC 9 ms
12,956 KB
testcase_13 AC 9 ms
12,940 KB
testcase_14 WA -
testcase_15 WA -
testcase_16 AC 9 ms
13,152 KB
testcase_17 AC 9 ms
13,188 KB
testcase_18 AC 10 ms
13,048 KB
testcase_19 AC 10 ms
13,152 KB
testcase_20 TLE -
testcase_21 -- -
testcase_22 -- -
testcase_23 -- -
testcase_24 -- -
権限があれば一括ダウンロードができます

ソースコード

diff # 

#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==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;
}
0