結果

問題 No.2613 Sum of Combination
ユーザー 蜜蜂蜜蜂
提出日時 2021-03-05 18:00:44
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 244 ms / 4,500 ms
コード長 2,244 bytes
コンパイル時間 5,259 ms
コンパイル使用メモリ 240,408 KB
実行使用メモリ 30,028 KB
最終ジャッジ日時 2024-01-19 20:50:14
合計ジャッジ時間 12,547 ms
ジャッジサーバーID
（参考情報） 		judge12 / judge14
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 31 ms
15,524 KB
testcase_01 AC 30 ms
15,524 KB
testcase_02 AC 34 ms
16,036 KB
testcase_03 AC 31 ms
15,524 KB
testcase_04 AC 31 ms
15,524 KB
testcase_05 AC 30 ms
15,524 KB
testcase_06 AC 30 ms
15,524 KB
testcase_07 AC 31 ms
15,524 KB
testcase_08 AC 30 ms
15,652 KB
testcase_09 AC 30 ms
15,524 KB
testcase_10 AC 30 ms
15,524 KB
testcase_11 AC 30 ms
15,524 KB
testcase_12 AC 30 ms
15,524 KB
testcase_13 AC 37 ms
15,908 KB
testcase_14 AC 36 ms
15,908 KB
testcase_15 AC 34 ms
15,780 KB
testcase_16 AC 36 ms
15,908 KB
testcase_17 AC 36 ms
15,908 KB
testcase_18 AC 35 ms
15,908 KB
testcase_19 AC 36 ms
16,036 KB
testcase_20 AC 31 ms
15,524 KB
testcase_21 AC 31 ms
15,524 KB
testcase_22 AC 41 ms
16,128 KB
testcase_23 AC 226 ms
28,088 KB
testcase_24 AC 222 ms
28,056 KB
testcase_25 AC 211 ms
25,944 KB
testcase_26 AC 232 ms
29,848 KB
testcase_27 AC 130 ms
23,072 KB
testcase_28 AC 220 ms
29,652 KB
testcase_29 AC 220 ms
28,868 KB
testcase_30 AC 236 ms
29,864 KB
testcase_31 AC 225 ms
28,792 KB
testcase_32 AC 220 ms
28,428 KB
testcase_33 AC 223 ms
30,028 KB
testcase_34 AC 220 ms
30,028 KB
testcase_35 AC 230 ms
30,028 KB
testcase_36 AC 232 ms
30,028 KB
testcase_37 AC 230 ms
30,028 KB
testcase_38 AC 221 ms
29,576 KB
testcase_39 AC 227 ms
29,640 KB
testcase_40 AC 218 ms
29,628 KB
testcase_41 AC 220 ms
29,688 KB
testcase_42 AC 244 ms
29,676 KB
testcase_43 AC 219 ms
30,028 KB
testcase_44 AC 209 ms
30,028 KB
testcase_45 AC 31 ms
15,524 KB
testcase_46 AC 31 ms
15,524 KB
testcase_47 AC 30 ms
15,524 KB
testcase_48 AC 31 ms
15,524 KB
testcase_49 AC 31 ms
15,524 KB
testcase_50 AC 210 ms
30,028 KB
testcase_51 AC 209 ms
30,028 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;

#include <atcoder/all>
using namespace atcoder;

using ll = long long;
using ld = long double;
#define fi first
#define se second
#define pb push_back

const int MAX=510000;
ll MOD;

ll fac[MAX],finv[MAX],inv[MAX];

void COMinit(){
  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 COM(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;
}

constexpr ll mod = 998244353;

random_device seed;
mt19937_64 randint(seed());

ll ggr(ll mi,ll ma) { // [mi, ma)
    return mi + randint() % (ma - mi);
}

int main(){
  ll n;
  int p;
  cin>>n>>p;

  int copyp=p;
  MOD=p;
  COMinit();

  vector<int> v;
  while(n>0){
    v.pb(n%p);
    n/=p;
  }

  if(p==2){
    ll ans=1;
    for(int i:v){
      if(i==1){
        ans*=2;
        ans%=mod;
      }
    }
    cout<<ans<<endl;
    return 0;
  }
  
  vector<int> divisor;
  p--;
  for(int i=2;i*i<=p;i++){
    if(p%i==0){
      divisor.pb(i);
      while(p%i==0){
        p/=i;
      }
    }
  }
  if(p>1){
    divisor.pb(p);
  }

  p=copyp;
  int root;

  while(true){
    int check=ggr(2,p);
    int flag=1;
    for(int i:divisor){
      if(pow_mod(check,(p-1)/i,p)==1){
        flag=0;
        break;
      }
    }
    if(flag==1){
      root=check;
      break;
    }
    if(p==2){
      root=1;
      break;
    }
  }
  
  vector<ll> powmod(p),invpowmod(p);
  //powmod[i] := root^i (mod p)
  //invpowmod[i] := root^(invpowmod[i]) = i (mod p)

  powmod[0]=1;
  for(int i=1;i<p;i++){
    powmod[i]=powmod[i-1]*root;
    powmod[i]%=p;
    invpowmod[powmod[i]]=i;
  }
  invpowmod[0]=-1;
  invpowmod[1]=0;
  
  vector<ll> a(p,0);
  a[0]=1;

  for(int now:v){
    vector<ll> b(p,0);
    for(int i=0;i<=now;i++){
      int next=COM(now,i);
      if(next==0){
        continue;
      }
      b[invpowmod[next]]++;
    }
    
    vector<ll> c=convolution(a,b);
    for(int i=0;i<p;i++){
      a[i]=0;
    }
    for(int i=0;i<2*p-1;i++){
      a[i%(p-1)]+=c[i];
    }
  }

  ll ans=0;
  for(int i=0;i<p;i++){
    ans+=powmod[i]*a[i];
    ans%=mod;
  }
  cout<<ans<<endl;
}
0