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

#pragma region kyopro_template
#include <bits/stdc++.h>
#define pb push_back
#define eb emplace_back
#define fi first
#define se second
#define each(x, v) for (auto &x : v)
#define all(v) (v).begin(), (v).end()
#define sz(v) ((int)(v).size())
#define mem(a, val) memset(a, val, sizeof(a))
#define ini(...)                                                               \\
  int __VA_ARGS__;                                                             \\
  in(__VA_ARGS__)
#define inl(...)                                                               \\
  long long __VA_ARGS__;                                                       \\
  in(__VA_ARGS__)
#define ins(...)                                                               \\
  string __VA_ARGS__;                                                          \\
  in(__VA_ARGS__)
#define inc(...)                                                               \\
  char __VA_ARGS__;                                                            \\
  in(__VA_ARGS__)
#define in2(s, t)                                                              \\
  for (int i = 0; i < (int)s.size(); i++) {                                    \\
    in(s[i], t[i]);                                                            \\
  }
#define in3(s, t, u)                                                           \\
  for (int i = 0; i < (int)s.size(); i++) {                                    \\
    in(s[i], t[i], u[i]);                                                      \\
  }
#define in4(s, t, u, v)                                                        \\
  for (int i = 0; i < (int)s.size(); i++) {                                    \\
    in(s[i], t[i], u[i], v[i]);                                                \\
  }
#define rep(i, N) for (long long i = 0; i < (long long)(N); i++)
#define repr(i, N) for (long long i = (long long)(N)-1; i >= 0; i--)
#define rep1(i, N) for (long long i = 1; i <= (long long)(N); i++)
#define repr1(i, N) for (long long i = (N); (long long)(i) > 0; i--)
using namespace std;
void solve();
using ll = long long;
template <class T = ll> using V = vector<T>;
using vi = vector<int>;
using vl = vector<long long>;
using vvi = vector<vector<int>>;
using vd = V<double>;
using vs = V<string>;
using vvl = vector<vector<long long>>;
using P = pair<long long, long long>;
using vp = vector<P>;
using pii = pair<int, int>;
using vpi = vector<pair<int, int>>;
constexpr int inf = 1001001001;
constexpr long long infLL = (1LL << 61) - 1;
template <typename T, typename U> inline bool amin(T &x, U y) {
  return (y < x) ? (x = y, true) : false;
}
template <typename T, typename U> inline bool amax(T &x, U y) {
  return (x < y) ? (x = y, true) : false;
}
template <typename T, typename U> ll ceil(T a, U b) { return (a + b - 1) / b; }
constexpr ll TEN(int n) {
  ll ret = 1, x = 10;
  while (n) {
    if (n & 1)
      ret *= x;
    x *= x;
    n >>= 1;
  }
  return ret;
}

template <typename T, typename U>
ostream &operator<<(ostream &os, const pair<T, U> &p) {
  os << p.first << " " << p.second;
  return os;
}
template <typename T, typename U>
istream &operator>>(istream &is, pair<T, U> &p) {
  is >> p.first >> p.second;
  return is;
}
template <typename T> ostream &operator<<(ostream &os, const vector<T> &v) {
  int s = (int)v.size();
  for (int i = 0; i < s; i++)
    os << (i ? " " : "") << v[i];
  return os;
}
template <typename T> istream &operator>>(istream &is, vector<T> &v) {
  for (auto &x : v)
    is >> x;
  return is;
}
void in() {}
template <typename T, class... U> void in(T &t, U &...u) {
  cin >> t;
  in(u...);
}
void out() { cout << "\\n"; }
template <typename T, class... U> void out(const T &t, const U &...u) {
  cout << t;
  if (sizeof...(u))
    cout << " ";
  out(u...);
}
template <typename T> void die(T x) {
  out(x);
  exit(0);
}

#ifdef NyaanDebug
#include "NyaanDebug.h"
#define trc(...)                                                               \\
  do {                                                                         \\
    cerr << #__VA_ARGS__ << " = ";                                             \\
    dbg_out(__VA_ARGS__);                                                      \\
  } while (0)
