Problem p02635 (C++) — buggy submission, verdict: Memory Limit Exceeded

#define _CRT_SECURE_NO_WARNINGS
#pragma GCC target("avx")
#pragma GCC optimize("O3")
#include <algorithm>
#include <bitset>
#include <cassert>
#include <cfloat>
#include <climits>
#include <cmath>
#include <complex>
#include <ctime>
#include <deque>
#include <fstream>
#include <functional>
#include <iomanip>
#include <iostream>
#include <iterator>
#include <list>
#include <map>
#include <memory>
#include <queue>
#include <random>
#include <set>
#include <stack>
#include <string.h>
#include <string>
#include <unordered_map>
#include <unordered_set>
#include <utility>
#include <vector>
#define rep(i, n) for (int i = 0; i < (lint)(n); i++)
#define REP(i, n) for (int i = 1; i <= (lint)(n); i++)
#define all(V) V.begin(), V.end()
#define stackReplaceBegin(size)                                                \\
  void *p = malloc(size);                                                      \\
  esp_new = (lint)p + size - 1;                                                \\
  void *stack_extend_memory_ = malloc(size);                                   \\
  void *stack_extend_origin_memory_;                                           \\
  char *stack_extend_dummy_memory_ = (char *)alloca(                           \\
      (1 + (int)(((long long)stack_extend_memory_) & 127)) * 16);              \\
  *stack_extend_dummy_memory_ = 0;                                             \\
  asm volatile("mov %%rsp, %%rbx\\nmov %%rax, %%rsp"                            \\
               : "=b"(stack_extend_origin_memory_)                             \\
               : "a"((char *)stack_extend_memory_ + (size)-1024));
typedef long long lint;
typedef unsigned long long ulint;
typedef std::pair<int, int> P;
constexpr int INF = INT_MAX / 2;
constexpr lint LINF = LLONG_MAX / 2;
constexpr double eps = DBL_EPSILON;
constexpr double PI = 3.141592653589793238462643383279;
lint esp_new;
template <class T>
class prique : public std::priority_queue<T, std::vector<T>, std::greater<T>> {
};
template <class T, class U> inline bool chmax(T &lhs, const U &rhs) {
  if (lhs < rhs) {
    lhs = rhs;
    return 1;
  }
  return 0;
}
template <class T, class U> inline bool chmin(T &lhs, const U &rhs) {
  if (lhs > rhs) {
    lhs = rhs;
    return 1;
  }
  return 0;
}
inline lint gcd(lint a, lint b) {
  while (b) {
    lint c = a;
    a = b;
    b = c % b;
  }
  return a;
}
inline lint lcm(lint a, lint b) { return a / gcd(a, b) * b; }
bool isprime(lint n) {
  if (n == 1)
    return false;
  for (int i = 2; i * i <= n; i++) {
    if (n % i == 0)
      return false;
  }
  return true;
}
template <typename T> T mypow(T a, unsigned int b) {
  T res(1), now(a);
  while (b) {
    if (b & 1)
      res *= now;
    b >>= 1;
    now *= now;
  }
  return res;
}
template <typename T> void printArray(std::vector<T> &vec) {
  rep(i, (int)vec.size() - 1) std::cout << vec[i] << " ";
  if (!vec.empty())
    std::cout << vec.back() << std::endl;
}
template <typename T> void printArray(T l, T r) {
  T rprev = r;
  rprev--;
  for (T i = l; i != rprev; i++) {
    std::cout << *i << " ";
  }
  std::cout << *rprev << std::endl;
}
class ModInt {
  lint value;

public:
  static const unsigned int modulo;
  ModInt() : value(0) {}
  template <typename T> ModInt(T value = 0) : value(value) {
    if (value < 0)
      value = -(lint)(-value % modulo) + modulo;
    this->value = value % modulo;
  }
  inline operator int() const { return value; }
  inline ModInt &operator+=(const ModInt &x) {
    value += x.value;
    if (value >= modulo)
      value -= modulo;
    return *this;
  }
  inline ModInt &operator++() {
    if (value == modulo - 1)
      value = 0;
    else
      value++;
    return *this;
  }
  inline ModInt operator-() const { return ModInt(0) -= *this; }
  inline ModInt &operator-=(const ModInt &x) {
    value -= x.value;
    if (value < 0)
      value += modulo;
    return *this;
  }
  inline ModInt &operator--() {
    if (value == 0)
      value = modulo - 1;
    else
      value--;
    return *this;
  }
  inline ModInt &operator*=(const ModInt &x) {
    value = value * x.value % modulo;
    return *this;
  }
  inline ModInt &operator/=(ModInt rhs) {
    int exp = modulo - 2;
    while (exp) {
      if (exp & 1)
        *this *= rhs;
      rhs *= rhs;
      exp >>= 1;
    }
    return *this;
  }
  template <typename T> ModInt operator+(const T &rhs) const {
    return ModInt(*this) += rhs;
  }
  template <typename T> ModInt &operator+=(const T &rhs) {
    return operator+=(ModInt(rhs));
  }
  template <typename T> ModInt operator-(const T &rhs) const {
    return ModInt(*this) -= rhs;
  }
  template <typename T> ModInt &operator-=(const T &rhs) {
    return operator-=(ModInt(rhs));
  }
  template <typename T> ModInt operator*(const T &rhs) const {
    return ModInt(*this) *= rhs;
  }
  template <typename T> ModInt &operator*=(const T &rhs) {
    return operator*=(ModInt(rhs));
  }
  template <typename T> ModInt operator/(const T &rhs) const {
    return ModInt(*this) /= rhs;
  }
  template <typename T> ModInt &operator/=(const T &rhs) {
    return operator/=(ModInt(rhs));
  }
};
std::istream &operator>>(std::istream &ist, ModInt &x) {
  lint a;
  ist >> a;
  x = a;
  return ist;
}
const unsigned int ModInt::modulo = 998244353;
int k;
std::string s;
std::vector<int> vec[2];
ModInt dp[310][310][610];
int main() {
  std::cin >> s >> k;
  int cnt = 0;
  rep(i, s.size()) {
    if (s[i] == '0') {
      vec[0].emplace_back(cnt);
      vec[1].emplace_back(cnt);
      cnt = 0;
    } else
      cnt++;
  }
  vec[0].emplace_back(cnt);
  vec[1].emplace_back(cnt);
  rep(i, vec[1].size() - 1) vec[1][i + 1] += vec[1][i];
  dp[0][0][0] = 1;
  chmin(k, vec[1].back());
  rep(i, vec[0].size()) {
    for (int j = (i == 0 ? 0 : vec[1][i - 1]); j <= vec[1].back(); j++) {
      rep(l, 2 * k + 1) {
        for (int m = std::max(0, vec[1][i] - j); j + m <= vec[1].back(); m++) {
          if (2 * k < l + std::abs(vec[0][i] - m))
            continue;
          dp[i + 1][j + m][l + std::abs(vec[0][i] - m)] += dp[i][j][l];
        }
      }
    }
  }
  ModInt ans = 0;
  for (int i = 0; i <= 2 * k; i += 2) {
    ans += dp[vec[0].size()][vec[1].back()][i];
  }
  std::cout << ans << std::endl;
  return 0;
}

