結果

問題 No.3213 depth max K
ユーザー asmin
提出日時 2025-07-25 22:17:26
言語 C++17(gcc12)
(gcc 12.3.0 + boost 1.87.0)
結果
MLE  
実行時間 -
コード長 4,257 bytes
コンパイル時間 6,354 ms
コンパイル使用メモリ 219,380 KB
実行使用メモリ 760,064 KB
最終ジャッジ日時 2025-07-25 22:17:54
合計ジャッジ時間 25,276 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 32 MLE * 9
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
using ll = long long;

template <uint32_t mod_, bool fast = false>
struct MontgomeryModInt {
private:
    using mint = MontgomeryModInt;
    using i32 = int32_t;
    using i64 = int64_t;
    using u32 = uint32_t;
    using u64 = uint64_t;

    static constexpr u32 get_r() {
        u32 ret = mod_;
        for (i32 i = 0; i < 4; i++) ret *= 2 - mod_ * ret;
        return ret;
    }

    static constexpr u32 r = get_r();

    static constexpr u32 n2 = -u64(mod_) % mod_;

    static_assert(r * mod_ == 1, "invalid, r * mod != 1");
    static_assert(mod_ < (1 << 30), "invalid, mod >= 2 ^ 30");
    static_assert((mod_ & 1) == 1, "invalid, mod % 2 == 0");

    u32 x;

    public:
    MontgomeryModInt() : x{} {}

    MontgomeryModInt(const i64 &a): x(reduce(u64(fast ? a : (a % mod() + mod())) * n2)) {}

    static constexpr u32 reduce(const u64 &b) {
        return u32(b >> 32) + mod() - u32((u64(u32(b) * r) * mod()) >> 32);
    }

    mint &operator+=(const mint &p) {
        if (i32(x += p.x - 2 * mod()) < 0) x += 2 * mod();
        return *this;
    }

    mint &operator-=(const mint &p) {
        if (i32(x -= p.x) < 0) x += 2 * mod();
        return *this;
    }

    mint &operator*=(const mint &p) {
        x = reduce(u64(x) * p.x);
        return *this;
    }

    mint &operator/=(const mint &p) {
        *this *= p.inv();
        return *this;
    }

    mint operator-() const { return mint() - *this; }

    mint operator+(const mint &p) const { return mint(*this) += p; }

    mint operator-(const mint &p) const { return mint(*this) -= p; }

    mint operator*(const mint &p) const { return mint(*this) *= p; }

    mint operator/(const mint &p) const { return mint(*this) /= p; }

    bool operator==(const mint &p) const {
        return (x >= mod() ? x - mod() : x) == (p.x >= mod() ? p.x - mod() : p.x);
    }

    bool operator!=(const mint &p) const {
        return (x >= mod() ? x - mod() : x) != (p.x >= mod() ? p.x - mod() : p.x);
    }

    u32 val() const {
        u32 ret = reduce(x);
        return ret >= mod() ? ret - mod() : ret;
    }

    mint pow(u64 n) const {
        mint ret(1), mul(*this);
        while (n > 0) {
            if (n & 1) ret *= mul;
            mul *= mul;
            n >>= 1;
        }
        return ret;
    }

    mint inv() const { return pow(mod() - 2); }

    friend ostream &operator<<(ostream &os, const mint &p) {
        return os << p.val();
    }

    friend istream &operator>>(istream &is, mint &a) {
        i64 t;
        is >> t;
        a = mint(t);
        return is;
    }

    static constexpr u32 mod() { return mod_; }
};

template <uint32_t mod>
using modint = MontgomeryModInt<mod>;
using mint = modint<998244353>;

void solve(){
    int N; cin >> N;
    int K; cin >> K;
    vector dp1(2 * N + 1, vector<mint>(N + 1, 0)), dp2(2 * N + 1, vector<mint>(N + 1, 0));
    dp1[0][0] = 1;
    mint ans = 0;
    for(int i = 0; i < 2 * N; ++i){
        for(int j = 0; j < K; ++j){
            if(j != 0) dp1[i + 1][j - 1] += dp1[i][j];
            dp1[i + 1][j + 1] += dp1[i][j];
        }
        dp2[i][K] += dp1[i][K];
        for(int j = 0; j <= N; ++j){
            if(j != 0){
                dp2[i + 1][j - 1] += dp2[i][j];
            }
            if(j != N){
                dp2[i + 1][j + 1] += dp2[i][j];
            }
        }
    }
    ans += dp2[2 * N][0];
    if(K == N){
        cout << ans << "\n";
        return;
    }

    vector dp3(2 * N + 1, vector<mint>(N + 1, 0)), dp4(2 * N + 1, vector<mint>(N + 1, 0));
    dp3[0][0] = 1;
    for(int i = 0; i < 2 * N; ++i){
        for(int j = 0; j <= K; ++j){
            if(j != 0) dp3[i + 1][j - 1] += dp3[i][j];
            dp3[i + 1][j + 1] += dp3[i][j];
        }
        dp4[i][K + 1] += dp3[i][K + 1];
        for(int j = 0; j <= N; ++j){
            if(j != 0){
                dp4[i + 1][j - 1] += dp4[i][j];
            }
            if(j != N){
                dp4[i + 1][j + 1] += dp4[i][j];
            }
        }
    }
    ans -= dp4[2 * N][0];
    cout << ans << "\n";
}


int main(){
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    cout << setprecision(10) << fixed;


    int T;
    T = 1;
    //cin >> T;
    for(;T--;) solve();
}
0