SRM撃墜大好きっ子に贈る：二分探索問題の撃墜 〜慎重に撃墜ケースを検討すべき一例〜

SRM582 Easy(250) "SpaceWarDiv1"

問題意訳

魔法少女（複数）と敵（複数）がいる。
魔法少女は自分と同等以下の敵を倒せるが、1人倒すたびにソウルジェムが濁っていく。
敵は全て倒したいが、ソウルジェムの濁りが特定の少女に集中し魔女化してしまうのを避けるため、戦闘回数を分散したい。
最大限にうまくやったとき、ソウルジェムの濁りはどこまで抑えられるか。

※雰囲気はそんな感じだけど正確なところは原文でちゃんと読んでください。writerは準急さん(semiexp)です。

入力

{魔法少女の強さ}, {敵の強さ}, {それぞれの強さの敵が何人いるか}

出力

一番多く敵を倒した魔法少女のソウルジェムの濁り

解き方

二分探索でどうよ

submitしたWAコード

(要ログイン)

二分探索が間違っている事に気づいたのに（あれ、これでも通るんだーとか思っちゃって）再提出せず

#include <iostream>
#include <cstdio>
#include <cmath>
#include <algorithm>
#include <string>
#include <vector>

using namespace std;
typedef long long ll;
#define sz(a)  int((a).size())
#define pb  push_back
#define rep(var,n)  for(int var=0;var<(n);var++)
#define all(c)  (c).begin(),(c).end()
#define tr(c,i)  for(typeof((c).begin()) i=(c).begin(); i!=(c).end(); i++)
 
vector<int> mgs;
int mgN, eN;
vector<pair<int,ll> > es;
 
class SpaceWarDiv1 {
  bool po(ll x) {
    vector<ll> res(mgN, x);
    int ix = 0;
    tr(es,it) {
      int e_s = it->first;
      ll e_c = it->second;
      while (e_c > 0) {
        if (ix == mgN) return false;
        if (mgs[ix] < e_s) return false;
 
        if (res[ix] > e_c) {
          res[ix] -= e_c;
          e_c = 0;
        } else if (res[ix] == e_c) {
          res[ix++] = e_c = 0;
        } else {
          e_c -= res[ix];
          res[ix++] = 0;
        }
      }
    }
    return true;
  }
 public:
  long long minimalFatigue(vector<int> magicalGirlStrength, vector<int> enemyStrength, vector<long long> enemyCount) {
    int maxES = *max_element(all(enemyStrength)), 
        maxGS = *max_element(all(magicalGirlStrength));
    if (maxES > maxGS) return -1LL;
 
    mgN = sz(magicalGirlStrength);
    mgs.assign(all(magicalGirlStrength));
    sort(all(mgs)); reverse(all(mgs));
 
    eN = sz(enemyStrength);
    es.resize(eN);
    rep(i,eN){
      es[i] = make_pair(enemyStrength[i], enemyCount[i]);
    }
    sort(all(es)); reverse(all(es));
 
    ll lo = 1LL, hi = LONG_LONG_MAX;
    while (true) {
      // lo:x hi:o
      if (lo+1 == hi) return hi;
      ll mi = (lo + hi)/2;
      if (po(mi)) hi = mi;
      else        lo = mi;
    }
  }
};

撃墜ケース

3人の撃墜ミスを誘いました。Room1ということもあり高レーティングです。

• ltaravilse (黄2083) : {1}{1}{1}
• hjvfy (赤2562) : {1}{1}{1}
• marcina007 (赤3015) : {1}{1}{1}

同室だったPetr様がこのコードを開いて手を出して来なかったのは流石と思いました。

入力 {1}{1}{1} に対する挙動

賢明な読者の皆さんは、このコードがどういう挙動をするかお分かりですよね？

判定関数 po() の値が lo で false, hi でtrueであることを想定して、区間 [1, LONG_LONG_MAX=0x7FFFFFFFFFFFFFFF] で二分探索をしています。はい。lower boundは0であるべきです。0にすればACです。

// あと、lo+hi > LONG_LONG_MAX なことがあり得るケースでは lo + (hi - lo)/2 が良いですね。

無限ループになる例

{5, 6, 10, 1, 1, 2, 2, 1, 10, 4, 9, 5, 6, 7, 3}, {6}, {20LL}*2

教訓

二分探索のupper/lower boundsがおかしいコードを撃墜する際は要注意。

区間 [1, LONG_LONG_MAX] で二分探索したらいけない理由は、1のケースを拾いそこねるからではなく、(1+LONG_LONG_MAX)/2 は負数になるので負の領域を含めてぐるんぐるんと探索しに行くこと、しかもそれでもほとんどのテストケースが通るので気づきにくいが、場合によっては無限ループに陥ること。