結果

問題 No.2383 Naphthol
ユーザー Iroha_3856
提出日時 2025-02-26 19:47:24
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,524 bytes
コンパイル時間 3,311 ms
コンパイル使用メモリ 280,352 KB
実行使用メモリ 6,820 KB
最終ジャッジ日時 2025-02-26 19:47:29
合計ジャッジ時間 4,376 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 16 WA * 2
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
#include <atcoder/modint>
using mint = atcoder::modint998244353;

#define rep(i, l, r) for (int i = (int)(l); i<(int)(r); i++)
#define ll long long

struct Combination {
    vector<mint> fac, infac;
    Combination(int n) {
        fac.resize(n+1);
        infac.resize(n+1);
        fac[0] = 1;
        for (int i = 1; i <= n; i++) fac[i] = fac[i-1] * i;
        infac[n] = fac[n].inv();
        for (int i = n; i >= 1; i--) infac[i-1] = infac[i] * i;
    }
    mint operator()(int n, int k) {
        if (k < 0 || k > n) return 0;
        return fac[n] * infac[k] * infac[n-k];
    }
};

void solve() {
    int N, K; cin >> N >> K;
    if (N == 1) {
        cout << "1133311"[K] << endl;
        return;
    }
    Combination comb(2*N+4);
    mint ans = 0;
    ans += comb(2*N+4, K);
    //-反転
    if (K%2 == 0) {
        ans += comb(N+2, K/2);
    }
    //|反転1
    if (N%2 == 0 and K%2 == 0) {
        ans += comb(N+2, K/2);
    }
    if (N%2 == 1) {
        if (K%2 == 0) {
            //真ん中は0, 2
            if (K >= 0) {
                ans += comb(N+1, K/2);
            }
            if (K >= 2) {
                ans += comb(N+1, (K-2)/2) * 2;
            }
        }
        if (K%2 == 1) {
            //真ん中は1
            ans += comb(N+1, (K-1)/2) * 2;
        }
    }
    if (K%2 == 0) {
        ans += comb(N+2, K/2);
    }
    ans /= 4;
    cout << ans.val() << endl;
}

int main() {
    int T = 1;
    //cin >> T;
    while(T--) solve();
}
0