結果

問題 No.626 Randomized 01 Knapsack
コンテスト
ユーザー Ryuhei Mori
提出日時 2017-12-18 21:49:02
言語 C++11（廃止可能性あり）
(gcc 13.3.0)
結果
AC  
実行時間 4 ms / 2,000 ms
コード長 3,529 bytes
コンパイル時間 932 ms
コンパイル使用メモリ 52,604 KB
実行使用メモリ 6,944 KB
最終ジャッジ日時 2024-06-26 03:29:28
合計ジャッジ時間 1,629 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 25
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function ‘int main()’:
main.cpp:93:8: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   93 |   scanf("%d%lld", &n, &w);
      |   ~~~~~^~~~~~~~~~~~~~~~~~
main.cpp:102:10: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
  102 |     scanf("%lld%lld",&vv,&ww);
      |     ~~~~~^~~~~~~~~~~~~~~~~~~~

ソースコード

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