結果

問題 No.3089 Base M Numbers, But Only 0~9
ユーザー umezo
提出日時 2025-04-04 23:08:24
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,170 bytes
コンパイル時間 3,471 ms
コンパイル使用メモリ 278,684 KB
実行使用メモリ 9,984 KB
最終ジャッジ日時 2025-04-04 23:08:29
合計ジャッジ時間 4,663 ms
ジャッジサーバーID
（参考情報） 		judge6 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 5
other AC * 14 WA * 5
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

#define rep(i,n) for(int i=0;i<(int)(n);i++)
#define ALL(v) v.begin(),v.end()
typedef long long ll;
template <class T> using V=vector<T>;
template <class T> using VV=V<V<T>>;

const int MOD=998244353;
 
int main(){
  ios::sync_with_stdio(false);
  std::cin.tie(nullptr);
  
  ll m;
  cin>>m;
  string s;
  cin>>s;
  int n=s.size();
  ll t=1;
  reverse(ALL(s));
  
  if(m==10){
    ll ans=0;
    ll t=1;
    rep(i,n){
      ans=(ans+(s[i]-'0')*t%MOD)%MOD;
      t=t*10;
    }
    cout<<ans<<endl;
    return 0;
  }
  
  V<ll> S(n),C(n);
  rep(i,n){
    ll a=s[i]-'0';
    if(a<=(m-1)%10){
      C[i]=(m-1)/10+1;
      if(a==0) C[i]--;
    }
    else{
      C[i]=(m-1)/10;
      if(a==0) C[i]--;
    }
    if(a!=0) S[i]=C[i]*(a+5*(C[i]-1)%MOD)%MOD;
    else S[i]=C[i]*(10+5*(C[i]-1)%MOD)%MOD;
  }
  V<ll> L(n),R(n);
  L[0]=C[0],R[n-1]=C[n-1];
  for(int i=1;i<n;i++) L[i]=L[i-1]*C[i]%MOD;
  for(int i=n-2;i>=0;i--) R[i]=R[i+1]*C[i]%MOD;
  ll ans=0;
  rep(i,n){
    ll tmp=S[i]*t%MOD;
    if(i) tmp=tmp*L[i-1]%MOD;
    if(i<n-1) tmp=tmp*R[i+1]%MOD;
    ans=(ans+tmp)%MOD;
    t=t*m%MOD;
  }
  cout<<ans<<endl;
    
  return 0;
}
0