結果

問題 No.2381 Gift Exchange Party
ユーザー umezo
提出日時 2023-07-14 23:05:13
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,167 bytes
コンパイル時間 1,874 ms
コンパイル使用メモリ 205,760 KB
最終ジャッジ日時 2025-02-15 14:37:27
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 15 WA * 2 TLE * 1 -- * 4
権限があれば一括ダウンロードができます

ソースコード

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;
  if(n<p){
    cout<<(fac[n]-1+MOD)%MOD<<endl;
    return 0;
  }
  
  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)%MOD<<endl;
  
  return 0;
}
0