제출 #631027

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
631027		thienbao1602	Knapsack (NOI18_knapsack)	C++17	100 / 100	97 ms	36028 KiB

knapsack

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

const int N = 2005;
const ll inf = -1e15;

int S, n;

void solve()
{
    cin >> S >> n;
    map<ll, vector<pair<ll, ll>>> weight;
    for(int i=0; i<n; i++)
    {
        ll w, val, sl;
        cin >> val >> w >> sl;
        weight[w].push_back({val, sl});
    }

    int id = 1;
    vector<vector<ll>> ans((int)weight.size()+1, vector<ll>(S+1, inf));
    ans[0][0] = 0;
    for(auto [w, items] : weight)
    {
        sort(items.begin(), items.end(), greater<pair<ll, ll>>());
        for(int i=0; i<=S; i++)
        {
            ans[id][i] = ans[id-1][i];
            ll sl = 0, profit = 0, type_at = 0, used = 0;
            while((sl+1) * w <= i && type_at < (ll)items.size())
            {
                sl += 1;
                profit += items[type_at].first;
                if (ans[id-1][i-sl*w] != inf)
                {
                    ans[id][i] = max(ans[id][i], ans[id-1][i-sl*w] + profit);
                }

                used++;
                if (used == (ll)items[type_at].second)
                {
                    used = 0;
                    type_at++;
                }
            }
        }
        id++;
    }

    cout << *max_element(ans.back().begin(), ans.back().end());
}

int main()
{
    ios_base::sync_with_stdio(false);
    cin.tie(0); cout.tie(0);
    solve();
    return 0;
}

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