結果

問題 No.2503 Typical Path Counting Problem on a Grid
ユーザー suisen
提出日時 2023-07-21 20:36:17
言語 C++17(gcc12)
(gcc 12.3.0 + boost 1.87.0)
結果
RE  
実行時間 -
コード長 2,013 bytes
コンパイル時間 2,253 ms
コンパイル使用メモリ 100,612 KB
最終ジャッジ日時 2025-02-15 16:13:00
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 7 RE * 3
権限があれば一括ダウンロードができます

ソースコード

diff #
プレゼンテーションモードにする

// min(n, m) = 0 の場合分け漏れ
#include <array>
#include <iostream>
#include <vector>
#include <atcoder/modint>
using mint = atcoder::modint998244353;
template <typename T, size_t N, size_t M>
using Matrix = std::array<std::array<T, N>, M>;
template <typename T, size_t N, size_t M, size_t K>
Matrix<T, N, K> operator*(const Matrix<T, N, M>& A, const Matrix<T, M, K>& B) {
    Matrix<T, N, K> C{};
    for (size_t i = 0; i < N; ++i) for (size_t j = 0; j < M; ++j) for (size_t k = 0; k < K; ++k) {
        C[i][k] += A[i][j] * B[j][k];
    }
    return C;
}
template <typename T, size_t N, size_t M>
std::array<T, N> operator*(const Matrix<T, N, M>& A, const std::array<T, M>& x) {
    std::array<T, N> y{};
    for (size_t i = 0; i < N; ++i) for (size_t j = 0; j < M; ++j) {
        y[i] += A[i][j] * x[j];
    }
    return y;
}
template <typename T, size_t N>
Matrix<T, N, N> pow(Matrix<T, N, N> A, unsigned long long k) {
    Matrix<T, N, N> B{};
    for (size_t i = 0; i < N; ++i) B[i][i] = 1;
    for (; k; k >>= 1) {
        if (k & 1) B = B * A;
        A = A * A;
    }
    return B;
}
constexpr size_t MAX_N = 10000000;
std::array<mint, MAX_N + 1> g;
void init_g() {
    g.fill(0);
    g[0] = 1;
    for (size_t i = 1; i <= MAX_N; ++i) {
        g[i] += g[i - 1] * (2 * i);
        if (i >= 2) {
            g[i] += g[i - 2] * (i - 1);
        }
    }
}
mint solve(uint32_t n, uint64_t m) {
    if (n > m) {
        uint64_t tmp = n;
        n = m, m = tmp;
    }
    Matrix<mint, 2, 2> A{
        0, 1,
        n, 2 * n + 1
    };
    mint fn1 = g[n - 1], fn = g[n];
    auto [fm1, fm] = pow(A, m - n) * std::array<mint, 2> { fn1, fn };
    return fm1 * fn1 * n + fm * fn;
}
int main() {
    std::ios::sync_with_stdio(false);
    std::cin.tie(nullptr);
    init_g();
    uint32_t t;
    std::cin >> t;
    for (uint32_t case_id = 0; case_id < t; ++case_id) {
        uint32_t n;
        uint64_t m;
        std::cin >> n >> m;
        std::cout << solve(n, m).val() << '\n';
    }
}
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0