#1168472 (C++23) No.3566 Subsequence Sum

• 問題: No.3566 Subsequence Sum
• ユーザー: kwm_t
• 提出日時: 2026-06-06 10:42:56
• 言語: C++23 (gcc 15.2.0 + boost 1.89.0)
• 結果: TLE
• 実行時間: -
• コード長: 3,840 bytes
• コンパイル時間: 4,428 ms
• コンパイル使用メモリ: 381,124 KB
• 実行使用メモリ: 11,004 KB
• 最終ジャッジ日時: 2026-06-06 10:43:31
• 合計ジャッジ時間: 8,540 ms
• ジャッジサーバーID （参考情報）: judge3_0 / judge2_0

ファイルパターン	結果
sample	AC * 3
other	AC * 7 TLE * 1 -- * 7

ソースコード

#include <bits/stdc++.h>
#include <atcoder/all>
using namespace std;
using namespace atcoder;
// using mint = modint1000000007;
// const int mod = 1000000007;
using mint = modint998244353;
const int mod = 998244353;
// const int INF = 1e9;
// const long long LINF = 1e18;
#define rep(i, n) for (int i = 0; i < (n); ++i)
#define rep2(i, l, r) for (int i = (l); i < (r); ++i)
#define rrep(i, n) for (int i = (n)-1; i >= 0; --i)
#define rrep2(i, l, r) for (int i = (r)-1; i >= (l); --i)
#define all(x) (x).begin(), (x).end()
#define allR(x) (x).rbegin(), (x).rend()
#define P pair<int, int>
template<typename A, typename B> inline bool chmax(A &a, const B &b) { if (a < b) { a = b; return true; } return false; }
template<typename A, typename B> inline bool chmin(A &a, const B &b) { if (a > b) { a = b; return true; } return false; }
#ifndef KWM_T_MATH_MATRIX_SIMPLE_MATRIX_HPP
#define KWM_T_MATH_MATRIX_SIMPLE_MATRIX_HPP

// #include
#include <vector>
#include <cassert>

/**
 * @brief シンプルな行列演算（vector版・軽量）
 *
 * 最低限の行列演算（乗算・累乗）を提供する。
 * とりあえず使う用の軽量実装。
 *
 * 典型用途:
 *   - 行列累乗（DP遷移）
 *   - 小規模な行列計算
 *
 * 計算量:
 *   - 乗算: O(N^3)
 *   - 累乗: O(N^3 log K)
 *
 * @tparam T
 *   - 要素型（+, *, += が定義されていること）
 *
 * 制約 / 注意:
 *   - 正方行列のみ想定（matrixPow）
 *   - サイズ不一致は assert
 *   - 単位元は T(1) を使用
 *
 * 使用例:
 *   vector<vector<long long>> A(n, vector<long long>(n));
 *   auto B = matrixPow(10, A);
 *
 * verified:
 *   - https://atcoder.jp/contests/awc0053/submissions/75169363
 */
namespace kwm_t::math::matrix {

template<typename T>
std::vector<std::vector<T>> matrixMul(
	const std::vector<std::vector<T>>& A,
	const std::vector<std::vector<T>>& B
) {
	assert(!A.empty() && !B.empty());
	assert(A[0].size() == B.size());

	int n = (int)A.size();
	int m = (int)B[0].size();
	int p = (int)A[0].size();

	std::vector<std::vector<T>> C(n, std::vector<T>(m, T(0)));

	for (int i = 0; i < n; ++i) {
		for (int k = 0; k < p; ++k) {
			for (int j = 0; j < m; ++j) {
				C[i][j] += A[i][k] * B[k][j];
			}
		}
	}
	return C;
}

template<typename T>
std::vector<std::vector<T>> matrixPow(
	long long n,
	const std::vector<std::vector<T>>& mat
) {
	assert(!mat.empty());
	assert(mat.size() == mat[0].size());

	int size = (int)mat.size();

	std::vector<std::vector<T>> res(size, std::vector<T>(size, T(0)));
	for (int i = 0; i < size; ++i) {
		res[i][i] = T(1);
	}

	auto base = mat;

	while (n > 0) {
		if (n & 1) res = matrixMul(res, base);
		base = matrixMul(base, base);
		n >>= 1;
	}

	return res;
}

} // namespace kwm_t::math::matrix

#endif // KWM_T_MATH_MATRIX_SIMPLE_MATRIX_HPP
int main() {
	std::ios::sync_with_stdio(false);
	std::cin.tie(nullptr);
	string s; cin >> s;
	long long k; cin >> k;
	// 21*21のmatrixをもって,前半をやって
	// 後半pow?
	vector mat(21, vector<mint>(21));
	rep(i, 21)mat[i][i] = 1;
	rep(i, s.size()) {
		int x = s[i] - '0';
		vector nmat(21, vector<mint>(21));
		// 0-9 cnt
		rep(i, 10) {
			if (i == x) {
				rep(j, 10)nmat[j][x] += 1;
				if (x != 0) nmat[20][x] += 1;
			}
			else {
				nmat[i][i] = 1;
			}
		}
		// 10-19 sum
		rep(i, 10) {
			if (i == x) {
				rep(k, 10) nmat[k][10 + x] += x;
				rep(k, 10) nmat[10 + k][10 + x] += 10;
				if (x != 0) nmat[20][10 + x] += x;
			}
			else {
				nmat[10 + i][10 + i] = 1;
			}
		}
		// 20 1
		nmat[20][20] = 1;
		mat = kwm_t::math::matrix::matrixMul(mat, nmat);
	}
	mat = kwm_t::math::matrix::matrixPow(k, mat);
	vector<vector<mint>>dp(1, vector<mint>(21));
	dp[0][20] = 1;
	dp = kwm_t::math::matrix::matrixMul(dp, mat);
	mint ans = 0;
	rep(i, 10)ans += dp[0][i + 10];
	cout << ans.val() << endl;
	return 0;
}