Problem p02574 (C++) — buggy submission, verdict: Time Limit Exceeded

#include <algorithm>
#include <bits/stdc++.h>
#include <bitset>
#include <cmath>
#include <cstring>
#include <deque>
#include <iomanip>
#include <iostream>
#include <limits>
#include <list>
#include <map>
#include <numeric>
#include <queue>
#include <random>
#include <regex>
#include <set>
#include <stack>
#include <string>
#include <tuple>
#include <type_traits>
#include <unordered_map>
#include <unordered_set>
#include <vector>

using namespace std;

typedef long long ll;
#define rep(i, n) for (ll i = 0; i < n; ++i)
#define exout(x) printf("%.10f\\n", x)
const double pi = acos(-1.0);
const ll MOD = 1000000007;
const ll INF = 1e18;
const ll MAX_N = 201010;

// 最大公約数
ll gcd(ll x, ll y) { return y ? gcd(y, x % y) : x; }

ll lcm(ll x, ll y) {
  if (x == 0 || y == 0)
    return 0;
  return (x / gcd(x, y) * y);
}

ll dx[4] = {0, 0, -1, 1};
ll dy[4] = {-1, 1, 0, 0};
ll ans[110][110];
bool flag[2][110];

const ll A = 1010101;

// long longしか使わない
// 素数は1より大きい
// lower_boundは指定したkey以上の要素の一番左のイテレータをかえす
// upper_boundは指定したkeyより大きい要素の一番左のイテレータをかえす
int main() {
  ll n;
  cin >> n;
  vector<ll> a(n);
  vector<ll> c(A);

  rep(i, n) {
    cin >> a[i];
    c[a[i]]++;
  }
  bool flag = true;
  for (ll i = 2; i < A; ++i) {
    ll cnt = 0;
    for (ll j = i; j < A; j += i) {
      cnt += c[j];
    }
    if (cnt > 1) {
      flag = false;
    }
  }
  if (flag) {
    puts("pairwise coprime");
    return 0;
  }
  ll g = 0;
  rep(i, n) { g = gcd(g, a[i]); }
  if (g == 1) {
    puts("setwise coprime");
    return 0;
  }
  puts("not coprime");
  return 0;
}


Error: TLE

Fixed submission:

#include <algorithm>
#include <bits/stdc++.h>
#include <bitset>
#include <cmath>
#include <cstring>
#include <deque>
#include <iomanip>
#include <iostream>
#include <limits>
#include <list>
#include <map>
#include <numeric>
#include <queue>
#include <random>
#include <regex>
#include <set>
#include <stack>
#include <string>
#include <tuple>
#include <type_traits>
#include <unordered_map>
#include <unordered_set>
#include <vector>

using namespace std;

typedef long long ll;
#define rep(i, n) for (ll i = 0; i < n; ++i)
#define exout(x) printf("%.10f\\n", x)
const double pi = acos(-1.0);
const ll MOD = 1000000007;
const ll INF = 1e18;
const ll MAX_N = 201010;

// 最大公約数
ll gcd(ll x, ll y) { return y ? gcd(y, x % y) : x; }

ll lcm(ll x, ll y) {
  if (x == 0 || y == 0)
    return 0;
  return (x / gcd(x, y) * y);
}

ll dx[4] = {0, 0, -1, 1};
ll dy[4] = {-1, 1, 0, 0};
ll ans[110][110];
bool flag[2][110];

// long longしか使わない
// 素数は1より大きい
// lower_boundは指定したkey以上の要素の一番左のイテレータをかえす
// upper_boundは指定したkeyより大きい要素の一番左のイテレータをかえす
int main() {
  ll n;
  cin >> n;
  vector<ll> a(n), b(n);
  map<ll, ll> mp;
  bool flag2 = true;
  rep(i, n) {
    cin >> a[i];
    b[i] = a[i];
    if (flag2) {
      for (ll j = 1; j <= sqrt(a[i]); ++j) {
        if (a[i] % j == 0) {
          if (mp[j]) {
            flag2 = false;
            break;
          }
          if (j != 1)
            mp[j]++;
          while (a[i] % j == 0) {
            a[i] /= j;
          }
        }
      }
      if (mp[a[i]]) {
        flag2 = false;
        break;
      }
      mp[a[i]]++;
    }
  }
  if (flag2) {
    puts("pairwise coprime");
    return 0;
  }
  ll res2 = b[0];
  rep(i, n - 1) { res2 = gcd(res2, b[i + 1]); }
  if (res2 == 1) {
    puts("setwise coprime");
    return 0;
  }
  puts("not coprime");
  return 0;
}