#define trca(v, N)                                                             \\
  do {                                                                         \\
    cerr << #v << " = ";                                                       \\
    array_out(v, N);                                                           \\
  } while (0)
#define trcc(v)                                                                \\
  do {                                                                         \\
    cerr << #v << " = {";                                                      \\
    each(x, v) { cerr << " " << x << ","; }                                    \\
    cerr << "}" << endl;                                                       \\
  } while (0)
#else
#define trc(...)
#define trca(...)
#define trcc(...)
int main() { solve(); }
#endif

/*struct IoSetupNya {
  IoSetupNya() {
    cin.tie(nullptr);
    ios::sync_with_stdio(false);
    cout << fixed << setprecision(15);
    cerr << fixed << setprecision(7);
  }
} iosetupnya;*/

inline int popcnt(unsigned long long a) { return __builtin_popcountll(a); }
inline int lsb(unsigned long long a) { return __builtin_ctzll(a); }
inline int msb(unsigned long long a) { return 63 - __builtin_clzll(a); }
template <typename T> inline int getbit(T a, int i) { return (a >> i) & 1; }
template <typename T> inline void setbit(T &a, int i) { a |= (1LL << i); }
template <typename T> inline void delbit(T &a, int i) { a &= ~(1LL << i); }
template <typename T> int lb(const vector<T> &v, const T &a) {
  return lower_bound(begin(v), end(v), a) - begin(v);
}
template <typename T> int ub(const vector<T> &v, const T &a) {
  return upper_bound(begin(v), end(v), a) - begin(v);
}
template <typename T> vector<T> mkrui(const vector<T> &v) {
  vector<T> ret(v.size() + 1);
  for (int i = 0; i < int(v.size()); i++)
    ret[i + 1] = ret[i] + v[i];
  return ret;
};
template <typename T> vector<T> mkuni(const vector<T> &v) {
  vector<T> ret(v);
  sort(ret.begin(), ret.end());
  ret.erase(unique(ret.begin(), ret.end()), ret.end());
  return ret;
}
template <typename F> vector<int> mkord(int N, F f) {
  vector<int> ord(N);
  iota(begin(ord), end(ord), 0);
  sort(begin(ord), end(ord), f);
  return ord;
}
template <typename T = int> vector<T> mkiota(int N) {
  vector<T> ret(N);
  iota(begin(ret), end(ret), 0);
  return ret;
}

#pragma endregion

constexpr int MOD = 998244353;

ll dp[516][101010];

void solve() {
  ini(N);
  vl v(N), w(N);
  in2(v, w);

  int thres = 512;
  int maxw = 100000;

  rep(i, thres) rep(j, maxw + 1) dp[i][j] = -infLL;
  dp[0][0] = 0;
  dp[0][w[0]] = v[0];
  rep(i, thres / 2) rep(j, maxw + 1) {
    if (i >= N)
      break;
    if (dp[i][j] < 0)
      continue;
    rep1(k, 2) {
      int dst = i * 2 + k;
      if (dst >= N)
        break;
      amax(dp[dst][j], dp[i][j]);
      if (j + w[dst] <= 100000)
        amax(dp[dst][j + w[dst]], dp[i][j] + v[dst]);
    }
  }
  rep(i, thres) {
    if (i >= N)
      break;
    rep(j, 100000) amax(dp[i][j + 1], dp[i][j]);
  }

  P ep[2048];

  ini(Q);
  rep(_, Q) {
    inl(n, ww);
    n--;
    if (n < thres) {
      out(dp[n][ww]);
      continue;
    }
    ep[0] = {0, 0};
    int epsz = 1;
    while (n >= thres) {
      rep(i, epsz) ep[i + epsz] = {ep[i].first + w[n], ep[i].second + v[n]};
      n = (n - 1) >> 1;
      epsz <<= 1;
    }
    ll ans = 0;
    rep(i, epsz) {
      auto p = ep[i];
      if (ww - p.first >= 0)
        amax(ans, p.second + dp[n][ww - p.first]);
    }
    out(ans);
  }
}

Error: TLE

Fixed submission:

