Problem p02662 (C++) — buggy submission, verdict: Time Limit Exceeded

#include <algorithm>
#include <bitset>
#include <cmath>
#include <cstdio>
#include <cstring>
#include <iomanip>
#include <iostream>
#include <map>
#include <queue>
#include <set>
#include <sstream>
#include <stack>
#include <unordered_map>
#include <vector>
// #include <bits/stdc++.h>
#define ALL(x) (x).begin(), (x).end()
#define sz(a) int(a.size())
#define rep(i, x, n) for (int i = x; i < n; i++)
#define repd(i, x, n) for (int i = x; i <= n; i++)
#define pii pair<int, int>
#define pll pair<long long, long long>
#define gbtb ios::sync_with_stdio(false), cin.tie(0), cout.tie(0)
#define MS0(X) memset((X), 0, sizeof((X)))
#define MSC0(X) memset((X), '\\0', sizeof((X)))
#define pb push_back
#define mp make_pair
#define fi first
#define se second
#define eps 1e-6
#define chu(x)                                                                 \\
  if (DEBUG_Switch)                                                            \\
  cout << "[" << #x << " " << (x) << "]" << endl
#define du3(a, b, c) scanf("%d %d %d", &(a), &(b), &(c))
#define du2(a, b) scanf("%d %d", &(a), &(b))
#define du1(a) scanf("%d", &(a));
using namespace std;
typedef long long ll;
ll gcd(ll a, ll b) { return b ? gcd(b, a % b) : a; }
ll lcm(ll a, ll b) { return a / gcd(a, b) * b; }
ll powmod(ll a, ll b, ll MOD) {
  if (a == 0ll) {
    return 0ll;
  }
  a %= MOD;
  ll ans = 1;
  while (b) {
    if (b & 1) {
      ans = ans * a % MOD;
    }
    a = a * a % MOD;
    b >>= 1;
  }
  return ans;
}
ll poww(ll a, ll b) {
  if (a == 0ll) {
    return 0ll;
  }
  ll ans = 1;
  while (b) {
    if (b & 1) {
      ans = ans * a;
    }
    a = a * a;
    b >>= 1;
  }
  return ans;
}
void Pv(const vector<int> &V) {
  int Len = sz(V);
  for (int i = 0; i < Len; ++i) {
    printf("%d", V[i]);
    if (i != Len - 1) {
      printf(" ");
    } else {
      printf("\\n");
    }
  }
}
void Pvl(const vector<ll> &V) {
  int Len = sz(V);
  for (int i = 0; i < Len; ++i) {
    printf("%lld", V[i]);
    if (i != Len - 1) {
      printf(" ");
    } else {
      printf("\\n");
    }
  }
}
inline long long readll() {
  long long tmp = 0, fh = 1;
  char c = getchar();
  while (c < '0' || c > '9') {
    if (c == '-')
      fh = -1;
    c = getchar();
  }
  while (c >= '0' && c <= '9')
    tmp = tmp * 10 + c - 48, c = getchar();
  return tmp * fh;
}
inline int readint() {
  int tmp = 0, fh = 1;
  char c = getchar();
  while (c < '0' || c > '9') {
    if (c == '-')
      fh = -1;
    c = getchar();
  }
  while (c >= '0' && c <= '9')
    tmp = tmp * 10 + c - 48, c = getchar();
  return tmp * fh;
}
void pvarr_int(int *arr, int n, int strat = 1) {
  if (strat == 0) {
    n--;
  }
  repd(i, strat, n) { printf("%d%c", arr[i], i == n ? '\\n' : ' '); }
}
void pvarr_LL(ll *arr, int n, int strat = 1) {
  if (strat == 0) {
    n--;
  }
  repd(i, strat, n) { printf("%lld%c", arr[i], i == n ? '\\n' : ' '); }
}
const int maxn = 3000 + 10;
const int inf = 0x3f3f3f3f;
/*** TEMPLATE CODE * * STARTS HERE ***/
#define DEBUG_Switch 0
ll dp[maxn][maxn];
int a[maxn];
const ll mod = 998244353;
int main() {
#if DEBUG_Switch
  freopen("C:\\\\code\\\\input.txt", "r", stdin);
#endif
  // freopen("C:\\\\code\\\\output.txt","r",stdin);
  int n = readint();
  int s = readint();
  repd(i, 1, n) { a[i] = readint(); }
  dp[0][0] = 1;
  repd(i, 1, n) {
    repd(j, a[i], s) {
      dp[i][j] += dp[i - 1][j - a[i]];
      dp[i][j] %= mod;
    }
    repd(j, 0, s) {
      dp[i][j] += dp[i - 1][j] * 2;
      dp[i][j] %= mod;
    }
  }
  printf("%lld\\n", dp[n][s]);
  return 0;
}


Error: TLE

Fixed submission:

