Problem p02824 (C++)

Buggy submission (status: Runtime Error):
#define STOPIT
#include <bits/stdc++.h>
#define loop(n)                                                                \
  for (lint ngtkana_is_a_genius = 0; ngtkana_is_a_genius < lint(n);            \
       ngtkana_is_a_genius++)
#define rep(i, begin, end) for (lint i = lint(begin); (i) < lint(end); i++)
#define all(v) v.begin(), v.end()
#define rand(l, r) std::uniform_int_distribution<>(l, r)(mt)
using lint = long long;
auto mt = std::mt19937_64(std::random_device{}());
auto cmn = [](auto &&a, auto b) {
  if (a > b) {
    a = b;
    return true;
  }
  return false;
};
auto cmx = [](auto &&a, auto b) {
  if (a < b) {
    a = b;
    return true;
  }
  return false;
};
void debug_impl() { std::cerr << std::endl; }
template <typename Head, typename... Tail>
void debug_impl(Head head, Tail... tail) {
  std::cerr << " " << head;
  debug_impl(tail...);
}
#ifndef STOPIT
#define debug(...)                                                             \
  do {                                                                         \
    std::cerr << std::boolalpha << "[" << #__VA_ARGS__ << "]:";                \
    debug_impl(__VA_ARGS__);                                                   \
    std::cerr << std::noboolalpha;                                             \
  } while (false)
#else
#define debug(...)                                                             \
  {}
#endif

template <typename Container, typename Value = typename Container::value_type,
          std::enable_if_t<!std::is_same<Container, std::string>::value,
                           std::nullptr_t> = nullptr>
std::istream &operator>>(std::istream &is, Container &v) {
  for (auto &x : v) {
    is >> x;
  }
  return is;
}

template <typename Container, typename Value = typename Container::value_type,
          std::enable_if_t<!std::is_same<Container, std::string>::value,
                           std::nullptr_t> = nullptr>
std::ostream &operator<<(std::ostream &os, Container const &v) {
  os << "{";
  for (auto it = v.begin(); it != v.end(); it++) {
    os << (it != v.begin() ? "," : "") << *it;
  }
  return os << "}";
}

/*
二分探索です。i 番目が行ける条件を考えます。
まず、i < p なら明らかに OK です。
そうでないとき、まず、全振りして a_{p-1} に勝てなければダメです。
それが大丈夫だとして、残りが始めの p-1 個と右のものだけに振れるなら OK です。
それがダメなら、[p-1,i[ に、忖度しながら振っていきます。
*/

int main() {
  std::cin.tie(0);
  std::cin.sync_with_stdio(false);
  lint n, m, v, p;
  std::cin >> n >> m >> v >> p;
  std::vector<lint> a(n);
  std::cin >> a;
  std::sort(all(a), std::greater<>{});
  debug(m, v, p, a);

  std::vector<lint> b = {0};
  std::partial_sum(all(a), std::back_inserter(b));
  debug(b);
  lint rival = a.at(p - 1);
  debug(rival);

  auto check = [&](lint i) -> bool {
    if (i < p)
      return true;
    debug("foo");
    lint me = a.at(i);
    if (rival > me + m)
      return false;
    lint u = n + p - i - 2;
    debug(u, "goo");
    if (v - 1 <= u)
      return true;
    debug("hoo");
    lint sum = b.at(i) - b.at(p - 1);
    lint k = v - 1 - u;
    debug(sum, k);
    debug(i - p + 1, me + m);
    return sum + k * m <= (i - p + 1) * (me + m);
  };
  // debug
  // rep(i,0,n) {
  //   bool b = check(i);
  //   debug(i,b);
  //   std::cerr<<std::endl;
  // }

  lint ok = 0, ng = n + 1;
  while (1 < ng - ok) {
    lint mid = (ng + ok) / 2;
    (check(mid) ? ok : ng) = mid;
  }
  lint ans = ok + 1;
  std::cout << ans << std::endl;
  return 0;
}

