Submission #1053978

#	Time	Username	Problem	Language	Result	Execution time	Memory
1053978		ssitaram	Knapsack (NOI18_knapsack)	C++17	73 / 100	1096 ms	2816 KiB

knapsack

#include <bits/stdc++.h>
using namespace std;

typedef long long ll;

struct item {
	ll v, w, k;
};

int main() {
	ios::sync_with_stdio(0);
	cin.tie(nullptr);
	ll s, n; cin >> s >> n;
	vector<item> v(n);
	for (ll i = 0; i < n; ++i) {
		cin >> v[i].v >> v[i].w >> v[i].k;
	}
	vector<ll> best(s + 1), valof(s + 1);
	for (ll i = 0; i < n; ++i) {
		vector<set<tuple<ll, ll, ll>>> prevs(v[i].w);
		ll last = 3001;
		for (ll j = 0; j <= s; ++j) {
			if (!(j % v[i].w)) --last;
			valof[j] = best[j] + v[i].v * last;
			prevs[j % v[i].w].insert({valof[j], j, best[j]});
			if (prevs[j % v[i].w].size() == v[i].k + 2) {
				ll idx = j - v[i].w * (v[i].k + 1);
				auto it = prevs[j % v[i].w].lower_bound({valof[idx], idx, 0});
				prevs[j % v[i].w].erase(it);
			}
			auto it = prev(prevs[j % v[i].w].end());
			best[j] = get<2>(*it) + (j - get<1>(*it)) / v[i].w * v[i].v;
		}
	}
	cout << best.back() << '\n';
	return 0;
}

Compilation message (stderr)

knapsack.cpp: In function 'int main()':
knapsack.cpp:26:33: warning: comparison of integer expressions of different signedness: 'std::set<std::tuple<long long int, long long int, long long int> >::size_type' {aka 'long unsigned int'} and 'll' {aka 'long long int'} [-Wsign-compare]
   26 |    if (prevs[j % v[i].w].size() == v[i].k + 2) {
      |        ~~~~~~~~~~~~~~~~~~~~~~~~~^~~~~~~~~~~~~

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