Submission #654375

#	Time	Username	Problem	Language	Result	Execution time	Memory
654375		Nafeeszx	Knapsack (NOI18_knapsack)	C++14	100 / 100	73 ms	19420 KiB

knapsack

#pragma GCC optimize("O3")
#pragma GCC optimize("unroll-loops")
#include <bits/stdc++.h>
using namespace std;
#define trav(a, x) for(auto& a : x)
#define FOR(i, a, b) for (int i=(a); i<=(signed)(b); i++)
#define ROF(i, a, b) for (int i=(a); i>=(signed)(b); i--)
#define F0R(i, a) for (int i=0; i<(signed)(a); i++)
#define vi vector<int>
#define all(v) (v).begin(), (v).end()
#define f first
#define s second
typedef long long ll;

const int mod = 1e9 + 7, MOD = 998244353;

vector<pair<int, int>> wt[2005];
vector<int> sortwt[2005];
int dp[2005][2005];

int main()
{	
    ios_base::sync_with_stdio(0); cin.tie(0);
    memset(dp, -1e9-15, sizeof(dp));
    int s, n; cin >> s >> n;
    F0R(i, n) {
        int k, w, v; cin >> v >> w >> k;
        wt[w].push_back(make_pair(v, k));
    }
    FOR(i, 1, s) sort(wt[i].rbegin(), wt[i].rend());
    FOR(i, 1, s) {
        int wwt = 0, vvs = 0;
        int j = 0;
        while(wwt <= s && j < wt[i].size()) {
            wwt+=i;
            vvs += wt[i][j].first; 
            sortwt[i].push_back(vvs);
            wt[i][j].second--;
            if(wt[i][j].second == 0) j++; 
        }
    }
    //trav(u, sortwt[1]) cout << u << " ";
    
    dp[0][0] = 0;
    FOR(i, 1, s) {
        F0R(j, sortwt[i].size()) {
            FOR(k, 0, s) {
                if(k - i * (j+1) >= 0) dp[i][k] = max(dp[i][k], dp[i-1][k - i * (j+1)] + sortwt[i][j]);
            }
        }
        FOR(k, 0, s) {
            dp[i][k] = max(dp[i][k], dp[i-1][k]);
        }
    }
    int res = 0;
    FOR(i, 0, s) {FOR(j, 0, s) res = max(res, dp[i][j]);}
    cout << res;
    return 0;
}

Compilation message (stderr)

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

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