Problem p03019 (C++)

Buggy submission (status: Runtime Error):
///////////////////////////////////////////////////////////////////////////////

#include <bits/stdc++.h>
#include <sys/time.h>
#include <time.h>
#include <unistd.h>

using namespace std;

///////////////////////////////////////////////////////////////////////////////

#define DEBUG 0

#define pb push_back
#define V vector
#define M unordered_map
#define S static

#define rep(i, n) for (ll i = 0LL; i < n; ++i)
#define srep(i, s, n) for (ll i = s; i < n; ++i)
#define rrep(i, n) for (ll i = n - 1LL; i >= 0LL; --i)
#define ALL(a) (a).begin(), (a).end()
#define RALL(a) (a).rbegin(), (a).rend()
#define CIN(x)                                                                 \
  do {                                                                         \
    assert(!cin.eof());                                                        \
    cin >> x;                                                                  \
    assert(!cin.fail());                                                       \
  } while (0);
#if DEBUG
#define debug_print(...) _debug_print(__VA_ARGS__)
#define debug_printf(...) printf(__VA_ARGS__)
#define debug_print_time _debug_print_time
#else // DEBUG
#define debug_print(...)
#define debug_printf(...)
#define debug_print_time
#endif // DEBUG

typedef long long ll;
typedef unsigned long long ull;
typedef tuple<ll, ll> t2;
typedef tuple<ll, ll, ll> t3;
typedef tuple<ll, ll, ll, ll> t4;
typedef tuple<ll, ll, ll, ll, ll> t5;
template <typename T>
using priority_queue_incr = priority_queue<T, V<T>, greater<T>>;

///////////////////////////////////////////////////////////////////////////////

template <typename TT, typename T> T get_m(M<TT, T> &m, TT k, T default_value) {
  if (m.find(k) == m.end())
    return m[k] = default_value;
  return m[k];
}

template <typename TT, typename T> void incr_m(M<TT, T> &m, TT k) {
  if (m.find(k) == m.end())
    m[k] = 0;
  m[k]++;
}

template <typename TT, typename T> void incr_m(map<TT, T> &m, TT k) {
  if (m.find(k) == m.end())
    m[k] = 0;
  m[k]++;
}

void llin(ll &a) { CIN(a); }

void llinl1(V<ll> &v, ll count) {
  for (ll i = 0LL; i < count; ++i) {
    ll a;
    CIN(a);
    v.push_back(a);
  }
}

void llinl2(V<t2> &v, ll count) {
  for (ll i = 0LL; i < count; ++i) {
    ll a, b;
    CIN(a >> b);
    v.push_back(t2(a, b));
  }
}

void llinl3(V<t3> &v, ll count) {
  for (ll i = 0LL; i < count; ++i) {
    ll a, b, c;
    CIN(a >> b >> c);
    v.push_back(t3(a, b, c));
  }
}

void llinl4(V<t4> &v, ll count) {
  for (ll i = 0LL; i < count; ++i) {
    ll a, b, c, d;
    CIN(a >> b >> c >> d);
    v.push_back(t4(a, b, c, d));
  }
}

void llina(V<ll> &v, ll count) { llinl1(v, count); }

template <typename T> void sort(V<T> &v) { sort(v.begin(), v.end()); }

template <typename T> void sort_reverse(V<T> &v) {
  sort(v.begin(), v.end(), greater<T>());
}

template <typename T> void _debug_print(T x) { cout << x << endl; }

void _debug_print(const t2 &x) {
  ll x1 = get<0>(x);
  ll x2 = get<1>(x);
  cout << "-- " << x1 << " -- " << x2 << endl;
}

void _debug_print(const t3 &x) {
  ll x1 = get<0>(x);
  ll x2 = get<1>(x);
  ll x3 = get<2>(x);
  cout << "-- " << x1 << " -- " << x2 << " -- " << x3 << endl;
}

void _debug_print(const t4 &x) {
  ll x1 = get<0>(x);
  ll x2 = get<1>(x);
  ll x3 = get<2>(x);
  ll x4 = get<3>(x);
  cout << "-- " << x1 << " -- " << x2 << " -- " << x3 << " -- " << x4 << endl;
}

