Submission #709267

#	Time	Username	Problem	Language	Result	Execution time	Memory
709267		nima_aryan	Knapsack (NOI18_knapsack)	C++14	29 / 100	661 ms	12760 KiB

knapsack

#include <bits/stdc++.h>

using namespace std;

#pragma GCC optimize("Ofast")
#pragma GCC target("avx2")

#ifdef LOCAL
#include "algo/debug.h"
#endif

using i64 = long long;

template<class Fun>
class y_combinator_result {
   Fun fun_;
public:
   template<class T>
   explicit y_combinator_result(T &&fun): fun_(std::forward<T>(fun)) {}
 
   template<class ...Args>
   decltype(auto) operator()(Args &&...args) {
      return fun_(std::ref(*this), std::forward<Args>(args)...);
   }
};
 
template<class Fun>
decltype(auto) y_combinator(Fun &&fun) {
   return y_combinator_result<std::decay_t<Fun>>(std::forward<Fun>(fun));
}

int main() {
   ios::sync_with_stdio(false);
   cin.tie(nullptr);
   int S, N;
   cin >> S >> N;
   vector<int> V(N + 1), W(N + 1), K(N + 1);
   for (int i = 1; i <= N; ++i) {
      cin >> V[i] >> W[i] >> K[i];
   }
   const i64 inf = 1e18;
   map<pair<int, int>, i64> memo;
   auto solve = y_combinator([&](auto self, int sum, int n) -> i64 {
      if (sum < 0) {
         return -inf;
      }
      if (!n) {
         return sum ? -inf : 0;
      }
      auto pr = make_pair(sum, n);
      if (memo.count(pr)) {
         return memo[pr];
      }
      i64 ret = 0;
      for (int cnt = 0; min(K[N], sum / W[n] + 10) >= cnt; ++cnt) {
         ret = max(ret, self(sum - cnt * W[n], n - 1) + cnt * V[n]);
      }
      return memo[pr] = ret;
   });
   i64 ans = 0;
   for (int sum = 0; sum <= S; ++sum) {
      ans = max(ans, solve(sum, N));
   }
   cout << ans << '\n';
}

/* stuff you should look for
 * int overflow, array bounds
 * special cases (n=1?)
 * do smth instead of nothing and stay organized
 * WRITE STUFF DOWN
 * DON'T GET STUCK ON ONE APPROACH
 */

• Subtask 1: 12 / 12
• Subtask 2: 17 / 17
• Subtask 3: 0 / 20
• Subtask 4: 0 / 24
• Subtask 5: 0 / 27