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

#include <algorithm>
#include <cmath>
#include <functional>
#include <iomanip>
#include <iostream>
#include <list>
#include <map>
#include <queue>
#include <set>
#include <stack>
#include <string>
#include <vector>

#define MOD (1000000000 + 7)

using namespace std;

typedef long long ll;
const int MAX = 3000000;

long long fac[MAX], finv[MAX], inv[MAX];

long long modinv(long long a, long long m) {
  long long b = m, u = 1, v = 0;
  while (b) {
    long long t = a / b;
    a -= t * b;
    swap(a, b);
    u -= t * v;
    swap(u, v);
  }
  u %= m;
  if (u < 0)
    u += m;
  return u;
}

ll modpow(ll a, ll n, ll mod) {
  ll res = 1;
  while (n > 0) {
    if (n & 1)
      res = res * a % mod;
    a = a * a % mod;
    n >>= 1;
  }
  return res;
}

ll gcd(ll a, ll b) { return b ? gcd(b, a % b) : a; }
ll lcm(ll a, ll b) { return a / gcd(a, b) * b; }
// テーブルを作る前処理
void COMinit() {
  fac[0] = fac[1] = 1;
  finv[0] = finv[1] = 1;
  inv[1] = 1;
  for (int i = 2; i < MAX; i++) {
    fac[i] = fac[i - 1] * i % MOD;
    inv[i] = MOD - inv[MOD % i] * (MOD / i) % MOD;
    finv[i] = finv[i - 1] * inv[i] % MOD;
  }
}

// 二項係数計算
long long COM(int n, int k) {
  if (n < k)
    return 0;
  if (n < 0 || k < 0)
    return 0;
  return fac[n] * (finv[k] * finv[n - k] % MOD) % MOD;
}

map<ll, ll> prime; // 素因数分解でそれぞれの素数がいくつ出てきたかを保存するmap

// O(√n)
// 整列済み(mapはkeyで自動で整列される)
void prime_factorize(ll n) {
  if (n <= 1)
    return;
  ll l = sqrt(n);
  for (ll i = 2; i <= (ll)(l); i++) { //
    if (n % i == 0) {
      prime_factorize(i);
      prime_factorize(ll(n / i));
      return;
    }
  }
  // mapでは存在しないkeyの場合も自動で構築される
  prime[n]++;
  return;
}

// 約数列挙？
vector<ll> divisor;
void list_divisor(ll n) {
  divisor.clear();
  for (int i = 1; i * i <= n; i++) {
    if (n % i == 0) {
      divisor.push_back(i);
      if (i * i != n)
        divisor.push_back(n / i);
    }
  }
  sort(divisor.begin(), divisor.end());
}

vector<bool> Eratosthenes(ll N) {
  vector<bool> primes(N, true);
  vector<ll> ret;
  primes[0] = false;
  primes[1] = false;
  for (ll i = 2; pow(i, 2) <= N; i++) {
    if (primes[i]) {
      for (int j = 2; i * j <= N; j++)
        primes[i * j] = false;
      ret.push_back(primes[i]);
    }
  }
  return primes; // ret;
}

template <typename T>
bool next_combination(const T first, const T last, int k) {
  // ** Sample **
  // vector<int> v{1, 2, 3, 4, 5, 6, 7};
  // do {
  //     for (int i = 0; i < k; i++) {
  //         cout << v[i] << " ";
  //     }
  //     cout << "| ";
  //     for (int i = k; i < v.size(); i++) {
  //         cout << v[i] << " ";
  //     }
  //     cout << endl;
  // } while(next_combination(v.begin(), v.end(), k));

  const T subset = first + k;
  // empty container | k = 0 | k == n
  if (first == last || first == subset || last == subset) {
    return false;
  }
  T src = subset;
  while (first != src) {
    src--;
    if (*src < *(last - 1)) {
      T dest = subset;
      while (*src >= *dest) {
        dest++;
      }
      iter_swap(src, dest);
      rotate(src + 1, dest + 1, last);
      rotate(subset, subset + (last - dest) - 1, last);
      return true;
    }
  }
  // restore
  rotate(first, subset, last);
  return false;
}

