結果

問題 No.2413 Multiple of 99
ユーザー 👑 Nachia
提出日時 2023-08-11 23:32:29
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 809 ms / 8,000 ms
コード長 1,751 bytes
コンパイル時間 2,019 ms
コンパイル使用メモリ 114,936 KB
最終ジャッジ日時 2025-02-16 02:11:44
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 21
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <string>
#include <vector>
#include <algorithm>
#include <atcoder/modint>
#include <atcoder/convolution>
using namespace std;
using i32 = int;
using u32 = unsigned int;
using i64 = long long;
using u64 = unsigned long long;
#define rep(i,n) for(int i=0; i<(int)(n); i++)
const i64 INF = 1001001001001001001;

using Modint = atcoder::static_modint<998244353>;

void mul10(vector<Modint>& A){
    A.resize(A.size() + 9);
    rep(i,A.size()-1) A[i+1] += A[i];
    for(int i=A.size()-1; i>=10; i--) A[i] -= A[i-10];
}

vector<Modint> powdig(int N){
    if(N == 0){
        return vector<Modint>(1,1);
    }
    auto buf = powdig(N/2);
    auto A = atcoder::convolution(buf, buf);
    if(N%2 == 1) mul10(A);
    return A;
}

int main(){
    int N, K; cin >> N >> K;
    auto d1 = powdig(N/2);
    auto d0 = d1;
    if(N%2 == 1) mul10(d0);

    int Z = N + 1;
    vector<Modint> raizeK(Z+1); {
        vector<int> lpf(Z+1);
        raizeK[1] = 1;
        for(int p=2; p<=Z; p++){
            if(lpf[p]){ raizeK[p] = raizeK[lpf[p]] * raizeK[p/lpf[p]]; continue; }
            raizeK[p] = Modint(p).pow(K);
            for(int q=p*2; q<=Z; q+=p) lpf[q] = p;
        }
    }
    
    Modint ans = 0;
    rep(c0,99){
        vector<Modint> A(N/2+2);
        for(int c=c0; c<(int)d0.size(); c+=99) A[c/9] = d0[c];
        vector<Modint> B(N/2+2);
        rep(i,d1.size()) if((i*10+c0)%99 == 0) B[(i+8)/9] = d1[i];
        A = atcoder::convolution(A, B);
        rep(i,A.size()) ans += A[i] * raizeK[i];
    }
    ans *= Modint(9).pow(K);
    cout << ans.val() << endl;
    return 0;
}


struct ios_do_not_sync{
    ios_do_not_sync(){
        ios::sync_with_stdio(false);
        cin.tie(nullptr);
    }
} ios_do_not_sync_instance;
0