Problem p02574 (C++)

Buggy submission (status: Runtime Error):
#include "bits/stdc++.h"
using namespace std;
typedef long long ll;
typedef pair<int, int> pii;
typedef pair<ll, ll> pll;
const int INF = 1e9;
const ll LINF = 1e18;
inline ll gcd(ll a, ll b) { return b ? gcd(b, a % b) : a; }
inline ll lcm(ll a, ll b) { return a / gcd(a, b) * b; }
template <class S, class T>
ostream &operator<<(ostream &out, const pair<S, T> &o) {
  out << "(" << o.first << "," << o.second << ")";
  return out;
}
template <class T> ostream &operator<<(ostream &out, const vector<T> &V) {
  for (int i = 0; i < V.size(); i++) {
    out << V[i];
    if (i != V.size() - 1)
      out << " ";
  }
  return out;
}
template <class T>
ostream &operator<<(ostream &out, const vector<vector<T>> &Mat) {
  for (int i = 0; i < Mat.size(); i++) {
    if (i != 0)
      out << endl;
    out << Mat[i];
  }
  return out;
}
template <class S, class T>
ostream &operator<<(ostream &out, const map<S, T> &mp) {
  out << "{ ";
  for (auto it = mp.begin(); it != mp.end(); it++) {
    out << it->first << ":" << it->second;
    if (mp.size() - 1 != distance(mp.begin(), it))
      out << ", ";
  }
  out << " }";
  return out;
}
template <typename T> vector<T> make_v(size_t a) { return vector<T>(a); }
template <typename T, typename... Ts> auto make_v(size_t a, Ts... ts) {
  return vector<decltype(make_v<T>(ts...))>(a, make_v<T>(ts...));
}
template <typename T, typename V>
typename enable_if<is_class<T>::value == 0>::type fill_v(T &t, const V &v) {
  t = v;
}
template <typename T, typename V>
typename enable_if<is_class<T>::value != 0>::type fill_v(T &t, const V &v) {
  for (auto &e : t)
    fill_v(e, v);
}
/*
 <url:>
 問題文============================================================
 =================================================================
 解説=============================================================
 ================================================================
 */

/**
 * 素因数分解 3
 * 与えられる値をその都度処理
 * 戻り値 {素因数の連想配列(値と個数)}
 */
map<ll, ll> PrimeFact(ll n) {
  map<ll, ll> mp;
  while (n != 1) {
    if (n == 2 || n == 3) {
      mp[n]++;
      n /= n;
      continue;
    }
    bool prime_flag = false;
    for (ll i = 2; i * i <= n; i++) {
      if (n % i == 0) {
        mp[i]++;
        n /= i;
        prime_flag = true;
        break;
      }
    }
    if (!prime_flag) {
      mp[n]++;
      n /= n;
    }
  }
}

template <class Type> Type solve(Type res = Type()) {
  int N;
  cin >> N;
  vector<ll> A(N);
  for (auto &in : A)
    cin >> in;

  { // check pairwise
    bool prime[1000001] = {false};
    bool ok = true;
    for (int i = N - 1; i >= 0; i--) {
      for (auto p : PrimeFact(A[i])) {
        if (prime[p.first]) {
          ok = false;
          break;
        } else {
          prime[p.first] = true;
        }
      }
    }
    if (ok) {
      return "pairwise coprime";
    }
  }
  {
    // check setwise
    ll GCD = 0;
    for (auto a : A)
      GCD = gcd(GCD, a);
    if (GCD == 1) {
      return "setwise coprime";
    }
  }
  return "not coprime";
  return res;
}
int main(void) {
  cin.tie(0);
  ios_base::sync_with_stdio(false);
  // solve(0);
  cout << fixed << setprecision(15) << solve<string>() << endl;
  return 0;
}


Fixed submission:
#include "bits/stdc++.h"
using namespace std;
typedef long long ll;
typedef pair<int, int> pii;
typedef pair<ll, ll> pll;
const int INF = 1e9;
const ll LINF = 1e18;
inline ll gcd(ll a, ll b) { return b ? gcd(b, a % b) : a; }
inline ll lcm(ll a, ll b) { return a / gcd(a, b) * b; }
template <class S, class T>
ostream &operator<<(ostream &out, const pair<S, T> &o) {
  out << "(" << o.first << "," << o.second << ")";
  return out;
}
template <class T> ostream &operator<<(ostream &out, const vector<T> &V) {
  for (int i = 0; i < V.size(); i++) {
    out << V[i];
    if (i != V.size() - 1)
      out << " ";
  }
  return out;
}
template <class T>
ostream &operator<<(ostream &out, const vector<vector<T>> &Mat) {
  for (int i = 0; i < Mat.size(); i++) {
    if (i != 0)
      out << endl;
    out << Mat[i];
  }
  return out;
}
template <class S, class T>
ostream &operator<<(ostream &out, const map<S, T> &mp) {
  out << "{ ";
  for (auto it = mp.begin(); it != mp.end(); it++) {
    out << it->first << ":" << it->second;
    if (mp.size() - 1 != distance(mp.begin(), it))
      out << ", ";
  }
  out << " }";
  return out;
}
template <typename T> vector<T> make_v(size_t a) { return vector<T>(a); }
template <typename T, typename... Ts> auto make_v(size_t a, Ts... ts) {
  return vector<decltype(make_v<T>(ts...))>(a, make_v<T>(ts...));
}
template <typename T, typename V>
typename enable_if<is_class<T>::value == 0>::type fill_v(T &t, const V &v) {
  t = v;
}
template <typename T, typename V>
typename enable_if<is_class<T>::value != 0>::type fill_v(T &t, const V &v) {
  for (auto &e : t)
    fill_v(e, v);
}
/*
 <url:>
 問題文============================================================
 =================================================================
 解説=============================================================
 ================================================================
 */

/**
 * 素因数分解 3
 * 与えられる値をその都度処理
 * 戻り値 {素因数の連想配列(値と個数)}
 */
map<ll, ll> PrimeFact(ll n) {
  map<ll, ll> mp;
  while (n != 1) {
    if (n == 2 || n == 3) {
      mp[n]++;
      n /= n;
      continue;
    }
    bool prime_flag = false;
    for (ll i = 2; i * i <= n; i++) {
      if (n % i == 0) {
        mp[i]++;
        n /= i;
        prime_flag = true;
        break;
      }
    }
    if (!prime_flag) {
      mp[n]++;
      n /= n;
    }
  }
  return mp;
}

template <class Type> Type solve(Type res = Type()) {
  int N;
  cin >> N;
  vector<ll> A(N);
  for (auto &in : A)
    cin >> in;

  { // check pairwise
    bool prime[1000001] = {false};
    bool ok = true;
    for (int i = N - 1; i >= 0; i--) {
      for (auto p : PrimeFact(A[i])) {
        if (prime[p.first]) {
          ok = false;
          break;
        } else {
          prime[p.first] = true;
        }
      }
    }
    if (ok) {
      return "pairwise coprime";
    }
  }
  {
    // check setwise
    ll GCD = 0;
    for (auto a : A)
      GCD = gcd(GCD, a);
    if (GCD == 1) {
      return "setwise coprime";
    }
  }
  return "not coprime";
  return res;
}
int main(void) {
  cin.tie(0);
  ios_base::sync_with_stdio(false);
  // solve(0);
  cout << fixed << setprecision(15) << solve<string>() << endl;
  return 0;
}