結果

問題 No.626 Randomized 01 Knapsack
コンテスト
ユーザー pekempey
提出日時 2017-12-15 21:54:34
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,691 bytes
コンパイル時間 756 ms
コンパイル使用メモリ 73,872 KB
実行使用メモリ 6,820 KB
最終ジャッジ日時 2024-12-15 18:55:07
合計ジャッジ時間 1,684 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 21 WA * 4
権限があれば一括ダウンロードができます

ソースコード

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