Submission #1276609

#TimeUsernameProblemLanguageResultExecution timeMemory
1276609almazKnapsack (NOI18_knapsack)C++20
100 / 100
262 ms4964 KiB
#include <bits/stdc++.h>
using namespace std;

#define int long long
// #define endl '\n'
#define ff first
#define ss second
#define pb push_back
#define all(a) a.begin(), a.end()
#define rall(a) a.rbegin(), a.rend()
#define ar array

const int MOD = 1e9 + 7,INF = 1e18, N = 2e5 + 5;
/*
*/

void solve(){
	int s , n;
	cin >> s >> n;
	
	vector <ar <int, 3>> a(n);
	set <int> st;
	map <int, vector <pair<int,int>>> b;
	
	for(int i = 0;i < n;i++){
		int v , w , k;
		cin >> v >> w >> k;
		a[i][0] = v;
		a[i][1] = w;
		a[i][2] = k;
		if(w > s) continue;
		// b[w].pb({v , k});
		st.insert(w);
	}
	vector <int> dp(s + 1, 0);
	vector <int> use(s + 1);
	use[0] = 1;
	for(int w : st){
		vector <pair<int,int>> c;
		for(int i = 0;i < n;i++){
			if(a[i][1] == w){
				c.pb({a[i][0] , a[i][2]});
			}
		}
		sort(rall(c));
		
		vector <int> d;
		
		for(int i = 0;i < (int)c.size();i++){
			for(int j = 0;j < c[i].ss;j++){
				d.pb(c[i].ff);
				if((int)d.size() == s){
					break;
				}
			}
			if((int)d.size() == s){
				break;
			}
		}
		// cout<<d.size()<<endl;
		
		for(int k = s;k >= 0;k--){
			if(use[k]){
				// cout<<k<<endl;
				int j = 0;
				for(int i = w + k;i <= s;i += w){
					if(j == (int)d.size()) break;
					dp[i] = max(dp[i] , dp[i - w] + d[j]);
					j++;
					use[i] = 1;
				}
			}
		}
		// for(int i = 0;i <= s;i++){
			// cout<<dp[i]<<' ';
		// }
		// cout<<endl;
		
	}
	
	int ans = 0;

	
	for(int i = 0;i <= s;i++){
		// cout<<dp[i]<<' ';
		ans = max(ans , dp[i]);
	}
	// cout<<endl;
	
	cout<<ans<<endl;
	
}


signed main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    int ti = 1;
    // cin >> ti;
    while (ti--) {
		solve();
    }
}
#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...