template <typename T> void _debug_print(T xarray[], ll n) {
  rep(i, n) _debug_print(xarray[i]);
}

template <typename T> void _debug_print(const V<T> &xlist) {
  for (auto x : xlist) {
    cout << "-- ";
    _debug_print(x);
  }
}

template <typename T> void _debug_print(const set<T> &xset) {
  for (auto x : xset) {
    cout << "-- ";
    _debug_print(x);
  }
}

template <typename TT, typename T> void _debug_print(const M<TT, T> &xlist) {
  for (auto x : xlist) {
    TT k = x.first;
    T v = x.second;
    cout << "====" << endl;
    cout << "K=";
    _debug_print(k);
    cout << "V=";
    _debug_print(v);
  }
}

template <typename TT, typename T> void _debug_print(const map<TT, T> &xlist) {
  for (auto x : xlist) {
    TT k = x.first;
    T v = x.second;
    cout << "====" << endl;
    cout << "K=";
    _debug_print(k);
    cout << "V=";
    _debug_print(v);
  }
}

void _debug_print_time(const char *prefix) {
  struct timeval tv;
  gettimeofday(&tv, NULL);
#if DEBUG
  struct tm *tm = localtime(&tv.tv_sec);
#endif
  debug_printf("-- %s %02d:%02d:%02d.%06ld\n", prefix, tm->tm_hour, tm->tm_min,
               tm->tm_sec, tv.tv_usec);
}

///////////////////////////////////////////////////////////////////////////////

void _main();
int main() {
  cout << setprecision(12);
#if !DEBUG
  ios::sync_with_stdio(false);
  cin.tie(0);
#endif

  _main();
  return 0;
}

void _main() {
  ll n;
  llin(n);
  ll max_a;
  llin(max_a);
  V<t3> blulist;
  llinl3(blulist, n);

  bool allmax = true;
  for (auto blu : blulist) {
    ll b = get<0>(blu);
    if (b < max_a)
      allmax = false;
  }
  if (allmax) {
    cout << max_a * n << endl;
    return;
  }

  ll aim = 0;
  for (auto blu : blulist) {
    ll b = get<0>(blu);
    ll left = get<1>(blu);
    ll right = get<2>(blu);
    aim += b * left;
  }
  debug_printf("---- aim = %lld\n", aim);

  V<t4> incr_blulist;
  rep(i, n) {
    ll b = get<0>(blulist[i]);
    ll left = get<1>(blulist[i]);
    ll right = get<2>(blulist[i]);
    incr_blulist.pb(t4(b * left + (max_a - b) * right, b, left, right));
  }
  sort_reverse(incr_blulist);

  ll base = 0;
  ll end = 0;

  for (auto item : incr_blulist) {
    ll incr = get<0>(item);
    ll b = get<1>(item);
    ll left = get<2>(item);
    ll right = get<3>(item);
    debug_printf("--- base=%lld incr=%lld b=%lld left=%lld right=%lld\n", base,
                 incr, b, left, right);

    if (base + incr <= aim) {
      base += incr;
      end++;
    } else
      break;
  }

  ll min_append = max_a;

  rep(i, n) {
    ll lbase = base;
    if (i < end) {
      lbase -= get<0>(incr_blulist[i]);
      lbase += get<0>(incr_blulist[end]);
    }
    ll incr = get<0>(incr_blulist[i]);
    ll b = get<1>(incr_blulist[i]);
    ll left = get<2>(incr_blulist[i]);
    ll right = get<3>(incr_blulist[i]);

    ll laim = aim - lbase;
    assert(laim > 0);

    ll floor = 0;
    ll ceil = max_a;
    ll nowpos = (floor + ceil) / 2LL;
    ll append = max_a;
    for (; floor <= ceil; nowpos = (floor + ceil) / 2LL) {
      ll lincr = left * min(nowpos, b);
      lincr += right * max(0LL, nowpos - b);
      if (lincr < laim) {
        floor = nowpos + 1LL;
      } else {
        ceil = nowpos - 1LL;
        append = min(append, nowpos);
      }
    }

    min_append = min(min_append, append);
  }

  debug_printf("---- ans\n");
  cout << end * max_a + min_append << endl;
}

