結果

問題 No.2381 Gift Exchange Party
ユーザー umezoumezo
提出日時 2023-07-14 22:49:40
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
WA  
実行時間 -
コード長 1,133 bytes
コンパイル時間 2,225 ms
コンパイル使用メモリ 209,356 KB
実行使用メモリ 13,240 KB
最終ジャッジ日時 2023-10-14 13:06:31
合計ジャッジ時間 3,531 ms
ジャッジサーバーID
（参考情報） 		judge12 / judge13
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 10 ms
13,192 KB
testcase_01 AC 9 ms
12,952 KB
testcase_02 AC 9 ms
13,000 KB
testcase_03 WA -
testcase_04 WA -
testcase_05 WA -
testcase_06 AC 8 ms
13,060 KB
testcase_07 WA -
testcase_08 WA -
testcase_09 WA -
testcase_10 WA -
testcase_11 AC 8 ms
12,984 KB
testcase_12 AC 8 ms
12,948 KB
testcase_13 AC 9 ms
13,008 KB
testcase_14 WA -
testcase_15 WA -
testcase_16 AC 8 ms
13,000 KB
testcase_17 AC 11 ms
12,968 KB
testcase_18 AC 8 ms
13,044 KB
testcase_19 AC 9 ms
13,176 KB
testcase_20 WA -
testcase_21 WA -
testcase_22 WA -
testcase_23 WA -
testcase_24 AC 10 ms
13,196 KB
権限があれば一括ダウンロードができます

ソースコード

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(c==0) return m[{x,c}]=1;
  ll sum=0,tmp=1;
  for(ll i=0;i*A[c]<=x;i++){
    if(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==1){
    cout<<0<<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-1+MOD)%MOD<<endl;
  
  return 0;
}
0