Submission #1272328

#TimeUsernameProblemLanguageResultExecution timeMemory
1272328sm.22Knapsack (NOI18_knapsack)C++20
0 / 100
13 ms16260 KiB
#include<bits/stdc++.h>
#pragma GCC optimize("Ofast")
using namespace std;
 
#define watch(x)        cerr << "\n" << (#x) << " is " << (x) << "\n";
#define int             long long 
#define ffor(i, a, b)   for(int i=a;i<b;i++)
#define rfor(i, a, b)   for(int i=a-1;i>=b;i--)
#define all(v)          v.begin(), v.end()
#define rall(v)         v.rbegin(), v.rend()
#define pb              push_back
#define fi              first
#define se              second
#define pii             pair<int, int>
#define vi              vector<int> 
#define vvi             vector<vi> 
#define show(a)         for(auto e : a) {cout<<e<<" ";} cout<<"\n"; 
#define pr(a, b, c)     ffor(i, b, c) {cout<<a[i]<<" ";} cout<<"\n";
#define cerr            if(false)cerr
mt19937 rng(chrono::steady_clock::now().time_since_epoch().count());

const int N = 1e3 + 10; 
int Mod = 1e9 + 7, INF = 1e18, BASE = 7, Block = 555;

vvi z[N]; 
        
void solve() {

        int n, s; cin>>s>>n; 
        
        ffor(i, 0, n) {
                int v, w, k; cin>>v>>w>>k; 
                z[w].pb({v, k});
        }

        ffor(i, 0, N) {
                sort(all(z[i])); reverse(all(z[i])); 
                // if(z[i].size()) {
                //         cout<<i<<" : "; 
                //         for(auto zz : z[i]) {
                //                 show(zz); 
                //         }
                // }
        }


        int dp[N][s + 1];
        memset(dp, 0, sizeof(dp)); 
        
        ffor(i, 1, N) {
                // go through all weights = i 
                ffor(c, 1, s + 1) {
                        
                        int cur = c - i, k = 0, used = 1, amt = 0; 
                        dp[i][c] = dp[i - 1][c]; 


                        while(cur >= 0 && k < (int)z[i].size()) {
                                amt += z[i][k][0];
                                dp[i][c] = max(dp[i][c], dp[i - 1][cur] + amt); 
                                cur -= i; 
                                used++; 
                                if(z[i][k][1] < used) {
                                        k++; used = 1; 
                                }
                        }
                }
        }
        
        int ans = 0; 
        ffor(i, 1, s + 1) ans = max(ans, dp[N - 1][i]); 
        cout<<ans<<"\n"; 
}

signed main() { 
 
        ios_base::sync_with_stdio(false); 
        cin.tie(NULL); 

        int t = 1; //cin>>t; 
 
        while(t--) solve();
 
        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...