#pragma region kyopro_template
#include <bits/stdc++.h>
#define pb push_back
#define eb emplace_back
#define fi first
#define se second
#define each(x, v) for (auto &x : v)
#define all(v) (v).begin(), (v).end()
#define sz(v) ((int)(v).size())
#define mem(a, val) memset(a, val, sizeof(a))
#define ini(...)                                                               \\
  int __VA_ARGS__;                                                             \\
  in(__VA_ARGS__)
#define inl(...)                                                               \\
  long long __VA_ARGS__;                                                       \\
  in(__VA_ARGS__)
#define ins(...)                                                               \\
  string __VA_ARGS__;                                                          \\
  in(__VA_ARGS__)
#define inc(...)                                                               \\
  char __VA_ARGS__;                                                            \\
  in(__VA_ARGS__)
#define in2(s, t)                                                              \\
  for (int i = 0; i < (int)s.size(); i++) {                                    \\
    in(s[i], t[i]);                                                            \\
  }
#define in3(s, t, u)                                                           \\
  for (int i = 0; i < (int)s.size(); i++) {                                    \\
    in(s[i], t[i], u[i]);                                                      \\
  }
#define in4(s, t, u, v)                                                        \\
  for (int i = 0; i < (int)s.size(); i++) {                                    \\
    in(s[i], t[i], u[i], v[i]);                                                \\
  }
#define rep(i, N) for (long long i = 0; i < (long long)(N); i++)
#define repr(i, N) for (long long i = (long long)(N)-1; i >= 0; i--)
#define rep1(i, N) for (long long i = 1; i <= (long long)(N); i++)
#define repr1(i, N) for (long long i = (N); (long long)(i) > 0; i--)
using namespace std;
void solve();
using ll = long long;
template <class T = ll> using V = vector<T>;
using vi = vector<int>;
using vl = vector<long long>;
using vvi = vector<vector<int>>;
using vd = V<double>;
using vs = V<string>;
using vvl = vector<vector<long long>>;
using P = pair<long long, long long>;
using vp = vector<P>;
using pii = pair<int, int>;
using vpi = vector<pair<int, int>>;
constexpr int inf = 1001001001;
constexpr long long infLL = (1LL << 61) - 1;
template <typename T, typename U> inline bool amin(T &x, U y) {
  return (y < x) ? (x = y, true) : false;
}
template <typename T, typename U> inline bool amax(T &x, U y) {
  return (x < y) ? (x = y, true) : false;
}
template <typename T, typename U> ll ceil(T a, U b) { return (a + b - 1) / b; }
constexpr ll TEN(int n) {
  ll ret = 1, x = 10;
  while (n) {
    if (n & 1)
      ret *= x;
    x *= x;
    n >>= 1;
  }
  return ret;
}

template <typename T, typename U>
ostream &operator<<(ostream &os, const pair<T, U> &p) {
  os << p.first << " " << p.second;
  return os;
}
template <typename T, typename U>
istream &operator>>(istream &is, pair<T, U> &p) {
  is >> p.first >> p.second;
  return is;
}
template <typename T> ostream &operator<<(ostream &os, const vector<T> &v) {
  int s = (int)v.size();
  for (int i = 0; i < s; i++)
    os << (i ? " " : "") << v[i];
  return os;
}
template <typename T> istream &operator>>(istream &is, vector<T> &v) {
  for (auto &x : v)
    is >> x;
  return is;
}
void in() {}
template <typename T, class... U> void in(T &t, U &...u) {
  cin >> t;
  in(u...);
}
void out() { cout << "\\n"; }
template <typename T, class... U> void out(const T &t, const U &...u) {
  cout << t;
  if (sizeof...(u))
    cout << " ";
  out(u...);
}
template <typename T> void die(T x) {
  out(x);
  exit(0);
}

#ifdef NyaanDebug
#include "NyaanDebug.h"
#define trc(...)                                                               \\
  do {                                                                         \\
    cerr << #__VA_ARGS__ << " = ";                                             \\
    dbg_out(__VA_ARGS__);                                                      \\
  } while (0)
