Submission #659959

#	Time	Username	Problem	Language	Result	Execution time	Memory
659959		chanhchuong123	Knapsack (NOI18_knapsack)	C++14	100 / 100	60 ms	3756 KiB

knapsack

#include <bits/stdc++.h>
using namespace std;
#define task "C"
#define fi first
#define se second
#define all(x) (x).begin(), (x).end()
#define rall(x) (x).rbegin(), (x).rend()

template <typename T1, typename T2> bool mini(T1 &a, T2 b) {
  if (a > b) {a = b; return true;} return false;
}
template <typename T1, typename T2> bool maxi(T1 &a, T2 b) {
  if (a < b) {a = b; return true;} return false;
}

const int N = 2020;
int s, n;
int dp[N];
vector<int> a[N];
vector<pair<int, int>> V[N];

main() {
  ios_base::sync_with_stdio(0);
  cin.tie(0); cout.tie(0);

  cin >> s >> n;
  for (int i = 1; i <= n; i++) {
    int v, w, k; cin >> v >> w >> k;
    V[w].push_back({v, k});
  }
  for (int i = 1; i <= s; i++) {
    int LIM = s / i; sort(all(V[i]));
    while (V[i].size() && LIM > 0) {
      if (V[i].back().se > 0) {
        a[i].push_back(V[i].back().fi);
        V[i].back().se--; LIM--;
      } else V[i].pop_back();
    }
  }
  for (int i = 1; i <= s; i++) for (int j: a[i]) {
    for (int k = s; k >= i; k--) maxi(dp[k], dp[k - i] + j);
  }
  cout << dp[s];
}

Compilation message (stderr)

knapsack.cpp:22:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
   22 | main() {
      | ^~~~

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