Error: MLE

Fixed submission:

#define _CRT_SECURE_NO_WARNINGS
#pragma GCC target("avx")
#pragma GCC optimize("O3")
#pragma GCC optimize("unroll-loops")
#include <algorithm>
#include <bitset>
#include <cassert>
#include <cfloat>
#include <climits>
#include <cmath>
#include <complex>
#include <ctime>
#include <deque>
#include <fstream>
#include <functional>
#include <iomanip>
#include <iostream>
#include <iterator>
#include <list>
#include <map>
#include <memory>
#include <queue>
#include <random>
#include <set>
#include <stack>
#include <string>
#include <unordered_map>
#include <unordered_set>
#include <utility>
#include <vector>
#define int long long
#define rep(i, n) for (int i = 0; i < (lint)(n); i++)
#define REP(i, n) for (int i = 1; i <= (lint)(n); i++)
#define all(V) V.begin(), V.end()
#define stackReplaceBegin(size)                                                \\
  void *p = malloc(size);                                                      \\
  esp_new = (lint)p + size - 1;                                                \\
  void *stack_extend_memory_ = malloc(size);                                   \\
  void *stack_extend_origin_memory_;                                           \\
  char *stack_extend_dummy_memory_ = (char *)alloca(                           \\
      (1 + (int)(((long long)stack_extend_memory_) & 127)) * 16);              \\
  *stack_extend_dummy_memory_ = 0;                                             \\
  asm volatile("mov %%rsp, %%rbx\\nmov %%rax, %%rsp"                            \\
               : "=b"(stack_extend_origin_memory_)                             \\
               : "a"((char *)stack_extend_memory_ + (size)-1024));
typedef long long lint;
typedef unsigned long long ulint;
typedef std::pair<lint, lint> P;
constexpr int INF = INT_MAX / 2;
constexpr lint LINF = LLONG_MAX / 2;
constexpr double eps = DBL_EPSILON;
constexpr double PI = 3.141592653589793238462643383279;
lint esp_new;
template <class T>
class prique : public std::priority_queue<T, std::vector<T>, std::greater<T>> {
};
template <class T, class U> inline bool chmax(T &lhs, const U &rhs) {
  if (lhs < rhs) {
    lhs = rhs;
    return 1;
  }
  return 0;
}
template <class T, class U> inline bool chmin(T &lhs, const U &rhs) {
  if (lhs > rhs) {
    lhs = rhs;
    return 1;
  }
  return 0;
}
inline lint gcd(lint a, lint b) {
  while (b) {
    lint c = a;
    a = b;
    b = c % b;
  }
  return a;
}
inline lint lcm(lint a, lint b) { return a / gcd(a, b) * b; }
bool isprime(lint n) {
  if (n == 1)
    return false;
  for (int i = 2; i * i <= n; i++) {
    if (n % i == 0)
      return false;
  }
  return true;
}
template <typename T> T mypow(T a, unsigned int b) {
  if (!b)
    return T(1);
  if (b & 1)
    return mypow(a, b - 1) * a;
  T memo = mypow(a, b >> 1);
  return memo * memo;
}
lint modpow(lint a, lint b, lint m) {
  if (!b)
    return 1;
  if (b & 1)
    return modpow(a, b - 1, m) * a % m;
  lint memo = modpow(a, b >> 1, m);
  return memo * memo % m;
}
template <typename T> void printArray(std::vector<T> &vec) {
  rep(i, vec.size() - 1) std::cout << vec[i] << " ";
  std::cout << vec.back() << std::endl;
}
template <typename T> void printArray(T l, T r) {
  T rprev = r;
  rprev--;
  for (T i = l; i != rprev; i++) {
    std::cout << *i << " ";
  }
  std::cout << *rprev << std::endl;
}

