結果

問題 No.2617 容量3のナップザック
コンテスト
ユーザー tnakao0123
提出日時 2024-06-17 18:49:51
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,815 bytes
コンパイル時間 834 ms
コンパイル使用メモリ 60,652 KB
最終ジャッジ日時 2025-02-21 23:07:18
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 17 WA * 23
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function ‘int main()’:
main.cpp:32:8: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   32 |   scanf("%d%d%d%d%d%d", &n, &k, &seed, &a, &b, &m);
      |   ~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

ソースコード

diff # 

/* -*- coding: utf-8 -*-
 *
 * 2617.cc:  No.2617 螳ｹ驥・縺ｮ繝翫ャ繝励じ繝・け - yukicoder
 */

#include<cstdio>
#include<vector>
#include<algorithm>
 
using namespace std;

/* constant */

const int MAX_N = 1000000;

/* typedef */

using ll = long long;
using vl = vector<ll>;

/* global variables */

int fs[MAX_N * 2];
vl wvs[4];

/* subroutines */

/* main */

int main() {
  int n, k, seed, a, b, m;
  scanf("%d%d%d%d%d%d", &n, &k, &seed, &a, &b, &m);

  for (int i = 0, f = seed; i < n * 2; i++) {
    fs[i] = f;
    f = ((ll)a * f + b) % m;
  }
  for (int i = 0; i < n; i++) {
    int w = fs[i] % 3 + 1;
    ll v = (ll)fs[n + i] * w;
    //printf(" %d,%lld", w, v);
    wvs[w].push_back(v);
  }
  //putchar('\n');
  for (int w = 1; w <= 3; w++) sort(wvs[w].begin(), wvs[w].end());

  ll sum = 0;
  int j1 = wvs[1].size() - 1;
  int j2 = wvs[2].size() - 1;
  int j3 = wvs[3].size() - 1;

  for (int i = 0; i < k; i++) {
    ll maxv = 0;
    int op = -1;
    if (j3 >= 0 && maxv < wvs[3][j3])
      maxv = wvs[3][j3], op = 0;
    if (j2 >= 0 && j1 >= 0 && maxv < wvs[2][j2] + wvs[1][j1])
      maxv = wvs[2][j2] + wvs[1][j1], op = 1;
    if (j1 >= 2 && maxv < wvs[1][j1] + wvs[1][j1 - 1] + wvs[1][j1 - 2])
      maxv = wvs[1][j1] + wvs[1][j1 - 1] + wvs[1][j1 - 2], op = 2;
    if (j2 >= 0 && maxv < wvs[2][j2])
      maxv = wvs[2][j2], op = 3;
    if (j1 >= 1 && maxv < wvs[1][j1] + wvs[1][j1 - 1])
      maxv = wvs[1][j1] + wvs[1][j1 - 1], op = 4;
    if (j1 >= 0 && maxv < wvs[1][j1])
      maxv = wvs[1][j1], op = 5;
    if (op < 0) break;

    sum += maxv;
    switch (op) {
    case 0: j3--; break;
    case 1: j2--, j1--; break;
    case 2: j1 -= 3; break;
    case 3: j2--; break;
    case 4: j1 -= 2; break;
    case 5: j1--; break;
    }
  }

  printf("%lld\n", sum);
  
  return 0;
}
0