///////////////////////////////////////////////////////////////////////////////


Fixed submission:
///////////////////////////////////////////////////////////////////////////////

#include <bits/stdc++.h>
#include <sys/time.h>
#include <time.h>
#include <unistd.h>

using namespace std;

///////////////////////////////////////////////////////////////////////////////

#define DEBUG 0

#define pb push_back
#define V vector
#define M unordered_map
#define S static

#define rep(i, n) for (ll i = 0LL; i < n; ++i)
#define srep(i, s, n) for (ll i = s; i < n; ++i)
#define rrep(i, n) for (ll i = n - 1LL; i >= 0LL; --i)
#define ALL(a) (a).begin(), (a).end()
#define RALL(a) (a).rbegin(), (a).rend()
#define CIN(x)                                                                 \
  do {                                                                         \
    assert(!cin.eof());                                                        \
    cin >> x;                                                                  \
    assert(!cin.fail());                                                       \
  } while (0);
#if DEBUG
#define debug_print(...) _debug_print(__VA_ARGS__)
#define debug_printf(...) printf(__VA_ARGS__)
#define debug_print_time _debug_print_time
#else // DEBUG
#define debug_print(...)
#define debug_printf(...)
#define debug_print_time
#endif // DEBUG

typedef long long ll;
typedef unsigned long long ull;
typedef tuple<ll, ll> t2;
typedef tuple<ll, ll, ll> t3;
typedef tuple<ll, ll, ll, ll> t4;
typedef tuple<ll, ll, ll, ll, ll> t5;
template <typename T>
using priority_queue_incr = priority_queue<T, V<T>, greater<T>>;

///////////////////////////////////////////////////////////////////////////////

template <typename TT, typename T> T get_m(M<TT, T> &m, TT k, T default_value) {
  if (m.find(k) == m.end())
    return m[k] = default_value;
  return m[k];
}

template <typename TT, typename T> void incr_m(M<TT, T> &m, TT k) {
  if (m.find(k) == m.end())
    m[k] = 0;
  m[k]++;
}

template <typename TT, typename T> void incr_m(map<TT, T> &m, TT k) {
  if (m.find(k) == m.end())
    m[k] = 0;
  m[k]++;
}

void llin(ll &a) { CIN(a); }

void llinl1(V<ll> &v, ll count) {
  for (ll i = 0LL; i < count; ++i) {
    ll a;
    CIN(a);
    v.push_back(a);
  }
}

void llinl2(V<t2> &v, ll count) {
  for (ll i = 0LL; i < count; ++i) {
    ll a, b;
    CIN(a >> b);
    v.push_back(t2(a, b));
  }
}

void llinl3(V<t3> &v, ll count) {
  for (ll i = 0LL; i < count; ++i) {
    ll a, b, c;
    CIN(a >> b >> c);
    v.push_back(t3(a, b, c));
  }
}

void llinl4(V<t4> &v, ll count) {
  for (ll i = 0LL; i < count; ++i) {
    ll a, b, c, d;
    CIN(a >> b >> c >> d);
    v.push_back(t4(a, b, c, d));
  }
}

void llina(V<ll> &v, ll count) { llinl1(v, count); }

template <typename T> void sort(V<T> &v) { sort(v.begin(), v.end()); }

template <typename T> void sort_reverse(V<T> &v) {
  sort(v.begin(), v.end(), greater<T>());
}

template <typename T> void _debug_print(T x) { cout << x << endl; }

void _debug_print(const t2 &x) {
  ll x1 = get<0>(x);
  ll x2 = get<1>(x);
  cout << "-- " << x1 << " -- " << x2 << endl;
}

void _debug_print(const t3 &x) {
  ll x1 = get<0>(x);
  ll x2 = get<1>(x);
  ll x3 = get<2>(x);
  cout << "-- " << x1 << " -- " << x2 << " -- " << x3 << endl;
}

void _debug_print(const t4 &x) {
  ll x1 = get<0>(x);
  ll x2 = get<1>(x);
  ll x3 = get<2>(x);
  ll x4 = get<3>(x);
  cout << "-- " << x1 << " -- " << x2 << " -- " << x3 << " -- " << x4 << endl;
}

