結果

問題 No.2617 容量3のナップザック
ユーザー 👑 emthrmemthrm
提出日時 2024-01-26 21:55:30
言語 C++23
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 417 ms / 2,000 ms
コード長 3,945 bytes
コンパイル時間 3,342 ms
コンパイル使用メモリ 272,200 KB
実行使用メモリ 42,116 KB
最終ジャッジ日時 2024-01-26 21:55:42
合計ジャッジ時間 11,066 ms
ジャッジサーバーID
（参考情報） 		judge15 / judge11
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,548 KB
testcase_01 AC 2 ms
6,548 KB
testcase_02 AC 2 ms
6,548 KB
testcase_03 AC 1 ms
6,548 KB
testcase_04 AC 2 ms
6,548 KB
testcase_05 AC 1 ms
6,548 KB
testcase_06 AC 2 ms
6,548 KB
testcase_07 AC 2 ms
6,548 KB
testcase_08 AC 2 ms
6,548 KB
testcase_09 AC 2 ms
6,548 KB
testcase_10 AC 2 ms
6,548 KB
testcase_11 AC 2 ms
6,548 KB
testcase_12 AC 268 ms
27,992 KB
testcase_13 AC 199 ms
32,028 KB
testcase_14 AC 265 ms
26,272 KB
testcase_15 AC 163 ms
20,416 KB
testcase_16 AC 85 ms
19,160 KB
testcase_17 AC 281 ms
27,464 KB
testcase_18 AC 258 ms
26,980 KB
testcase_19 AC 27 ms
6,548 KB
testcase_20 AC 189 ms
30,672 KB
testcase_21 AC 128 ms
15,156 KB
testcase_22 AC 222 ms
32,400 KB
testcase_23 AC 356 ms
37,524 KB
testcase_24 AC 9 ms
6,548 KB
testcase_25 AC 85 ms
22,592 KB
testcase_26 AC 91 ms
19,960 KB
testcase_27 AC 97 ms
13,280 KB
testcase_28 AC 11 ms
6,548 KB
testcase_29 AC 61 ms
14,004 KB
testcase_30 AC 223 ms
31,300 KB
testcase_31 AC 264 ms
23,772 KB
testcase_32 AC 330 ms
35,004 KB
testcase_33 AC 384 ms
35,460 KB
testcase_34 AC 412 ms
36,064 KB
testcase_35 AC 340 ms
34,364 KB
testcase_36 AC 166 ms
34,132 KB
testcase_37 AC 417 ms
36,064 KB
testcase_38 AC 237 ms
34,060 KB
testcase_39 AC 152 ms
33,588 KB
testcase_40 AC 381 ms
35,420 KB
testcase_41 AC 137 ms
42,116 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
#define FOR(i,m,n) for(int i=(m);i<(n);++i)
#define REP(i,n) FOR(i,0,n)
using ll = long long;
constexpr int INF = 0x3f3f3f3f;
constexpr long long LINF = 0x3f3f3f3f3f3f3f3fLL;
constexpr double EPS = 1e-8;
constexpr int MOD = 998244353;
// constexpr int MOD = 1000000007;
constexpr int DY4[]{1, 0, -1, 0}, DX4[]{0, -1, 0, 1};
constexpr int DY8[]{1, 1, 0, -1, -1, -1, 0, 1};
constexpr int DX8[]{0, -1, -1, -1, 0, 1, 1, 1};
template <typename T, typename U>
inline bool chmax(T& a, U b) { return a < b ? (a = b, true) : false; }
template <typename T, typename U>
inline bool chmin(T& a, U b) { return a > b ? (a = b, true) : false; }
struct IOSetup {
  IOSetup() {
    std::cin.tie(nullptr);
    std::ios_base::sync_with_stdio(false);
    std::cout << fixed << setprecision(20);
  }
} iosetup;

// https://ei1333.github.io/library/structure/others/priority-sum-structure.hpp
template< typename T, typename Compare = less< T >, typename RCompare = greater< T > >
struct PrioritySumStructure {

  size_t k;
  T sum;

  priority_queue< T, vector< T >, Compare > in, d_in;
  priority_queue< T, vector< T >, RCompare > out, d_out;

  PrioritySumStructure(int k) : k(k), sum(0) {}

  void modify() {
    while(in.size() - d_in.size() < k && !out.empty()) {
      auto p = out.top();
      out.pop();
      if(!d_out.empty() && p == d_out.top()) {
        d_out.pop();
      } else {
        sum += p;
        in.emplace(p);
      }
    }
    while(in.size() - d_in.size() > k) {
      auto p = in.top();
      in.pop();
      if(!d_in.empty() && p == d_in.top()) {
        d_in.pop();
      } else {
        sum -= p;
        out.emplace(p);
      }
    }
    while(!d_in.empty() && in.top() == d_in.top()) {
      in.pop();
      d_in.pop();
    }
  }

  T query() const {
    return sum;
  }

  void insert(T x) {
    in.emplace(x);
    sum += x;
    modify();
  }

  void erase(T x) {
    assert(size());
    if(!in.empty() && in.top() == x) {
      sum -= x;
      in.pop();
    } else if(!in.empty() && RCompare()(in.top(), x)) {
      sum -= x;
      d_in.emplace(x);
    } else {
      d_out.emplace(x);
    }
    modify();
  }

  void set_k(size_t kk) {
    k = kk;
    modify();
  }

  size_t get_k() const {
    return k;
  }

  size_t size() const {
    return in.size() + out.size() - d_in.size() - d_out.size();
  }
};

template< typename T >
using MaximumSum = PrioritySumStructure< T, greater< T >, less< T > >;

template< typename T >
using MinimumSum = PrioritySumStructure< T, less< T >, greater< T > >;

int main() {
  int n, k; cin >> n >> k;
  vector<int> w(n);
  vector<ll> v(n);
  {
    vector<ll> f(n * 2);
    int a, b, m; cin >> f.front() >> a >> b >> m;
    FOR(i, 1, n * 2) f[i] = (a * f[i - 1] + b) % m;
    REP(i, n) w[i] = f[i] % 3 + 1;
    REP(i, n) v[i] = w[i] * f[i + n];
  }
  array<vector<ll>, 4> item{};
  REP(i, n) item[w[i]].emplace_back(v[i]);
  FOR(i, 1, 3) ranges::sort(item[i], greater<ll>());
  ll ans = 0;
  REP(md, 3) {
    if (md > k || item[2].size() < md) break;
    ll cur = 0;
    REP(i, md) cur += item[2][i] + (i < item[1].size() ? item[1][i] : 0);
    vector<ll> tmp;
    tmp.reserve((item[1].size() + 2) / 3);
    for (int i = md; i < item[1].size(); i += 3) {
      tmp.emplace_back(0);
      REP(j, 3) {
        if (i + j < item[1].size()) tmp.back() += item[1][i + j];
      }
    }
    ranges::reverse(tmp);
    assert(ranges::is_sorted(tmp));
    MaximumSum<ll> ms(k - md);
    for (const ll v_i : tmp) ms.insert(v_i);
    for (const ll v_i : item[3]) ms.insert(v_i);
    for (int i = md; i <= k;) {
      ms.set_k(k - i);
      chmax(ans, cur + ms.query());
      if (i >= item[2].size()) break;
      REP(_, 3) {
        if (i < item[2].size()) cur += item[2][i];
        ++i;
      }
      if (!tmp.empty()) {
        cur += tmp.back();
        ms.erase(tmp.back());
        tmp.pop_back();
      }
    }
  }
  cout << ans << '\n';
  return 0;
}
0