Submission #886801

#	Time	Username	Problem	Language	Result	Execution time	Memory
886801		xn_17	Knapsack (NOI18_knapsack)	C++17	100 / 100	92 ms	4956 KiB

knapsack

//oj.uz--NOI18_KNAPSACK
//USACO Knapsack DP

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

int main()
{
    int s,n; cin >> s >> n;

    vector<pair<ll,ll>> weights[s+1];

    //pair<int,pair<int,int>> items[n];

    for(int i = 0; i < n; i++)
    {
        int v, w, k; cin >> v >> w >> k;
        //items[i] = mp(w, mp(v, k));
        
        if(w <= s)
        {
            weights[w].pb(mp(v,k));
        }
    }

    for(int i = 0; i <= s; i++)
    {
        sort(weights[i].begin(), weights[i].end(), greater());
    }

    ll dp[s+1];

    for(int i = 0; i <= s; i++)
    {
        dp[i] = 0;
    }

    for(int w = 1; w <= s; w++)
    {
        int i = 0; int tot = 0;

        while(i < weights[w].size() && tot <= s)
        {
            ll v,k; tie(v,k) = weights[w][i];

            for(int j = s; j >= w; j--)
            {
                dp[j] = max(dp[j], dp[j-w] + v);
            }

            weights[w][i].second -= 1; 

            if(k == 1)
            {
                i++;
            }

            tot += w;
        }
    }

    cout << dp[s];
}

Compilation message (stderr)

knapsack.cpp: In function 'int main()':
knapsack.cpp:45:17: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<long long int, long long int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   45 |         while(i < weights[w].size() && tot <= s)
      |               ~~^~~~~~~~~~~~~~~~~~~

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