Problem p02616 (C++) — buggy submission, verdict: Runtime Error

#include <bits/stdc++.h>
using namespace std;
#define rep(i, n) for (int i = 0; i < (int)(n); ++i)
#define rep2(i, s, n) for (int i = (s); i < (int)(n); i++)

int main() {
  int mod = 1000000007;
  long n, k;
  cin >> n >> k;
  vector<long> a(n);
  vector<long> pl, mi;
  priority_queue<long> p;
  priority_queue<long> m;
  rep(i, n) {
    cin >> a.at(i);
    if (a.at(i) >= 0) {
      pl.push_back(a[i]);
    } else if (a.at(i) < 0) {
      mi.push_back(a[i]);
    }
  }
  sort(pl.begin(), pl.end());
  reverse(pl.begin(), pl.end());
  sort(mi.begin(), mi.end());
  reverse(mi.begin(), mi.end());
  long ans = 1;
  if (pl.size() == 0) {
    if (k % 2 == 1) {
      rep(i, k) {
        ans *= abs(mi[i]);
        ans %= mod;
      }
      cout << (mod - ans) % mod << endl;
    } else {
      reverse(mi.begin(), mi.end());
      rep(i, k) {
        ans *= abs(mi[i]);
        ans %= mod;
      }
      cout << ans << endl;
    }
  } else if (n == k) {
    rep(i, n) {
      ans *= abs(a[i]);
      ans %= mod;
    }
    if (mi.size() % 2 == 0) {
      cout << ans << endl;
    } else {
      cout << (mod - ans) % mod << endl;
    }
  } else {
    reverse(mi.begin(), mi.end());
    if (k % 2 == 1) {
      ans *= pl[0];
      rep(i, (pl.size() - 1) / 2) { p.push(pl[i * 2 + 1] * pl[i * 2 + 2]); }
      k--;
    } else {
      rep(i, pl.size() / 2) { p.push(pl[i * 2] * pl[i * 2 + 1]); }
    }
    rep(i, mi.size() / 2) { m.push(mi[i * 2] * mi[i * 2 + 1]); }
    int co = 0;
    while (co < k / 2) {
      if (!p.empty() && !m.empty()) {
        if (p.top() < m.top()) {
          ans *= m.top() % mod;
          ans %= mod;
          m.pop();
        } else {
          ans *= p.top() % mod;
          ans %= mod;
          p.pop();
        }
      } else if (p.empty()) {
        ans *= m.top() % mod;
        ans %= mod;
        m.pop();
      } else if (m.empty()) {
        ans *= p.top() % mod;
        ans %= mod;
        p.pop();
      }
      co++;
    }
    cout << ans << endl;
  }
}

Error: 0

Fixed submission:

#include <bits/stdc++.h>
using namespace std;
#define rep(i, n) for (int i = 0; i < (int)(n); ++i)
#define rep2(i, s, n) for (int i = (s); i < (int)(n); i++)

int main() {
  int mod = 1000000007;
  long n, k;
  cin >> n >> k;
  vector<long> a(n);
  vector<long> pl, mi;
  priority_queue<long> p;
  priority_queue<long> m;
  rep(i, n) {
    cin >> a.at(i);
    if (a.at(i) >= 0) {
      pl.push_back(a[i]);
    } else if (a.at(i) < 0) {
      mi.push_back(a[i]);
    }
  }
  long u = pl.size();
  long v = mi.size();
  sort(pl.begin(), pl.end());
  reverse(pl.begin(), pl.end());
  sort(mi.begin(), mi.end());
  reverse(mi.begin(), mi.end());
  long ans = 1;
  if (pl.size() == 0) {
    if (k % 2 == 1) {
      rep(i, k) {
        ans *= abs(mi[i]);
        ans %= mod;
      }
      cout << (mod - ans) % mod << endl;
    } else {
      reverse(mi.begin(), mi.end());
      rep(i, k) {
        ans *= abs(mi[i]);
        ans %= mod;
      }
      cout << ans << endl;
    }
  } else if (n == k) {
    rep(i, n) {
      ans *= abs(a[i]);
      ans %= mod;
    }
    if (k % 2 == 0) {
      cout << ans << endl;
    } else {
      cout << (mod - ans) % mod << endl;
    }
  } else {
    reverse(mi.begin(), mi.end());
    if (k % 2 == 1) {
      ans *= pl[0];
      rep(i, (pl.size() - 1) / 2) { p.push(pl[i * 2 + 1] * pl[i * 2 + 2]); }
      k--;
    } else {
      rep(i, pl.size() / 2) { p.push(pl[i * 2] * pl[i * 2 + 1]); }
    }
    rep(i, mi.size() / 2) { m.push(mi[i * 2] * mi[i * 2 + 1]); }
    int co = 0;
    while (co < k / 2) {
      if (p.top() < m.top()) {
        ans *= m.top() % mod;
        ans %= mod;
        m.pop();
      } else {
        ans *= p.top() % mod;
        ans %= mod;
        p.pop();
      }
      co++;
    }
    cout << ans << endl;
  }
}