Submission #1016874

#	Time	Username	Problem	Language	Result	Execution time	Memory
1016874		sexo69696969	Knapsack (NOI18_knapsack)	C++14	100 / 100	57 ms	4444 KiB

knapsack

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

#ifdef local
  #include "debug.hpp"
  #define pr(...) debug(#__VA_ARGS__, __VA_ARGS__)
  #define prs(...) debug_nameless(__VA_ARGS__)
#else
  #define pr(...) 69
  #define prs(...) 69
#endif

#define endl '\n'

const int inf = 1e9;
const long long binf = 1e18;

struct Item{
  int p, w, f;
};

void solve(int tc){
  int s, n;
  cin >> s >> n;
  vector <vector<Item>> items(s+1);
  for(int i = 0; i < n; i++){
    int p, w, f;
    cin >> p >> w >> f;
    items[w].push_back({p, w, f});
  } 
  auto cmp = [&](Item a, Item b){
    assert(a.w == b.w);
    return a.p > b.p;
  };
  for(int i = 1; i <= s; i++){
    sort(items[i].begin(), items[i].end(), cmp);
  }
  vector <pair<int,int>> arr;
  for(int i = 1; i <= s; i++){
    int total = 0;
    for(int pos = 0; pos < items[i].size(); pos++){
      for(int j = 0; j < items[i][pos].f && total < s; j++){
        if(total + items[i][pos].w <= s){
          arr.push_back({items[i][pos].w, items[i][pos].p});  
        }
        total += items[i][pos].w;
      }
      if(pos > 0){
        assert(items[i][pos-1].p >= items[i][pos].p && items[i][pos-1].w == items[i][pos].w);
      }
    }
  }
  vector <long long> dp(s+1, 0);
  for(int i = 1; i <= arr.size(); i++){
    for(int j = s; j >= 0; j--){
      if(j - arr[i-1].first >= 0){
        dp[j] = max(dp[j], dp[j-arr[i-1].first] + arr[i-1].second);
      }
    }
  }
  cout << dp[s] << endl;
}  

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

  int tc = -1;
  tc = 1;
  if(tc != 1) cin >> tc;

  for(int t = 0; t < tc; t++){
    pr(t); prs(string(50, '-'));
    solve(t);
    prs(string(50, '-') + "\n");
  }

  return 0;
}

Compilation message (stderr)

knapsack.cpp: In function 'void solve(int)':
knapsack.cpp:41:26: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<Item>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   41 |     for(int pos = 0; pos < items[i].size(); pos++){
      |                      ~~~~^~~~~~~~~~~~~~~~~
knapsack.cpp:54:20: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<int, int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   54 |   for(int i = 1; i <= arr.size(); i++){
      |                  ~~^~~~~~~~~~~~~
knapsack.cpp: In function 'int main()':
knapsack.cpp:9:19: warning: statement has no effect [-Wunused-value]
    9 |   #define pr(...) 69
      |                   ^~
knapsack.cpp:72:5: note: in expansion of macro 'pr'
   72 |     pr(t); prs(string(50, '-'));
      |     ^~
knapsack.cpp:10:20: warning: statement has no effect [-Wunused-value]
   10 |   #define prs(...) 69
      |                    ^~
knapsack.cpp:72:12: note: in expansion of macro 'prs'
   72 |     pr(t); prs(string(50, '-'));
      |            ^~~
knapsack.cpp:10:20: warning: statement has no effect [-Wunused-value]
   10 |   #define prs(...) 69
      |                    ^~
knapsack.cpp:74:5: note: in expansion of macro 'prs'
   74 |     prs(string(50, '-') + "\n");
      |     ^~~

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