Problem p02616 (C++)

Buggy submission (status: Runtime Error):
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
#define rep(i, n) for (ll i = 0, i##_len = (n); i < i##_len; i++)
#define reps(i, s, n) for (ll i = (s), i##_len = (n); i < i##_len; i++)
#define rrep(i, n) for (ll i = (n)-1; i >= 0; i--)
#define rreps(i, e, n) for (ll i = (n)-1; i >= (e); i--)
#define all(x) (x).begin(), (x).end()
#define rall(x) (x).rbegin(), (x).rend()
#define sz(x) ((ll)(x).size())
#define len(x) ((ll)(x).length())
#define endl "\n"

struct mint {
  const int mod = 1000000007;
  long long x;
  mint(long long x = 0) : x((x % mod + mod) % mod) {}
  mint operator-() const { return mint(-x); }
  mint &operator+=(const mint a) {
    if ((x += a.x) >= mod)
      x -= mod;
    return *this;
  }
  mint &operator-=(const mint a) {
    if ((x += mod - a.x) >= mod)
      x -= mod;
    return *this;
  }
  mint &operator*=(const mint a) {
    (x *= a.x) %= mod;
    return *this;
  }
  mint operator+(const mint a) const { return mint(*this) += a; }
  mint operator-(const mint a) const { return mint(*this) -= a; }
  mint operator*(const mint a) const { return mint(*this) *= a; }
  mint pow(long long t) const {
    if (!t)
      return 1;
    mint a = pow(t >> 1);
    a *= a;
    if (t & 1)
      a *= *this;
    return a;
  }
  mint inv() const { return pow(mod - 2); }
  mint &operator/=(const mint a) { return *this *= a.inv(); }
  mint operator/(const mint a) const { return mint(*this) /= a; }
};

int main() {
  cin.tie(0);
  ios::sync_with_stdio(false);
  // ifstream in("input.txt");
  // cin.rdbuf(in.rdbuf());
  ll n, k;
  cin >> n >> k;
  vector<ll> a(n);
  rep(i, n) cin >> a[i];
  multiset<ll> minus, plus;
  ll zero = 0;
  rep(i, n) {
    if (a[i] == 0) {
      zero++;
    } else if (a[i] < 0) {
      minus.insert(-a[i]);
    } else if (a[i] > 0) {
      plus.insert(a[i]);
    }
  }
  // 答えがどうしてもマイナスになる場合
  if (((k % 2 == 1) && (zero == 0) && (sz(plus) == 0))) {
    mint ans = 1;
    ll cnt = 0;
    for (auto it = minus.begin(); it != minus.end(); it++) {
      ans *= -*it;
      cnt++;
      if (cnt == k)
        break;
    }
    cout << ans.x << endl;
    return 0;
  }
  // 答えがどうしてもマイナスになる場合2
  if ((n == k) && (zero == 0) && (sz(minus) % 2 == 1)) {
    mint ans = 1;
    ll cnt = 0;
    for (auto it = minus.begin(); it != minus.end(); it++) {
      ans *= -*it;
    }
    for (auto it = plus.begin(); it != plus.end(); it++) {
      ans *= *it;
    }
    cout << ans.x << endl;
    return 0;
  }
  // マイナスになりそうな答えを0にできる場合
  if (((k % 2 == 1) && (zero != 0) && (sz(plus) == 0))) {
    cout << 0 << endl;
    return 0;
  }
  // その2
  if ((n == k) && (zero != 0) && (sz(minus) % 2 == 1)) {
    cout << 0 << endl;
    return 0;
  }
  // 答えをプラスにできる場合
  priority_queue<pair<ll, ll>> pq;
  auto pit = plus.rbegin();
  rep(i, sz(plus) / 2) {
    ll v1 = *pit;
    pit++;
    ll v2 = *pit;
    pit++;
    pq.push(make_pair(v1 * v2, 0));
  }
  auto mit = minus.rbegin();
  rep(i, sz(minus) / 2) {
    ll v1 = *mit;
    mit++;
    ll v2 = *mit;
    mit++;
    pq.push(make_pair(v1 * v2, 1));
  }
  mint ans = 1;
  if (k % 2 == 0) {
    rep(i, k / 2) {
      ans *= pq.top().first;
      pq.pop();
    }
  } else {
    ll puse = 0;
    rep(i, k / 2) {
      while ((((puse + 1) * 2) == sz(plus)) && (pq.top().second == 0)) {
        pq.pop();
      }
      pair<ll, ll> p = pq.top();
      pq.pop();
      ans *= p.first;
      if (p.second == 0)
        puse++;
    }
    pit = plus.rbegin();
    rep(i, puse * 2) pit++;
    ans *= *pit;
  }
  cout << ans.x << endl;
  return 0;
}