#include <algorithm>
#include <bitset>
#include <cmath>
#include <cstdio>
#include <cstring>
#include <iomanip>
#include <iostream>
#include <map>
#include <queue>
#include <set>
#include <sstream>
#include <stack>
#include <unordered_map>
#include <vector>
// #include <bits/stdc++.h>
#define ALL(x) (x).begin(), (x).end()
#define sz(a) int(a.size())
#define rep(i, x, n) for (int i = x; i < n; i++)
#define repd(i, x, n) for (int i = x; i <= n; i++)
#define pii pair<int, int>
#define pll pair<long long, long long>
#define gbtb ios::sync_with_stdio(false), cin.tie(0), cout.tie(0)
#define MS0(X) memset((X), 0, sizeof((X)))
#define MSC0(X) memset((X), '\\0', sizeof((X)))
#define pb push_back
#define mp make_pair
#define fi first
#define se second
#define eps 1e-6
#define chu(x)                                                                 \\
  if (DEBUG_Switch)                                                            \\
  cout << "[" << #x << " " << (x) << "]" << endl
#define du3(a, b, c) scanf("%d %d %d", &(a), &(b), &(c))
#define du2(a, b) scanf("%d %d", &(a), &(b))
#define du1(a) scanf("%d", &(a));
using namespace std;
typedef long long ll;
ll gcd(ll a, ll b) { return b ? gcd(b, a % b) : a; }
ll lcm(ll a, ll b) { return a / gcd(a, b) * b; }
ll powmod(ll a, ll b, ll MOD) {
  if (a == 0ll) {
    return 0ll;
  }
  a %= MOD;
  ll ans = 1;
  while (b) {
    if (b & 1) {
      ans = ans * a % MOD;
    }
    a = a * a % MOD;
    b >>= 1;
  }
  return ans;
}
ll poww(ll a, ll b) {
  if (a == 0ll) {
    return 0ll;
  }
  ll ans = 1;
  while (b) {
    if (b & 1) {
      ans = ans * a;
    }
    a = a * a;
    b >>= 1;
  }
  return ans;
}
void Pv(const vector<int> &V) {
  int Len = sz(V);
  for (int i = 0; i < Len; ++i) {
    printf("%d", V[i]);
    if (i != Len - 1) {
      printf(" ");
    } else {
      printf("\\n");
    }
  }
}
void Pvl(const vector<ll> &V) {
  int Len = sz(V);
  for (int i = 0; i < Len; ++i) {
    printf("%lld", V[i]);
    if (i != Len - 1) {
      printf(" ");
    } else {
      printf("\\n");
    }
  }
}
inline long long readll() {
  long long tmp = 0, fh = 1;
  char c = getchar();
  while (c < '0' || c > '9') {
    if (c == '-')
      fh = -1;
    c = getchar();
  }
  while (c >= '0' && c <= '9')
    tmp = tmp * 10 + c - 48, c = getchar();
  return tmp * fh;
}
inline int readint() {
  int tmp = 0, fh = 1;
  char c = getchar();
  while (c < '0' || c > '9') {
    if (c == '-')
      fh = -1;
    c = getchar();
  }
  while (c >= '0' && c <= '9')
    tmp = tmp * 10 + c - 48, c = getchar();
  return tmp * fh;
}
void pvarr_int(int *arr, int n, int strat = 1) {
  if (strat == 0) {
    n--;
  }
  repd(i, strat, n) { printf("%d%c", arr[i], i == n ? '\\n' : ' '); }
}
void pvarr_LL(ll *arr, int n, int strat = 1) {
  if (strat == 0) {
    n--;
  }
  repd(i, strat, n) { printf("%lld%c", arr[i], i == n ? '\\n' : ' '); }
}
const int maxn = 1000010;
const int inf = 0x3f3f3f3f;
/*** TEMPLATE CODE * * STARTS HERE ***/
#define DEBUG_Switch 0
int n;
int s;
int a[maxn];
int dp[maxn];
std::vector<int> f[maxn];
const ll mod = 998244353;
int main() {
#if DEBUG_Switch
  freopen("C:\\\\code\\\\input.txt", "r", stdin);
#endif
  // freopen("C:\\\\code\\\\output.txt","r",stdin);
  n = readint();
  s = readint();
  repd(i, 1, n) { a[i] = readint(); }
  dp[0] = 1;
  f[0].pb(0);
  repd(j, 1, n) {
    for (int i = s; i >= a[j]; --i) {
      if (dp[i - a[j]]) {
        dp[i] = 1;
        for (auto &x : f[i - a[j]]) {
          f[i].pb(x + 1);
        }
      }
    }
  }
  ll ans = 0ll;
  for (auto &x : f[s]) {
    ans += powmod(2ll, n - x, mod);
    ans %= mod;
  }
  printf("%lld\\n", ans);
  return 0;
}