// union by size + path having
class UnionFind {
public:
  vector<ll> par; // 各元の親を表す配列
  vector<ll> siz; // 素集合のサイズを表す配列(1 で初期化)

  // Constructor
  UnionFind(ll sz_) : par(sz_), siz(sz_, 1LL) {
    for (ll i = 0; i < sz_; ++i)
      par[i] = i; // 初期では親は自分自身
  }
  void init(ll sz_) {
    par.resize(sz_);
    siz.assign(sz_, 1LL); // resize だとなぜか初期化されなかった
    for (ll i = 0; i < sz_; ++i)
      par[i] = i; // 初期では親は自分自身
  }

  // Member Function
  // Find
  ll root(ll x) { // 根の検索
    while (par[x] != x) {
      x = par[x] = par[par[x]]; // x の親の親を x の親とする
    }
    return x;
  }

  // Union(Unite, Merge)
  bool merge(ll x, ll y) {
    x = root(x);
    y = root(y);
    if (x == y)
      return false;
    // merge technique（データ構造をマージするテク．小を大にくっつける）
    if (siz[x] < siz[y])
      swap(x, y);
    siz[x] += siz[y];
    par[y] = x;
    return true;
  }

  bool issame(ll x, ll y) { // 連結判定
    return root(x) == root(y);
  }

  ll size(ll x) { // 素集合のサイズ
    return siz[root(x)];
  }
};

int main() {
  ll N;
  cin >> N;

  vector<ll> A(N, 0);
  vector<ll> five(N, 0);
  vector<ll> two(N, 0);
  vector<vector<ll>> count(20, vector<ll>(20, 0));
  vector<vector<ll>> cum(20, vector<ll>(20, 0));
  for (ll i = 0; i < N; i++) {
    string d;
    cin >> d;
    ll small = 0;
    if (d.find('.') != string::npos) {
      small = d.length() - d.find('.') - 1;
      d.erase(remove(d.begin(), d.end(), '.'), d.end());
    }
    A[i] = stoll(d);
    for (int j = 1; j <= 9 - small; j++) {
      A[i] *= 10;
    }

    // prime.clear();
    // prime_factorize(A[i]);

    ll twov = 0;
    ll fivev = 0;
    while (true) {
      if (A[i] % 2 != 0)
        break;
      A[i] /= 2;
      twov++;
    }
    while (true) {
      if (A[i] % 5 != 0)
        break;
      A[i] /= 5;
      fivev++;
    }

    five[i] = fivev >= 20 ? 19 : fivev;
    two[i] = twov >= 20 ? 19 : twov;
    // cout << d << "->" << stoll(d) << " -> " << A[i] << ":" << two[i] <<"," <<
    // five[i] << endl;
    count[two[i]][five[i]]++;
  }
  // return 0;
  for (ll i = 19; i >= 0; i--) {
    for (ll j = 19; j >= 0; j--) {
      if (i == 19 && j == 19)
        cum[i][j] = count[i][j];
      else if (i == 19)
        cum[i][j] = cum[i][j + 1] + count[i][j];
      else if (j == 19)
        cum[i][j] = cum[i + 1][j] + count[i][j];
      else
        cum[i][j] =
            cum[i][j + 1] + cum[i + 1][j] + count[i][j] - cum[i + 1][j + 1];
    }
  }
  // cout << "test " << endl;
  ll ret = 0;
  for (ll i = 0; i < N; i++) {
    ll res = (two[i] >= 18 - two[i] && five[i] >= 18 - five[i]) ? 1 : 0;
    ret += cum[max((ll)0, 18 - two[i])][max((ll)0, 18 - five[i])] - res;
    // cout << A[i] << "," << cum[18-two[i]][18-five[i]] - res << endl;
  }

  cout << ret / 2 << endl;

  return 0;
}


