結果

問題 No.2503 Typical Path Counting Problem on a Grid
ユーザー suisensuisen
提出日時 2023-07-21 20:36:17
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
RE  
実行時間 -
コード長 2,013 bytes
コンパイル時間 677 ms
コンパイル使用メモリ 77,868 KB
実行使用メモリ 42,496 KB
最終ジャッジ日時 2024-10-01 19:45:35
合計ジャッジ時間 4,684 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 RE -
testcase_01 RE -
testcase_02 AC 119 ms
42,496 KB
testcase_03 RE -
testcase_04 AC 158 ms
42,368 KB
testcase_05 AC 133 ms
42,496 KB
testcase_06 AC 166 ms
42,496 KB
testcase_07 AC 165 ms
42,368 KB
testcase_08 AC 142 ms
42,496 KB
testcase_09 AC 158 ms
42,408 KB
権限があれば一括ダウンロードができます

ソースコード

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';
    }
}
0