제출 #709288

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
709288		nima_aryan	Knapsack (NOI18_knapsack)	C++17	49 / 100	1090 ms	10180 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 (!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
 */

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