Submission #709288

#TimeUsernameProblemLanguageResultExecution timeMemory
709288nima_aryanKnapsack (NOI18_knapsack)C++17
49 / 100
1090 ms10180 KiB
#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 (!n) {
         return sum <= S ? 0 : -inf;
      }
      auto pr = make_pair(sum, n);
      if (memo.count(pr)) {
         return memo[pr];
      }
      i64 ret = 0;
      for (int cnt = 0; sum - cnt * W[n] >= 0 && cnt <= K[n]; ++cnt) {
         ret = max(ret, self(sum - cnt * W[n], n - 1) + cnt * V[n]);
      }
      return memo[pr] = ret;
   });
   cout << solve(S, N) << '\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
 */
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...