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

#include <algorithm>     //sort,二分探索,など
#include <bitset>        //固定長bit集合
#include <cmath>         //pow,logなど
#include <complex>       //複素数
#include <deque>         //両端アクセスのキュー
#include <functional>    //sortのgreater
#include <iomanip>       //setprecision(浮動小数点の出力の誤差)
#include <iostream>      //入出力
#include <iterator>      //集合演算(積集合,和集合,差集合など)
#include <map>           //map(辞書)
#include <numeric>       //iota(整数列の生成),gcdとlcm(c++17)
#include <queue>         //キュー
#include <set>           //集合
#include <stack>         //スタック
#include <string>        //文字列
#include <unordered_map> //イテレータあるけど順序保持しないmap
#include <unordered_set> //イテレータあるけど順序保持しないset
#include <utility>       //pair
#include <vector>        //可変長配列

using namespace std;
typedef long long ll;

// マクロ
// forループ関係
// 引数は、(ループ内変数,動く範囲)か(ループ内変数,始めの数,終わりの数)、のどちらか
// Dがついてないものはループ変数は1ずつインクリメントされ、Dがついてるものはループ変数は1ずつデクリメントされる
#define REP(i, n) for (ll i = 0; i < (ll)(n); i++)
#define REPD(i, n) for (ll i = n - 1; i >= 0; i--)
#define FOR(i, a, b) for (ll i = a; i <= (ll)(b); i++)
#define FORD(i, a, b) for (ll i = a; i >= (ll)(b); i--)
// xにはvectorなどのコンテナ
#define ALL(x) (x).begin(), (x).end() // sortなどの引数を省略したい
#define SIZE(x) ((ll)(x).size())      // sizeをsize_tからllに直しておく
#define MAX(x) *max_element(ALL(x))   // 最大値を求める
#define MIN(x) *min_element(ALL(x))   // 最小値を求める
// 定数
#define INF 9000000000000000000 // 9e18:極めて大きい値
#define MOD 998244353
#define MAXR 100000 // 10^5:配列の最大のrange(素数列挙などで使用)
#define LOGMAXR 18  // floor(log2(10^5)) + 1: 配列の最大のrangeのlog
// 略記
// #define PB push_back //vectorヘの挿入
// #define MP make_pair //pairのコンストラクタ
// #define F first //pairの一つ目の要素
// #define S second //pairの二つ目の要素

ll modpow(ll n, ll p, ll mod) {
  ll ret = 1;
  while (p) {
    if (p & 1) {
      ret *= n;
      ret %= mod;
    }
    p >>= 1;
    n = n * n;
    n %= mod;
  }
  return ret;
}

int main() {
  ll N, S;
  cin >> N >> S;
  ll A[N];
  REP(n, N) cin >> A[n];

  ll memo[S + 1];
  REP(s, S + 1) memo[s] = 0;
  memo[0] = 1;
  for (auto &a : A) {
    REPD(n, S + 1) {
      if (n + a < S + 1) {
        memo[n + a] += memo[n];
        memo[n + a] %= MOD;
      }
      memo[n] *= 2;
      memo[n] %= MOD;
    }
  }

  // for(auto& s : memo) cout << s << endl;

  ll ret = memo[S];
  cout << ret << endl;

  return 0;
}


Error: TLE

Fixed submission:

#include <algorithm>     //sort,二分探索,など
#include <bitset>        //固定長bit集合
#include <cmath>         //pow,logなど
#include <complex>       //複素数
#include <deque>         //両端アクセスのキュー
#include <functional>    //sortのgreater
#include <iomanip>       //setprecision(浮動小数点の出力の誤差)
#include <iostream>      //入出力
#include <iterator>      //集合演算(積集合,和集合,差集合など)
#include <map>           //map(辞書)
#include <numeric>       //iota(整数列の生成),gcdとlcm(c++17)
#include <queue>         //キュー
#include <set>           //集合
#include <stack>         //スタック
#include <string>        //文字列
#include <unordered_map> //イテレータあるけど順序保持しないmap
#include <unordered_set> //イテレータあるけど順序保持しないset
#include <utility>       //pair
#include <vector>        //可変長配列

using namespace std;
typedef long long ll;

// マクロ
// forループ関係
// 引数は、(ループ内変数,動く範囲)か(ループ内変数,始めの数,終わりの数)、のどちらか
// Dがついてないものはループ変数は1ずつインクリメントされ、Dがついてるものはループ変数は1ずつデクリメントされる
#define REP(i, n) for (ll i = 0; i < (ll)(n); i++)
#define REPD(i, n) for (ll i = n - 1; i >= 0; i--)
#define FOR(i, a, b) for (ll i = a; i <= (ll)(b); i++)
#define FORD(i, a, b) for (ll i = a; i >= (ll)(b); i--)
// xにはvectorなどのコンテナ
#define ALL(x) (x).begin(), (x).end() // sortなどの引数を省略したい
#define SIZE(x) ((ll)(x).size())      // sizeをsize_tからllに直しておく
#define MAX(x) *max_element(ALL(x))   // 最大値を求める
#define MIN(x) *min_element(ALL(x))   // 最小値を求める
// 定数
#define INF 9000000000000000000 // 9e18:極めて大きい値
#define MOD 998244353
#define MAXR 100000 // 10^5:配列の最大のrange(素数列挙などで使用)
#define LOGMAXR 18  // floor(log2(10^5)) + 1: 配列の最大のrangeのlog
// 略記
// #define PB push_back //vectorヘの挿入
// #define MP make_pair //pairのコンストラクタ
// #define F first //pairの一つ目の要素
// #define S second //pairの二つ目の要素

ll modpow(ll n, ll p, ll mod) {
  ll ret = 1;
  while (p) {
    if (p & 1) {
      ret *= n;
      ret %= mod;
    }
    p >>= 1;
    n = n * n;
    n %= mod;
  }
  return ret;
}

int main() {
  ll N, S;
  cin >> N >> S;
  ll A[N];
  REP(n, N) cin >> A[n];

  map<ll, map<ll, ll>> memo, memo_tmp;
  for (auto a : A) {
    for (auto &it0 : memo) {
      ll s = it0.first;
      if (s + a > S)
        continue;
      for (auto &it1 : it0.second) {
        ll n = it1.first;
        ll c = it1.second;
        memo_tmp[s + a][n + 1] = c;
      }
    }
    if (a <= S) {
      // cout << a << endl;
      if (memo_tmp[a].count(1)) {
        memo_tmp[a][1] += 1;
      } else {
        memo_tmp[a][1] = 1;
      }
    }
    for (auto &it0 : memo_tmp) {
      ll s = it0.first;
      // cout << s << endl;
      for (auto &it1 : it0.second) {
        ll n = it1.first;
        ll c = it1.second;
        // cout << n << "  " << c << endl;
        if (memo[s].count(n)) {
          memo[s][n] += c;
        } else {
          memo[s][n] = c;
        }
      }
    }
    memo_tmp.clear();
  }

  // for(auto& it0 : memo){
  //   ll s = it0.first;
  //   cout << s << endl;
  //   for(auto& it1 : it0.second){
  //     ll n = it1.first;
  //     ll c = it1.second;
  //     cout << n << "  " << c << endl;
  //   }
  // }

  ll ret = 0;
  for (auto &it : memo[S]) {
    ll n = it.first;
    ll c = it.second;
    // cout << n << "  " << c << endl;
    ll tmp = modpow(2, N - n, MOD) * c;
    tmp %= MOD;
    ret += tmp;
    ret %= MOD;
  }
  cout << ret << endl;

  return 0;
}