結果
問題 | No.1710 Minimum OR is X |
ユーザー |
![]() |
提出日時 | 2021-10-16 02:09:13 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 5 ms / 2,000 ms |
コード長 | 4,331 bytes |
コンパイル時間 | 2,338 ms |
コンパイル使用メモリ | 204,348 KB |
最終ジャッジ日時 | 2025-01-25 01:29:55 |
ジャッジサーバーID (参考情報) |
judge3 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 30 |
ソースコード
#include <bits/stdc++.h>#define REP_(i, a_, b_, a, b, ...) \for (int i = (a), END_##i = (b); i < END_##i; ++i)#define REP(i, ...) REP_(i, __VA_ARGS__, __VA_ARGS__, 0, __VA_ARGS__)#define ALL(x) std::begin(x), std::end(x)using i64 = long long;#include <atcoder/modint>using Mint = atcoder::modint998244353;std::ostream &operator<<(std::ostream &os, const Mint &m) {return os << m.val();}template<typename T, typename U>inline bool chmax(T &a, U b) {return a < b and ((a = std::move(b)), true);}template<typename T, typename U>inline bool chmin(T &a, U b) {return a > b and ((a = std::move(b)), true);}template<typename T>inline int ssize(const T &a) {return (int) a.size();}template<class T>inline std::ostream &print_one(const T &x, char endc) {if constexpr (std::is_same_v<T, bool>) {return std::cout << (x ? "Yes" : "No") << endc;} else {return std::cout << x << endc;}}template<class T>inline std::ostream &print(const T &x) { return print_one(x, '\n'); }template<typename T, typename... Ts>std::ostream &print(const T &head, Ts... tail) {return print_one(head, ' '), print(tail...);}inline std::ostream &print() { return std::cout << '\n'; }template<typename Container>std::ostream &print_seq(const Container &a, std::string_view sep = " ",std::string_view ends = "\n",std::ostream &os = std::cout) {auto b = std::begin(a), e = std::end(a);for (auto it = std::begin(a); it != e; ++it) {if (it != b) os << sep;os << *it;}return os << ends;}template<typename T, typename = void>struct is_iterable : std::false_type {};template<typename T>struct is_iterable<T, std::void_t<decltype(std::begin(std::declval<T>())),decltype(std::end(std::declval<T>()))>>: std::true_type {};template<typename T, typename = std::enable_if_t<is_iterable<T>::value && !std::is_same<T, std::string>::value>>std::ostream &operator<<(std::ostream &os, const T &a) {return print_seq(a, ", ", "", (os << "{")) << "}";}struct VersatileInput {template<typename T>operator T() const {T x;std::cin >> x;return x;}struct Sized {std::size_t n;template<typename T>operator T() const {T x(n);for (auto &e: x) std::cin >> e;return x;}};Sized operator()(std::size_t n) const { return {n}; }} const in;inline void check(bool cond, const char *message = "!ERROR!") {if (not cond) throw std::runtime_error(message);}#ifdef MY_DEBUG#include "debug_dump.hpp"#else#define DUMP(...)#define cerr if(false)std::cerr#endifusing namespace std;template<typename T>std::vector<T> pow_seq(int n, int base = 2) {std::vector<T> p(n + 1);p[0] = 1;for (int i = 0; i < n; ++i) {p[i + 1] = p[i] * base;}return p;}template<class T = Mint>std::vector<std::vector<T>> comb_table(int n) {std::vector<std::vector<T>> C(n + 1, std::vector<T>(n + 1, 0));for (int i = 0; i <= n; ++i) {C[i][0] = C[i][i] = 1;}for (int i = 1; i <= n; ++i) {for (int j = 1; j < i; ++j) {C[i][j] = C[i - 1][j - 1] + C[i - 1][j];}}return C;}auto solve() {const int n = in, m = in, K = in;string X = in;auto C = comb_table(n);auto dp = vector(2, vector(n + 1, Mint(0)));dp[0][n] = 1;auto pow2 = pow_seq<Mint>(n);auto ccum = vector(n + 1, Mint(0));for (int i = K; i <= n; ++i) {Mint s = 0;for (int j = 0; j < K; ++j) {s += C[i][j];}ccum[i] = s * pow2[n - i];}REP(i, m) {if (X[i] == '1') {REP(j, K, n + 1) {if (dp[0][j].val() == 0 or ccum[j].val() == 0) continue;dp[1][j] += dp[0][j] * ccum[j];}} else {REP(j, K, n + 1) {if (dp[0][j].val() == 0) continue;auto q = dp[0][j] * pow2[n - j];for (int j2 = j; j2 >= K; --j2) {dp[1][j2] += q * C[j][j2];}}}dp[0].swap(dp[1]);dp[1].assign(n + 1, 0);}Mint ans = 0;for (int i = K; i <= n; ++i) {ans += dp[0][i];}return ans;}int main() {ios_base::sync_with_stdio(false), cin.tie(nullptr);cout << std::fixed << std::setprecision(18);const int T = 1;//in;REP(t, T) {auto ans = solve();print(ans);}}