結果

問題 No.626 Randomized 01 Knapsack
ユーザー Ryuhei MoriRyuhei Mori
提出日時 2017-12-18 21:49:02
言語 C++11
(gcc 11.4.0)
結果
AC  
実行時間 4 ms / 2,000 ms
コード長 3,529 bytes
コンパイル時間 753 ms
コンパイル使用メモリ 57,424 KB
実行使用メモリ 4,384 KB
最終ジャッジ日時 2023-09-08 10:08:08
合計ジャッジ時間 2,612 ms
ジャッジサーバーID
（参考情報） 		judge12 / judge13
このコードへのチャレンジ
（要ログイン）

テストケース

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

ソースコード

diff # 

#include <cstdio>
#include <vector>
#include <algorithm>

typedef long long int ll;

struct state {
  ll aw;
  ll av;
  state(ll aw=0, ll av=0): aw(aw), av(av) {}
};

struct item {
  ll w;
  ll v;
  double d;
  item(ll w, ll v): w(w), v(v), d((double) v/w) { }
};
 
int n;
ll w;

ll lower;
 
std::vector<item> is;
std::vector<state> dp[2];
 
void add(const std::vector<state> &s, const item &k, std::vector<state> &u, ll uw){
  auto i = s.cbegin();
  auto j = s.cbegin();
  while(i != s.cend() && j != s.cend()){
    if(i->aw > uw){ i = s.cend(); break; }
    ll jav = j->av + k.v;
    ll jaw = j->aw + k.w;
    if(jaw > uw){ j = s.cend(); break; }
    if(i->aw <= jaw && i->av >= jav){ ++j; }
    else if(i->aw >= jaw && i->av <= jav){ ++i; }
    else if(i->aw < jaw){ u.push_back(*i++); }
    else { u.emplace_back(state(jaw, jav)); ++j;}
  }
  while(i != s.cend()){
    if(i->aw > uw) return;
    u.push_back(*i++);
  }
  while(j != s.cend()){
    ll jav = j->av + k.v;
    ll jaw = j->aw + k.w;
    if(jaw > uw) return;
    u.emplace_back(state(jaw, jav));
    ++j;
  }
}

void remove(const std::vector<state> &s, const item &k, std::vector<state> &u, ll uw){
  auto i = s.cbegin();
  auto j = s.cbegin();
  while(i != s.cend() && j != s.cend()){
    if(i->aw > uw){ i = s.cend(); break; }
    ll jav = j->av - k.v;
    ll jaw = j->aw - k.w;
    if(jaw > uw){ j = s.cend(); break; }
    if(i->aw <= jaw && i->av >= jav){ ++j; }
    else if(i->aw >= jaw && i->av <= jav){ ++i; }
    else if(i->aw < jaw){ u.push_back(*i++); }
    else { u.emplace_back(state(jaw, jav)); ++j;}
  }
  while(i != s.cend()){
    if(i->aw > uw) return;
    u.push_back(*i++);
  }
  while(j != s.cend()){
    ll jav = j->av - k.v;
    ll jaw = j->aw - k.w;
    if(jaw > uw) return;
    u.emplace_back(state(jaw, jav));
    ++j;
  }
}

ll upper;

 
inline void shrink(const std::vector<state> &s, double d0, double d1, std::vector<state> &u){
  for(auto x: s){
    ll t = (x.aw > w) ? x.av - (x.aw - w) * d0 : x.av + (w - x.aw) * d1;
    if(t >= lower) u.push_back(x);
    if(t > upper) upper = t;
  }
}

 
int main(){
  scanf("%d%lld", &n, &w);
  is.reserve(5000);
  dp[0].reserve(300);
  dp[1].reserve(300);

  is.emplace_back(item(1,0));

  for(int i=0;i<n;i++){
    ll ww, vv;
    scanf("%lld%lld",&vv,&ww);
    if(ww <= w) is.emplace_back(item(ww,vv));
  }
  n = is.size()-1;

  std::sort(is.begin()+1, is.end(), [](const item &x, const item &y){ return  x.d > y.d; });
  is.emplace_back(item(1,0));


  ll aw = 0;
  ll av = 0;
  int ii;
  for(ii=1;ii<=n;ii++){
    const item &x = is[ii];
    if(aw + x.w > w) break;
    aw += x.w;
    av += x.v;
  }

  dp[0].emplace_back(state(aw, av));

  lower = av;
  ll sumw = w+aw;

  for(int i=ii-1, j=ii; i>0||j<=n; ){
    if(i>0){
      sumw -= is[i].w;
      remove(dp[0], is[i], dp[1], sumw);
      dp[0].resize(0);
      lower = (std::upper_bound(dp[1].cbegin(), dp[1].cend(), w, [](ll w, const state &x){ return x.aw > w; })-1)->av;
      shrink(dp[1], is[i-1].d, is[j].d, dp[0]);
      dp[1].resize(0);
      --i;
    }

    if(j<=n){
      add(dp[0], is[j], dp[1], sumw);
      dp[0].resize(0);
      lower = (std::upper_bound(dp[1].cbegin(), dp[1].cend(), w, [](ll w, const state &x){ return x.aw > w; })-1)->av;
      shrink(dp[1], is[i].d, is[j+1].d, dp[0]);
      dp[1].resize(0);
      ++j;
    }
    if(lower == upper) break;
  }

  lower = (std::upper_bound(dp[0].cbegin(), dp[0].cend(), w, [](ll w, const state &x){ return x.aw > w; })-1)->av;
  printf("%lld\n", lower);
 
  return 0;
}
0