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

#include <bits/stdc++.h>
using namespace std;
const long long MOD = 1e9 + 7;
typedef long long ll;

// auto mod int
// https://youtu.be/L8grWxBlIZ4?t=9858
// https://youtu.be/ERZuLAxZffQ?t=4807 : optimize
// https://youtu.be/8uowVvQ_-Mo?t=1329 : division
const int mod = 1000000007;
struct mint {
  ll x; // typedef long long ll;
  mint(ll 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(ll t) const {
    if (!t)
      return 1;
    mint a = pow(t >> 1);
    a *= a;
    if (t & 1)
      a *= *this;
    return a;
  }

  // for prime mod
  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; }
};
istream &operator>>(istream &is, mint &a) { return is >> a.x; }
ostream &operator<<(ostream &os, const mint &a) { return os << a.x; }
int main() {
  long long n, k;
  cin >> n >> k;
  vector<long long> a(n);
  vector<long long> pos; // 0以上の数を受け取る
  vector<long long> neg; // 負の値を受け取る
  for (long long i = 0; i < n; i++) {
    long long x;
    cin >> x;
    a[i] = x;
    if (x >= 0)
      pos.push_back(x);
    else
      neg.push_back(x);
  }
  sort(pos.begin(), pos.end());
  sort(neg.begin(), neg.end());    // 絶対値が大きい順
  reverse(pos.begin(), pos.end()); // 大きい順にする
  mint ans = 1;
  // k=nの時
  if (k == n) {
    for (long long i = 0; i < n; i++) {
      ans *= a[i];
    }
    cout << ans << endl;
    return 0;
  }
  // 正の数がないとき
  if (pos.empty()) {
    // 答えが負になるとき絶対値の小さい順
    if (k % 2) {
      reverse(neg.begin(), neg.end());
      for (long long i = 0; i < k; i++) {
        ans *= (MOD + neg[i]);
      }
      cout << ans << endl;
      return 0;
    }
    // 答えが正になるとき大きい順でとる
    else {
      for (long long i = 0; i < k; i++) {
        ans *= (MOD + neg[i]);
      }
      cout << ans << endl;
      return 0;
    }
  }

  // kが奇数の時正の数の最大値をあらかじめ取っておく
  if (k % 2) {
    ans *= pos[0];
    pos.erase(pos.begin());
  }

  long long p = pos.size();
  long long ne = neg.size();
  vector<long long> pos_pair(p / 2);
  vector<long long> neg_pair(ne / 2);
  for (long long i = 0; i < p / 2; i++) {
    pos_pair[i] = pos[i * 2] * pos[i * 2 + 1];
  }
  for (long long i = 0; i < ne / 2; i++) {
    neg_pair[i] = neg[i * 2] * neg[i * 2 + 1];
  }
  long long pk = 0, nk = 0;
  for (long long i = 0; i < k / 2; i++) {
    if (pk == p / 2) {
      ans *= neg_pair[nk];
      nk++;
    } else if (nk == ne / 2) {
      ans *= pos_pair[pk];
      pk++;
    } else {
      if (pos_pair[pk] > neg_pair[nk]) {
        ans *= pos_pair[pk];
        pk++;
      } else {
        ans *= neg_pair[nk];
        nk++;
      }
    }
  }
  cout << ans << endl;
}


Error: 0

Fixed submission:

#include <bits/stdc++.h>
using namespace std;
const long long MOD = 1e9 + 7;
int main() {
  int n, k;
  cin >> n >> k;
  vector<long long> a(n);
  vector<long long> pos;
  vector<long long> neg;
  for (int i = 0; i < n; i++) {
    long long x;
    cin >> x;
    a[i] = x;
    if (x >= 0)
      pos.push_back(x);
    else {
      x *= -1;
      neg.push_back(x);
    }
  }
  sort(pos.begin(), pos.end());
  sort(neg.begin(), neg.end());
  reverse(pos.begin(), pos.end());
  reverse(neg.begin(), neg.end());
  long long ans = 1;
  if (k == n) {
    for (int i = 0; i < k; i++) {
      ans *= a[i];
      ans %= MOD;
    }
    cout << ans << endl;
    return 0;
  }

  if (pos.empty()) {
    if (k % 2) {
      reverse(neg.begin(), neg.end());
      for (int i = 0; i < k; i++) {
        ans *= neg[i];
        ans %= MOD;
      }
      ans *= -1;
      ans += MOD;
      cout << ans << endl;
      return 0;
    } else {
      for (int i = 0; i < k; i++) {
        ans *= neg[i];
        ans %= MOD;
      }
      cout << ans << endl;
      return 0;
    }
  }

  if (k % 2) {
    ans *= pos[0];
    pos.erase(pos.begin());
  }

  long long p = pos.size();
  long long ne = neg.size();
  vector<long long> pos_pair(p / 2);
  vector<long long> neg_pair(ne / 2);
  for (int i = 0; i < p / 2; i++) {
    pos_pair[i] = (pos[i * 2] * pos[i * 2 + 1]) % MOD;
  }
  for (int i = 0; i < ne / 2; i++) {
    neg_pair[i] = (neg[i * 2] * neg[i * 2 + 1]) % MOD;
  }
  long long pk = 0, nk = 0;
  for (int i = 0; i < k / 2; i++) {
    if (pk == p / 2) {
      ans *= neg_pair[nk];
      ans %= MOD;
      nk++;
    } else if (nk == n / 2) {
      ans *= pos_pair[pk];
      ans %= MOD;
      pk++;
    } else {
      if (pos_pair[pk] > neg_pair[nk]) {
        ans *= pos_pair[pk];
        ans %= MOD;
        pk++;
      } else {
        ans *= neg_pair[nk];
        ans %= MOD;
        nk++;
      }
    }
  }
  cout << ans << endl;
}