結果
問題 | No.1832 NAND Reversible |
ユーザー |
![]() |
提出日時 | 2022-02-04 22:20:37 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 43 ms / 2,000 ms |
コード長 | 1,299 bytes |
コンパイル時間 | 2,456 ms |
コンパイル使用メモリ | 197,368 KB |
最終ジャッジ日時 | 2025-01-27 19:16:35 |
ジャッジサーバーID (参考情報) |
judge5 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 23 |
ソースコード
#include <bits/stdc++.h> using namespace std; const long long MOD = 998244353; long long modpow(long long a, long long b){ long long ans = 1; while (b > 0){ if (b % 2 == 1){ ans *= a; ans %= MOD; } a *= a; a %= MOD; b /= 2; } return ans; } long long modinv(long long a){ return modpow(a, MOD - 2); } vector<long long> mf = {1}; vector<long long> mfi = {1}; long long modfact(int n){ if (mf.size() > n){ return mf[n]; } else { for (int i = mf.size(); i <= n; i++){ long long next = mf.back() * i % MOD; mf.push_back(next); mfi.push_back(modinv(next)); } return mf[n]; } } long long modfactinv(int n){ if (mfi.size() > n){ return mfi[n]; } else { return modinv(modfact(n)); } } long long modbinom(int n, int k){ if (n < 0 || k < 0 || k > n){ return 0; } else { return modfact(n) * modfactinv(k) % MOD * modfactinv(n - k) % MOD; } } int main(){ int N, K; cin >> N >> K; if (K == 0){ cout << 1 << endl; } else if (K == 1){ if (N % 2 == 0){ cout << 2 << endl; } else { cout << N - 2 << endl; } } else { long long ans = 0; ans += modbinom(N - 2, K - 2); for (int i = 2; i <= N - K; i += 2){ ans += modbinom(N - 2 - i, K - 2) * (i + 1) % MOD; } ans %= MOD; cout << ans << endl; } }