제출 #849089

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
849089		AbdalrhmanMohammad	Knapsack (NOI18_knapsack)	C++14	73 / 100	816 ms	262144 KiB

knapsack

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

#define FAST                                                                   \
  ios_base::sync_with_stdio(false);                                            \
  cin.tie(nullptr);                                                            \
  cout.tie(nullptr);

#define ll long long
#define yn(tf) cout << (tf ? "YES\n" : "NO\n")
#define pr(v) v.begin(), v.end()
#define Endl endl
#define ld long double

const ld PI = 3.14159265359;
const ll MOD = 998244353;
const ll N = 1e5 + 7;

int s, n;
ll V[N], W[N], K[N];
vector<vector<ll>> v;
vector<vector<ll>> mem;
ll fun(int i, int rem) {
  if (rem < 0)
    return -1e15;
  if (i == v.size() || rem == 0)
    return 0;
  ll &res = mem[i][rem];
  if (res != -1)
    return res;
  res = fun(i + 1, rem);
  for (ll j = 1; j <= min(2000LL, v[i][2]); j++)
    res = max(res, fun(i + 1, rem - j * v[i][1]) + v[i][0] * j);
  return res;
}
int main() {
  FAST

  cin >> s >> n;
  map<int, vector<pair<int, int>>> mp;
  for (int i = 0; i < n; i++) {
    cin >> V[i] >> W[i] >> K[i];
    mp[W[i]].push_back({V[i], K[i]});
  }
  for (pair<int, vector<pair<int, int>>> p : mp) {
    vector<pair<int, int>> &tmp = p.second;
    int wei = p.first;
    sort(pr(tmp));
    reverse(pr(tmp));
    int sum = 0;
    for (int i = 0; i < tmp.size(); i++) {
      if (i > 0) {
        if (v.back()[0] == tmp[i].first) {
          v.back()[2] += max(0, min(tmp[i].second, 2000 - sum));
          sum += tmp[i].second;
        } else {
          v.push_back(
              {tmp[i].first, wei, max(0, min(tmp[i].second, 2000 - sum))});
          sum += tmp[i].second;
        }
      } else {
        v.push_back(
            {tmp[i].first, wei, max(0, min(tmp[i].second, 2000 - sum))});
        sum += tmp[i].second;
      }
    }
  }
  mem = vector<vector<ll>>(v.size() + 1, vector<ll>(s + 1));
  for (int i = 0; i <= v.size(); i++)
    for (int j = 0; j <= s; j++)
      mem[i][j] = -1;
  cout << fun(0, s);
  return 0;
}

/*

*/

// BEFORE coding are you sure you understood the statement correctly?
// PLEASE do not forget to read the sample explanation carefully.
// WATCH out for overflows & RTs in general.
// TEST your idea or code on the corner cases.
// ANALYZE each idea you have thoroughly.

컴파일 시 표준 에러 (stderr) 메시지

knapsack.cpp: In function 'long long int fun(int, int)':
knapsack.cpp:27:9: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::vector<long long int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   27 |   if (i == v.size() || rem == 0)
      |       ~~^~~~~~~~~~~
knapsack.cpp: In function 'int main()':
knapsack.cpp:52:23: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<int, int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   52 |     for (int i = 0; i < tmp.size(); i++) {
      |                     ~~^~~~~~~~~~~~
knapsack.cpp:70:21: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::vector<long long int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   70 |   for (int i = 0; i <= v.size(); i++)
      |                   ~~^~~~~~~~~~~

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