Submission #1299871

#TimeUsernameProblemLanguageResultExecution timeMemory
1299871sonnydaysKnapsack (NOI18_knapsack)C++20
73 / 100
1095 ms1992 KiB
#include <bits/stdc++.h>

using namespace std;

using ll = long long;
using ull = unsigned long long;
using pii = pair<int, int>;
using vi = vector<int>;
#define all(x) (x).begin(), (x).end()
ll v[100000];
int w[100000], k[100000];
ll dp[2][2001];

int main() {
    cin.tie(0);
    cout.tie(0);
    ios_base::sync_with_stdio(0);
    
    int s, n;
    cin >> s >> n;


    for (ll i = 0; i < n; ++i) {
        cin >> v[i] >> w[i] >> k[i];
    }

    for (int i = 0; i < n; ++i) {
        for (int r= 0; r < w[i]; ++r) {
            stack<pair<pair<ll, int>, ll>> st1, st2;

            for (int a = 0; a * w[i] + r <= s; ++a) {
                if (st2.empty()) {
                    while (!st1.empty()) {
                        auto element = st1.top().first;
                        st1.pop();
                        ll maximum = st2.empty() ? element.first : max(element.first, st2.top().second);
                        st2.push({element, maximum});
                    }
                }

                while (!st2.empty() && a - st2.top().first.second > k[i]) {
                    st2.pop();
                }

                ll cur_val = dp[(i + 1) & 1][a * w[i] + r] - v[i] * a;
                ll maximum = st1.empty() ? cur_val : max(cur_val, st1.top().second);
                st1.push({{cur_val, a}, maximum});

                if (!st2.empty()) maximum = max(maximum, st2.top().second);
                dp[i & 1][a * w[i] + r] =  maximum + a * v[i];
            }
        }
    }

    cout << dp[(n + 1) & 1][s] << '\n';

    return 0;
}
#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...