Fixed submission:
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
#define rep(i, n) for (ll i = 0, i##_len = (n); i < i##_len; i++)
#define reps(i, s, n) for (ll i = (s), i##_len = (n); i < i##_len; i++)
#define rrep(i, n) for (ll i = (n)-1; i >= 0; i--)
#define rreps(i, e, n) for (ll i = (n)-1; i >= (e); i--)
#define all(x) (x).begin(), (x).end()
#define rall(x) (x).rbegin(), (x).rend()
#define sz(x) ((ll)(x).size())
#define len(x) ((ll)(x).length())
#define endl "\n"

struct mint {
  const int mod = 1000000007;
  long long x;
  mint(long long x = 0) : x((x % mod + mod) % mod) {}
  mint operator-() const { return mint(-x); }
  mint &operator+=(const mint a) {
    if ((x += a.x) >= mod)
      x -= mod;
    return *this;
  }
  mint &operator-=(const mint a) {
    if ((x += mod - a.x) >= mod)
      x -= mod;
    return *this;
  }
  mint &operator*=(const mint a) {
    (x *= a.x) %= mod;
    return *this;
  }
  mint operator+(const mint a) const { return mint(*this) += a; }
  mint operator-(const mint a) const { return mint(*this) -= a; }
  mint operator*(const mint a) const { return mint(*this) *= a; }
  mint pow(long long t) const {
    if (!t)
      return 1;
    mint a = pow(t >> 1);
    a *= a;
    if (t & 1)
      a *= *this;
    return a;
  }
  mint inv() const { return pow(mod - 2); }
  mint &operator/=(const mint a) { return *this *= a.inv(); }
  mint operator/(const mint a) const { return mint(*this) /= a; }
};

int main() {
  cin.tie(0);
  ios::sync_with_stdio(false);
  // ifstream in("input.txt");
  // cin.rdbuf(in.rdbuf());
  ll n, k;
  cin >> n >> k;
  vector<ll> a(n);
  rep(i, n) cin >> a[i];
  multiset<ll> minus, plus;
  ll zero = 0;
  rep(i, n) {
    if (a[i] == 0) {
      zero++;
    } else if (a[i] < 0) {
      minus.insert(-a[i]);
    } else if (a[i] > 0) {
      plus.insert(a[i]);
    }
  }
  // 答えがどうしてもマイナスになる場合
  if (((k % 2 == 1) && (zero == 0) && (sz(plus) == 0))) {
    mint ans = 1;
    ll cnt = 0;
    for (auto it = minus.begin(); it != minus.end(); it++) {
      ans *= -*it;
      cnt++;
      if (cnt == k)
        break;
    }
    cout << ans.x << endl;
    return 0;
  }
  // 答えがどうしてもマイナスになる場合2
  if ((n == k) && (zero == 0) && (sz(minus) % 2 == 1)) {
    mint ans = 1;
    ll cnt = 0;
    for (auto it = minus.begin(); it != minus.end(); it++) {
      ans *= -*it;
    }
    for (auto it = plus.begin(); it != plus.end(); it++) {
      ans *= *it;
    }
    cout << ans.x << endl;
    return 0;
  }
  // マイナスになりそうな答えを0にできる場合
  if (((k % 2 == 1) && (zero != 0) && (sz(plus) == 0))) {
    cout << 0 << endl;
    return 0;
  }
  // その2
  if ((n == k) && (zero != 0) && (sz(minus) % 2 == 1)) {
    cout << 0 << endl;
    return 0;
  }
  // どうしても0になる場合
  if ((sz(plus) + sz(minus)) < k) {
    cout << 0 << endl;
    return 0;
  }
  // 答えをプラスにできる場合
  priority_queue<pair<ll, ll>> pq;
  auto pit = plus.rbegin();
  rep(i, sz(plus) / 2) {
    ll v1 = *pit;
    pit++;
    ll v2 = *pit;
    pit++;
    pq.push(make_pair(v1 * v2, 0));
  }
  auto mit = minus.rbegin();
  rep(i, sz(minus) / 2) {
    ll v1 = *mit;
    mit++;
    ll v2 = *mit;
    mit++;
    pq.push(make_pair(v1 * v2, 1));
  }
  mint ans = 1;
  if (k % 2 == 0) {
    rep(i, k / 2) {
      ans *= pq.top().first;
      pq.pop();
    }
  } else {
    ll puse = 0;
    rep(i, k / 2) {
      while ((((puse + 1) * 2) == sz(plus)) && (pq.top().second == 0)) {
        pq.pop();
      }
      pair<ll, ll> p = pq.top();
      pq.pop();
      ans *= p.first;
      if (p.second == 0)
        puse++;
    }
    pit = plus.rbegin();
    rep(i, puse * 2) pit++;
    ans *= *pit;
  }
  cout << ans.x << endl;
  return 0;
}