#1199083 (C++23) No.3799 1 + 6 = 7 (Another version)

• 問題: No.3799 1 + 6 = 7 (Another version)
• ユーザー: kwm_t
• 提出日時: 2026-11-14 01:30:24
• 言語: C++23 (gcc 15.3.0 + boost 1.92.0 + ACL)
• 結果: AC
• 実行時間: 2 ms / 2,000 ms
• コード長: 3,258 bytes
• コンパイル時間: 5,024 ms
• コンパイル使用メモリ: 385,004 KB
• 実行使用メモリ: 6,400 KB
• 最終ジャッジ日時: 2026-11-14 01:31:02
• 合計ジャッジ時間: 7,236 ms
• ジャッジサーバーID （参考情報）: judge5_0 / judge2_0

ファイルパターン	結果
sample	AC * 3
other	AC * 36

ソースコード

#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; }
/**
 * @file simple_matrix.hpp
 * @brief 基本的な行列
 * @author kwm_t
 * @see https://github.com/kwm-t/cp-library/blob/main/math/matrix/simple_matrix.hpp
 */
#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);
	long long n, m; cin >> n >> m;
	vector<vector<mint>>mat = {
		{-1,1},
		{0,m}
	};
	mat = kwm_t::math::matrix::matrixPow(n, mat);
	auto ans = mat[0][0] + mat[0][1];
	cout << ans.val() << endl;
	return 0;
}