Fixed submission:
#define STOPIT
#include <bits/stdc++.h>
#define loop(n)                                                                \
  for (lint ngtkana_is_a_genius = 0; ngtkana_is_a_genius < lint(n);            \
       ngtkana_is_a_genius++)
#define rep(i, begin, end) for (lint i = lint(begin); (i) < lint(end); i++)
#define all(v) v.begin(), v.end()
#define rand(l, r) std::uniform_int_distribution<>(l, r)(mt)
using lint = long long;
auto mt = std::mt19937_64(std::random_device{}());
auto cmn = [](auto &&a, auto b) {
  if (a > b) {
    a = b;
    return true;
  }
  return false;
};
auto cmx = [](auto &&a, auto b) {
  if (a < b) {
    a = b;
    return true;
  }
  return false;
};
void debug_impl() { std::cerr << std::endl; }
template <typename Head, typename... Tail>
void debug_impl(Head head, Tail... tail) {
  std::cerr << " " << head;
  debug_impl(tail...);
}
#ifndef STOPIT
#define debug(...)                                                             \
  do {                                                                         \
    std::cerr << std::boolalpha << "[" << #__VA_ARGS__ << "]:";                \
    debug_impl(__VA_ARGS__);                                                   \
    std::cerr << std::noboolalpha;                                             \
  } while (false)
#else
#define debug(...)                                                             \
  {}
#endif

template <typename Container, typename Value = typename Container::value_type,
          std::enable_if_t<!std::is_same<Container, std::string>::value,
                           std::nullptr_t> = nullptr>
std::istream &operator>>(std::istream &is, Container &v) {
  for (auto &x : v) {
    is >> x;
  }
  return is;
}

template <typename Container, typename Value = typename Container::value_type,
          std::enable_if_t<!std::is_same<Container, std::string>::value,
                           std::nullptr_t> = nullptr>
std::ostream &operator<<(std::ostream &os, Container const &v) {
  os << "{";
  for (auto it = v.begin(); it != v.end(); it++) {
    os << (it != v.begin() ? "," : "") << *it;
  }
  return os << "}";
}

/*
二分探索です。i 番目が行ける条件を考えます。
まず、i < p なら明らかに OK です。
そうでないとき、まず、全振りして a_{p-1} に勝てなければダメです。
それが大丈夫だとして、残りが始めの p-1 個と右のものだけに振れるなら OK です。
それがダメなら、[p-1,i[ に、忖度しながら振っていきます。
*/

int main() {
  std::cin.tie(0);
  std::cin.sync_with_stdio(false);
  lint n, m, v, p;
  std::cin >> n >> m >> v >> p;
  std::vector<lint> a(n);
  std::cin >> a;
  std::sort(all(a), std::greater<>{});
  debug(m, v, p, a);

  std::vector<lint> b = {0};
  std::partial_sum(all(a), std::back_inserter(b));
  debug(b);
  lint rival = a.at(p - 1);
  debug(rival);

  auto check = [&](lint i) -> bool {
    if (i < p)
      return true;
    debug("foo");
    lint me = a.at(i);
    if (rival > me + m)
      return false;
    lint u = n + p - i - 2;
    debug(u, "goo");
    if (v - 1 <= u)
      return true;
    debug("hoo");
    lint sum = b.at(i) - b.at(p - 1);
    lint k = v - 1 - u;
    debug(sum, k);
    debug(i - p + 1, me + m);
    return sum + k * m <= (i - p + 1) * (me + m);
  };
  // debug
  // rep(i,0,n) {
  //   bool b = check(i);
  //   debug(i,b);
  //   std::cerr<<std::endl;
  // }

  lint ok = 0, ng = n;
  while (1 < ng - ok) {
    lint mid = (ng + ok) / 2;
    (check(mid) ? ok : ng) = mid;
  }
  lint ans = ok + 1;
  std::cout << ans << std::endl;
  return 0;
}