結果

問題 No.626 Randomized 01 Knapsack
ユーザー pekempeypekempey
提出日時 2017-12-15 21:54:34
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
WA  
実行時間 -
コード長 1,691 bytes
コンパイル時間 665 ms
コンパイル使用メモリ 75,064 KB
実行使用メモリ 6,944 KB
最終ジャッジ日時 2024-05-09 07:57:41
合計ジャッジ時間 1,611 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 2 ms
5,376 KB
testcase_02 AC 2 ms
5,376 KB
testcase_03 AC 2 ms
5,376 KB
testcase_04 AC 2 ms
5,376 KB
testcase_05 AC 2 ms
5,376 KB
testcase_06 WA -
testcase_07 AC 3 ms
5,376 KB
testcase_08 AC 3 ms
5,376 KB
testcase_09 AC 4 ms
5,376 KB
testcase_10 AC 3 ms
5,376 KB
testcase_11 AC 6 ms
5,376 KB
testcase_12 WA -
testcase_13 AC 7 ms
5,376 KB
testcase_14 AC 6 ms
5,376 KB
testcase_15 AC 7 ms
5,376 KB
testcase_16 AC 7 ms
5,376 KB
testcase_17 AC 7 ms
5,376 KB
testcase_18 AC 7 ms
5,376 KB
testcase_19 WA -
testcase_20 AC 8 ms
5,376 KB
testcase_21 WA -
testcase_22 AC 7 ms
5,376 KB
testcase_23 AC 8 ms
5,376 KB
testcase_24 AC 6 ms
5,376 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
#include <algorithm>

using namespace std;

const int N = 5000;
int n;
long long C;
long long V[N];
long long W[N];
int p[N];
long long sumV[N + 1];
long long sumW[N + 1];

long long ans;

double estimate(long long A, int k) {
  int ok = k;
  int ng = n;
  while (ng - ok > 1) {
    int mid = (ok + ng) / 2;
    if (sumW[mid + 1] - sumW[k] <= A) {
      ok = mid;
    } else {
      ng = mid;
    }
  }
  double ret = sumV[ok + 1] - sumV[k];
  A -= sumW[ok + 1] - sumW[k];
  if (ok + 1 < n) {
    ret += A * (double)V[ p[ok + 1] ] / W[ p[ok + 1] ];
  }

  return ret;
}

// ***************************........
// [--------greedy-----------]
//
// **********************.*...*.**....
// [--------core--------]
void dfs(long long w, long long v, int k) {
  if (w > C) return;
  ans = max(ans, v);

  for (int i = k; i < n; i++) {
    // You can estimate the upper bound by Rational Knapsack problem.
    if (v + estimate(C - w, i) < ans - 1e-8) break;
    dfs(w + W[ p[i] ], v + V[ p[i] ], i + 1);
  }
}

int main() {
  cin >> n >> C;

  for (int i = 0; i < n; i++) {
    cin >> V[i] >> W[i];
    p[i] = i;
  }

  sort(p, p + n, [&](int i, int j) {
   return (double)V[i] / W[i] > (double)V[j] / W[j]; 
  });

  for (int i = 0; i < n; i++) {
    sumV[i + 1] = sumV[i] + V[ p[i] ];
    sumW[i + 1] = sumW[i] + W[ p[i] ];
  }

  long long v = 0;
  long long w = 0;
  int k = 0;
  while (k < n) {
    v += V[ p[k] ];
    w += W[ p[k] ];
    k++;
    if (w > C) break;
  }

  for (int i = k - 1; i >= 0; i--) {
    w -= W[ p[i] ];
    v -= V[ p[i] ];
    if (v + estimate(C - w, i + 1) < ans - 1e-8) break;
    dfs(w, v, i + 1);
  }

  cout << ans << endl;
}
0