Error: TLE

Fixed submission:

#include <algorithm>
#include <cmath>
#include <functional>
#include <iomanip>
#include <iostream>
#include <list>
#include <map>
#include <queue>
#include <set>
#include <stack>
#include <string>
#include <vector>

#define MOD (1000000000 + 7)

using namespace std;

typedef long long ll;
const int MAX = 3000000;

long long fac[MAX], finv[MAX], inv[MAX];

long long modinv(long long a, long long m) {
  long long b = m, u = 1, v = 0;
  while (b) {
    long long t = a / b;
    a -= t * b;
    swap(a, b);
    u -= t * v;
    swap(u, v);
  }
  u %= m;
  if (u < 0)
    u += m;
  return u;
}

ll modpow(ll a, ll n, ll mod) {
  ll res = 1;
  while (n > 0) {
    if (n & 1)
      res = res * a % mod;
    a = a * a % mod;
    n >>= 1;
  }
  return res;
}

ll gcd(ll a, ll b) { return b ? gcd(b, a % b) : a; }
ll lcm(ll a, ll b) { return a / gcd(a, b) * b; }
// テーブルを作る前処理
void COMinit() {
  fac[0] = fac[1] = 1;
  finv[0] = finv[1] = 1;
  inv[1] = 1;
  for (int i = 2; i < MAX; i++) {
    fac[i] = fac[i - 1] * i % MOD;
    inv[i] = MOD - inv[MOD % i] * (MOD / i) % MOD;
    finv[i] = finv[i - 1] * inv[i] % MOD;
  }
}

// 二項係数計算
long long COM(int n, int k) {
  if (n < k)
    return 0;
  if (n < 0 || k < 0)
    return 0;
  return fac[n] * (finv[k] * finv[n - k] % MOD) % MOD;
}

map<ll, ll> prime; // 素因数分解でそれぞれの素数がいくつ出てきたかを保存するmap

// O(√n)
// 整列済み(mapはkeyで自動で整列される)
void prime_factorize(ll n) {
  if (n <= 1)
    return;
  ll l = sqrt(n);
  for (ll i = 2; i <= (ll)(l); i++) { //
    if (n % i == 0) {
      prime_factorize(i);
      prime_factorize(ll(n / i));
      return;
    }
  }
  // mapでは存在しないkeyの場合も自動で構築される
  prime[n]++;
  return;
}

// 約数列挙？
vector<ll> divisor;
void list_divisor(ll n) {
  divisor.clear();
  for (int i = 1; i * i <= n; i++) {
    if (n % i == 0) {
      divisor.push_back(i);
      if (i * i != n)
        divisor.push_back(n / i);
    }
  }
  sort(divisor.begin(), divisor.end());
}

vector<bool> Eratosthenes(ll N) {
  vector<bool> primes(N, true);
  vector<ll> ret;
  primes[0] = false;
  primes[1] = false;
  for (ll i = 2; pow(i, 2) <= N; i++) {
    if (primes[i]) {
      for (int j = 2; i * j <= N; j++)
        primes[i * j] = false;
      ret.push_back(primes[i]);
    }
  }
  return primes; // ret;
}

template <typename T>
bool next_combination(const T first, const T last, int k) {
  // ** Sample **
  // vector<int> v{1, 2, 3, 4, 5, 6, 7};
  // do {
  //     for (int i = 0; i < k; i++) {
  //         cout << v[i] << " ";
  //     }
  //     cout << "| ";
  //     for (int i = k; i < v.size(); i++) {
  //         cout << v[i] << " ";
  //     }
  //     cout << endl;
  // } while(next_combination(v.begin(), v.end(), k));

  const T subset = first + k;
  // empty container | k = 0 | k == n
  if (first == last || first == subset || last == subset) {
    return false;
  }
  T src = subset;
  while (first != src) {
    src--;
    if (*src < *(last - 1)) {
      T dest = subset;
      while (*src >= *dest) {
        dest++;
      }
      iter_swap(src, dest);
      rotate(src + 1, dest + 1, last);
      rotate(subset, subset + (last - dest) - 1, last);
      return true;
    }
  }
  // restore
  rotate(first, subset, last);
  return false;
}

