Submission #788056

#	Time	Username	Problem	Language	Result	Execution time	Memory
788056		Andrey	Knapsack (NOI18_knapsack)	C++14	100 / 100	52 ms	5068 KiB

knapsack

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

vector<pair<long long,long long>> haha[2001];

int main()
{
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);
    long long n,a,b,c,s,y,ans = 0;
    cin >> s >> n;
    for(long long i = 0; i < n; i++) {
        cin >> a >> b >> c;
        haha[b].push_back({a,c});
    }
    vector<pair<long long,long long>> bruh(0);
    for(long long i = 1; i <= s; i++) {
        sort(haha[i].begin(),haha[i].end());
        reverse(haha[i].begin(),haha[i].end());
        c = s/i;
        y = 0;
        while(c > 0 && !haha[i].empty()) {
            if(haha[i][y].second == 0) {
                y++;
                if(y >= haha[i].size()) {
                    break;
                }
            }
            bruh.push_back({i,haha[i][y].first});
            haha[i][y] = {haha[i][y].first,haha[i][y].second-1};
            c--;
        }
    }
    vector<long long> dp(s+1);
    for(long long i = 0; i < bruh.size(); i++) {
        a = bruh[i].first;
        b = bruh[i].second;
        for(long long j = s; j >= a; j--) {
            dp[j] = max(dp[j],dp[j-a]+b);
            ans = max(ans,dp[j]);
        }
    }
    cout << ans;
    return 0;
}

Compilation message (stderr)

knapsack.cpp: In function 'int main()':
knapsack.cpp:26:22: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<std::pair<long long int, long long int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   26 |                 if(y >= haha[i].size()) {
      |                    ~~^~~~~~~~~~~~~~~~~
knapsack.cpp:36:28: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<std::pair<long long int, long long int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   36 |     for(long long i = 0; i < bruh.size(); i++) {
      |                          ~~^~~~~~~~~~~~~

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