class ModInt {
  lint value;

public:
  static const lint modulo;
  ModInt() : value(0) {}
  template <typename T> ModInt(T value = 0) : value(value) {
    if (value < 0)
      value = -(lint)(-value % modulo) + modulo;
    this->value = value % modulo;
  }
  inline operator int() const { return value; }
  inline ModInt &operator+=(const ModInt &x) {
    value += x.value;
    if (value >= modulo)
      value -= modulo;
    return *this;
  }
  inline ModInt &operator++() {
    if (value == modulo - 1)
      value = 0;
    else
      value++;
    return *this;
  }
  inline ModInt operator-() const { return ModInt(0) -= *this; }
  inline ModInt &operator-=(const ModInt &x) {
    value -= x.value;
    if (value < 0)
      value += modulo;
    return *this;
  }
  inline ModInt &operator--() {
    if (value == 0)
      value = modulo - 1;
    else
      value--;
    return *this;
  }
  inline ModInt &operator*=(const ModInt &x) {
    value = value * x.value % modulo;
    return *this;
  }
  inline ModInt &operator/=(ModInt rhs) {
    int exp = modulo - 2;
    while (exp) {
      if (exp & 1)
        *this *= rhs;
      rhs *= rhs;
      exp >>= 1;
    }
    return *this;
  }
  template <typename T> ModInt operator+(const T &rhs) const {
    return ModInt(*this) += rhs;
  }
  template <typename T> ModInt &operator+=(const T &rhs) {
    return operator+=(ModInt(rhs));
  }
  template <typename T> ModInt operator-(const T &rhs) const {
    return ModInt(*this) -= rhs;
  }
  template <typename T> ModInt &operator-=(const T &rhs) {
    return operator-=(ModInt(rhs));
  }
  template <typename T> ModInt operator*(const T &rhs) const {
    return ModInt(*this) *= rhs;
  }
  template <typename T> ModInt &operator*=(const T &rhs) {
    return operator*=(ModInt(rhs));
  }
  template <typename T> ModInt operator/(const T &rhs) const {
    return ModInt(*this) /= rhs;
  }
  template <typename T> ModInt &operator/=(const T &rhs) {
    return operator/=(ModInt(rhs));
  }
};
std::istream &operator>>(std::istream &ist, ModInt &x) {
  lint a;
  ist >> a;
  x = a;
  return ist;
}
const lint ModInt::modulo = 998244353;
std::string s;
lint k;
std::vector<P> vec;
ModInt dp[610][610][610], fact[1010], inv[1010];
ModInt comb(int a, int b) { return fact[a] * inv[a - b] * inv[b]; }
signed main() {
  std::cin >> s >> k;
  rep(i, s.size()) {
    if (s[i] == '0')
      vec.emplace_back(0, 1);
    else {
      if (!vec.empty() && vec.back().first == 1)
        vec.back().second++;
      else
        vec.emplace_back(1, 1);
    }
  }
  fact[0] = 1;
  REP(i, 909) fact[i] = fact[i - 1] * i;
  inv[909] = ModInt(1) / fact[909];
  for (int i = 908; i >= 0; i--)
    inv[i] = inv[i + 1] * (i + 1);
  lint zeroc = 0, onesum = 0;
  dp[0][0][0] = 1;
  rep(i, vec.size() + 1) {
    if (vec[i].first == 0)
      zeroc++;
    rep(j, onesum + 1) {
      rep(k, s.size()) {
        dp[i + 1][j][k] += dp[i][j][k];
        if (zeroc - k <= 0)
          continue;
        if (vec[i].first == 0)
          continue;
        REP(l, vec[i].second) {
          dp[i + 1][j + l][k + 1] += dp[i][j][k] * comb(zeroc - k + l - 1, l);
        }
      }
    }
    if (vec[i].first == 1)
      onesum += vec[i].second;
  }
  ModInt ans = 0;
  rep(i, std::min((lint)s.size(), k + 1)) {
    rep(j, s.size()) { ans += dp[vec.size()][i][j]; }
  }
  std::cout << ans << std::endl;
  return 0;
}