結果
問題 | No.2413 Multiple of 99 |
ユーザー |
👑 ![]() |
提出日時 | 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 |
ソースコード
#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;