結果
問題 | No.2383 Naphthol |
ユーザー |
![]() |
提出日時 | 2023-07-15 01:14:52 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 46 ms / 2,000 ms |
コード長 | 1,821 bytes |
コンパイル時間 | 4,670 ms |
コンパイル使用メモリ | 304,924 KB |
最終ジャッジ日時 | 2025-02-15 14:54:39 |
ジャッジサーバーID (参考情報) |
judge1 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 18 |
ソースコード
#include <bits/extc++.h>#include <atcoder/all>using namespace std;using ll = long long;#define REP(i,n) for(int i=0;i<int(n);i++)#define FOR(i,a,b) for(int i=a;i<=int(b);i++)#define ALL(x) x.begin(),x.end()#define INF INT_MAX#define INFL LLONG_MAX / 4using namespace atcoder;using mint = modint998244353;template<typename T> void chmin(T& a, T b) { a = min(a, b); }template<typename T> void chmax(T& a, T b) { a = max(a, b); }#define PR(x) cerr << #x << "=" << x << endlusing i128 = __int128_t;struct nCkmod {vector<long long> fanc;vector<long long> fancinv;long long mod;nCkmod(long long n, long long mod_ = 998244353): fanc(n + 1), fancinv(n + 1), mod(mod_) {fanc[0] = 1;fancinv[0] = 1;for(long long i = 1; i <= n; i++) {fanc[i] = (fanc[i - 1] * i) % mod;fancinv[i] = (fancinv[i - 1] * modinv(i)) % mod;}}long long modinv(long long a) {long long b = mod, u = 1, v = 0;while (b) {long long t = a / b;a -= t * b; swap(a, b);u -= t * v; swap(u, v);}u %= mod;if (u < 0) u += mod;return u;}long long nck(int n, int k) {if(n < k) return 0;return fanc[n] * fancinv[k] % mod * fancinv[n - k] % mod;}};int main() {ll n, k;cin >> n >> k;if(n == 1) {int ans[7] = {0, 1, 3, 3, 3, 1, 1};cout << ans[k];return 0;}nCkmod nck(2 * n + 10);mint ans = nck.nck(2 * n + 4, k);if(k % 2 == 0) {ans += nck.nck(n + 2, k / 2);if(n % 2 == 0) {ans += nck.nck(n + 2, k / 2);} else {ans += nck.nck(n+1, k / 2);ans += nck.nck(n+1, k / 2 - 1);}ans += nck.nck(n + 2, k / 2);} else {if(n % 2 == 0) {} else {ans += 2 * nck.nck(n+1, k / 2);}}ans /= 4;cout << ans.val();return 0;}