結果
問題 |
No.2383 Naphthol
|
ユーザー |
![]() |
提出日時 | 2025-02-26 19:50:28 |
言語 | C++23 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 5 ms / 2,000 ms |
コード長 | 1,520 bytes |
コンパイル時間 | 3,397 ms |
コンパイル使用メモリ | 279,560 KB |
実行使用メモリ | 6,820 KB |
最終ジャッジ日時 | 2025-02-26 19:50:33 |
合計ジャッジ時間 | 4,528 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 18 |
ソースコード
#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); } } 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(); }