// union by size + path having
class UnionFind {
public:
  vector<ll> par; // 各元の親を表す配列
  vector<ll> siz; // 素集合のサイズを表す配列(1 で初期化)

  // Constructor
  UnionFind(ll sz_) : par(sz_), siz(sz_, 1LL) {
    for (ll i = 0; i < sz_; ++i)
      par[i] = i; // 初期では親は自分自身
  }
  void init(ll sz_) {
    par.resize(sz_);
    siz.assign(sz_, 1LL); // resize だとなぜか初期化されなかった
    for (ll i = 0; i < sz_; ++i)
      par[i] = i; // 初期では親は自分自身
  }

  // Member Function
  // Find
  ll root(ll x) { // 根の検索
    while (par[x] != x) {
      x = par[x] = par[par[x]]; // x の親の親を x の親とする
    }
    return x;
  }

  // Union(Unite, Merge)
  bool merge(ll x, ll y) {
    x = root(x);
    y = root(y);
    if (x == y)
      return false;
    // merge technique（データ構造をマージするテク．小を大にくっつける）
    if (siz[x] < siz[y])
      swap(x, y);
    siz[x] += siz[y];
    par[y] = x;
    return true;
  }

  bool issame(ll x, ll y) { // 連結判定
    return root(x) == root(y);
  }

  ll size(ll x) { // 素集合のサイズ
    return siz[root(x)];
  }
};

int main() {
  ll N;
  cin >> N;

  vector<ll> A(N, 0);
  vector<ll> five(N, 0);
  vector<ll> two(N, 0);
  vector<vector<ll>> count(20, vector<ll>(20, 0));
  vector<vector<ll>> cum(20, vector<ll>(20, 0));
  for (ll i = 0; i < N; i++) {
    string d;
    cin >> d;
    ll small = 0;
    if (d.find('.') != string::npos) {
      small = d.length() - d.find('.') - 1;
      d.erase(remove(d.begin(), d.end(), '.'), d.end());
    }
    A[i] = stoll(d);
    for (int j = 1; j <= 9 - small; j++) {
      A[i] *= 10;
    }

    prime.clear();
    prime_factorize(A[i]);
    five[i] = prime[5] >= 20 ? 19 : prime[5];
    two[i] = prime[2] >= 20 ? 19 : prime[2];
    // cout << d << "->" << stoll(d) << " -> " << A[i] << ":" << two[i] <<"," <<
    // five[i] << endl;
    count[two[i]][five[i]]++;
  }
  // return 0;
  for (ll i = 19; i >= 0; i--) {
    for (ll j = 19; j >= 0; j--) {
      if (i == 19 && j == 19)
        cum[i][j] = count[i][j];
      else if (i == 19)
        cum[i][j] = cum[i][j + 1] + count[i][j];
      else if (j == 19)
        cum[i][j] = cum[i + 1][j] + count[i][j];
      else
        cum[i][j] =
            cum[i][j + 1] + cum[i + 1][j] + count[i][j] - cum[i + 1][j + 1];
    }
  }
  // cout << "test " << endl;
  ll ret = 0;
  for (ll i = 0; i < N; i++) {
    ll res = (two[i] >= 18 - two[i] && five[i] >= 18 - five[i]) ? 1 : 0;
    ret += cum[max((ll)0, 18 - two[i])][max((ll)0, 18 - five[i])] - res;
    // cout << A[i] << "," << cum[18-two[i]][18-five[i]] - res << endl;
  }

  cout << ret / 2 << endl;

  return 0;
}