template <typename T> void _debug_print(T xarray[], ll n) {
  rep(i, n) _debug_print(xarray[i]);
}

template <typename T> void _debug_print(const V<T> &xlist) {
  for (auto x : xlist) {
    cout << "-- ";
    _debug_print(x);
  }
}

template <typename T> void _debug_print(const set<T> &xset) {
  for (auto x : xset) {
    cout << "-- ";
    _debug_print(x);
  }
}

template <typename TT, typename T> void _debug_print(const M<TT, T> &xlist) {
  for (auto x : xlist) {
    TT k = x.first;
    T v = x.second;
    cout << "====" << endl;
    cout << "K=";
    _debug_print(k);
    cout << "V=";
    _debug_print(v);
  }
}

template <typename TT, typename T> void _debug_print(const map<TT, T> &xlist) {
  for (auto x : xlist) {
    TT k = x.first;
    T v = x.second;
    cout << "====" << endl;
    cout << "K=";
    _debug_print(k);
    cout << "V=";
    _debug_print(v);
  }
}

void _debug_print_time(const char *prefix) {
  struct timeval tv;
  gettimeofday(&tv, NULL);
#if DEBUG
  struct tm *tm = localtime(&tv.tv_sec);
#endif
  debug_printf("-- %s %02d:%02d:%02d.%06ld\n", prefix, tm->tm_hour, tm->tm_min,
               tm->tm_sec, tv.tv_usec);
}

///////////////////////////////////////////////////////////////////////////////

void _main();
int main() {
  cout << setprecision(12);
#if !DEBUG
  ios::sync_with_stdio(false);
  cin.tie(0);
#endif

  _main();
  return 0;
}

void _main() {
  ll n;
  llin(n);
  ll max_a;
  llin(max_a);
  V<t3> blulist;
  llinl3(blulist, n);

  bool allmax = true;
  for (auto blu : blulist) {
    ll b = get<0>(blu);
    if (b < max_a)
      allmax = false;
  }
  if (allmax) {
    cout << max_a * n << endl;
    return;
  }

  ll aim = 0;
  for (auto blu : blulist) {
    ll b = get<0>(blu);
    ll left = get<1>(blu);
    ll right = get<2>(blu);
    aim += b * left;
  }
  debug_printf("---- aim = %lld\n", aim);

  V<t4> incr_blulist;
  rep(i, n) {
    ll b = get<0>(blulist[i]);
    ll left = get<1>(blulist[i]);
    ll right = get<2>(blulist[i]);
    incr_blulist.pb(t4(b * left + (max_a - b) * right, b, left, right));
  }
  sort_reverse(incr_blulist);

  ll base = 0;
  ll end = 0;

  for (auto item : incr_blulist) {
    ll incr = get<0>(item);
    ll b = get<1>(item);
    ll left = get<2>(item);
    ll right = get<3>(item);
    debug_printf("--- base=%lld incr=%lld b=%lld left=%lld right=%lld\n", base,
                 incr, b, left, right);

    if (base + incr <= aim) {
      base += incr;
      end++;
    } else
      break;
  }

  ll min_append = max_a;

  rep(i, n) {
    ll lbase = base;
    if (i < end) {
      lbase -= get<0>(incr_blulist[i]);
      lbase += get<0>(incr_blulist[end]);
    }
    ll incr = get<0>(incr_blulist[i]);
    ll b = get<1>(incr_blulist[i]);
    ll left = get<2>(incr_blulist[i]);
    ll right = get<3>(incr_blulist[i]);

    ll laim = aim - lbase;
    assert(laim >= 0);

    ll floor = 0;
    ll ceil = max_a;
    ll nowpos = (floor + ceil) / 2LL;
    ll append = max_a;
    for (; floor <= ceil; nowpos = (floor + ceil) / 2LL) {
      ll lincr = left * min(nowpos, b);
      lincr += right * max(0LL, nowpos - b);
      if (lincr < laim) {
        floor = nowpos + 1LL;
      } else {
        ceil = nowpos - 1LL;
        append = min(append, nowpos);
      }
    }

    min_append = min(min_append, append);
  }

  debug_printf("---- ans\n");
  cout << end * max_a + min_append << endl;
}

///////////////////////////////////////////////////////////////////////////////