結果

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

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 RE -
testcase_01 AC 43 ms
6,820 KB
testcase_02 TLE -
testcase_03 -- -
testcase_04 -- -
testcase_05 -- -
testcase_06 -- -
testcase_07 -- -
testcase_08 -- -
testcase_09 -- -
権限があれば一括ダウンロードができます

ソースコード

diff #

// TLE: T(n+m)

#include <array>
#include <iostream>
#include <vector>

#include <atcoder/modint>

using mint = atcoder::modint998244353;

mint solve(uint32_t n, uint64_t m) {
    if (n > m) {
        uint64_t tmp = n;
        n = m, m = tmp;
    }
    std::vector<mint> f(n + m + 1);
    f[0] = 1;
    for (uint64_t i = 1; i <= n + m; ++i) {
        if (i <= n) {
            f[i] += f[i - 1] * 2 * i;
            if (i >= 2) f[i] += f[i - 2] * (i - 1);
        } else if (i <= m) {
            f[i] += f[i - 1] * (2 * n + 1);
            if (i >= 2) f[i] += f[i - 2] * n;
        } else {
            f[i] += f[i - 1] * 2 * (n + m + 1 - i);
            if (i >= 2) f[i] += f[i - 2] * (n + m + 1 - i);
        }
    }
    return f[n + m];
}

int main() {
    std::ios::sync_with_stdio(false);
    std::cin.tie(nullptr);

    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