#define trca(v, N)                                                             \\
  do {                                                                         \\
    cerr << #v << " = ";                                                       \\
    array_out(v, N);                                                           \\
  } while (0)
#define trcc(v)                                                                \\
  do {                                                                         \\
    cerr << #v << " = {";                                                      \\
    each(x, v) { cerr << " " << x << ","; }                                    \\
    cerr << "}" << endl;                                                       \\
  } while (0)
#else
#define trc(...)
#define trca(...)
#define trcc(...)
int main() { solve(); }
#endif

/*struct IoSetupNya {
  IoSetupNya() {
    cin.tie(nullptr);
    ios::sync_with_stdio(false);
    cout << fixed << setprecision(15);
    cerr << fixed << setprecision(7);
  }
} iosetupnya;*/

inline int popcnt(unsigned long long a) { return __builtin_popcountll(a); }
inline int lsb(unsigned long long a) { return __builtin_ctzll(a); }
inline int msb(unsigned long long a) { return 63 - __builtin_clzll(a); }
template <typename T> inline int getbit(T a, int i) { return (a >> i) & 1; }
template <typename T> inline void setbit(T &a, int i) { a |= (1LL << i); }
template <typename T> inline void delbit(T &a, int i) { a &= ~(1LL << i); }
template <typename T> int lb(const vector<T> &v, const T &a) {
  return lower_bound(begin(v), end(v), a) - begin(v);
}
template <typename T> int ub(const vector<T> &v, const T &a) {
  return upper_bound(begin(v), end(v), a) - begin(v);
}
template <typename T> vector<T> mkrui(const vector<T> &v) {
  vector<T> ret(v.size() + 1);
  for (int i = 0; i < int(v.size()); i++)
    ret[i + 1] = ret[i] + v[i];
  return ret;
};
template <typename T> vector<T> mkuni(const vector<T> &v) {
  vector<T> ret(v);
  sort(ret.begin(), ret.end());
  ret.erase(unique(ret.begin(), ret.end()), ret.end());
  return ret;
}
template <typename F> vector<int> mkord(int N, F f) {
  vector<int> ord(N);
  iota(begin(ord), end(ord), 0);
  sort(begin(ord), end(ord), f);
  return ord;
}
template <typename T = int> vector<T> mkiota(int N) {
  vector<T> ret(N);
  iota(begin(ret), end(ret), 0);
  return ret;
}

#pragma endregion

constexpr int MOD = 998244353;

ll dp[516][101010];

void solve() {
  ini(N);
  vl v(N), w(N);
  in2(v, w);

  int thres = 4;
  int maxw = 100000;

  rep(i, thres) rep(j, maxw + 1) dp[i][j] = -infLL;
  dp[0][0] = 0;
  dp[0][w[0]] = v[0];
  rep(i, thres / 2) rep(j, maxw + 1) {
    if (i >= N)
      break;
    if (dp[i][j] < 0)
      continue;
    rep1(k, 2) {
      int dst = i * 2 + k;
      if (dst >= N)
        break;
      amax(dp[dst][j], dp[i][j]);
      if (j + w[dst] <= 100000)
        amax(dp[dst][j + w[dst]], dp[i][j] + v[dst]);
    }
  }
  rep(i, thres) {
    if (i >= N)
      break;
    rep(j, 100000) amax(dp[i][j + 1], dp[i][j]);
  }

  vp ep, fp;
  ep.reserve(thres);
  fp.reserve(thres);

  ini(Q);
  rep(_, Q) {
    inl(n, ww);
    n--;
    if (n < thres) {
      out(dp[n][ww]);
      continue;
    }
    ep.clear();
    fp.clear();
    ep.emplace_back(0, 0);
    while (n >= thres) {
      fp.clear();
      each(p, ep) {
        fp.pb(p);
        if (w[n] + p.first <= ww)
          fp.eb(p.first + w[n], p.second + v[n]);
      }
      swap(ep, fp);
      n = (n - 1) >> 1;
    }
    ll ans = 0;
    each(p, ep) if (ww - p.first >= 0)
        amax(ans, p.second + dp[n][ww - p.first]);
    out(ans);
  }
}