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

#include <bits/stdc++.h>
#include <stdlib.h>
#include <sys/time.h>
#include <unistd.h>

using namespace std;

typedef long long ll;
typedef pair<ll, ll> P;
typedef pair<P, ll> T;
typedef pair<long double, ll> Ps;
typedef pair<ll, bool> Pb;

const ll INF = 1e18;
const ll fact_table = 3200008;
long double Pi = 3.1415926535897932384626;

priority_queue<ll> pql;
priority_queue<P> pqp;
priority_queue<P> bag;
// big priority queue
priority_queue<ll, vector<ll>, greater<ll>> pqls;
priority_queue<P, vector<P>, greater<P>> pqps;
// small priority queue
// top pop

ll dx[8] = {1, 0, -1, 0, 1, 1, -1, -1};
ll dy[8] = {0, 1, 0, -1, 1, -1, -1, 1};
// ↓,→,↑,←

/*
#define endl "\\n"
#ifdef ENJAPMA
#undef endl
#endif
*/
#define p(x) cout << x << endl;
#define el cout << endl;
#define pe(x) cout << x << " ";
#define ps(x) cout << fixed << setprecision(25) << x << endl;
#define pu(x) cout << (x);
#define pb push_back
#define lb lower_bound
#define ub upper_bound
#define CLEAR(a) a = decltype(a)();
#define pc(x) cout << x << ",";
#define rep(i, n) for (ll i = 0; i < (n); i++)

// const ll mod = 998244353ll;
const ll mod = 1000000007ll;

ll mypow(ll a, ll b, ll mod) {
  ll x = 1;
  while (b) {
    while (!(b & 1)) {
      (a *= a) %= mod;
      b >>= 1;
    }
    (x *= a) %= mod;
    b--;
  }
  return x;
}
void YES(bool cond) {
  if (cond) {
    p("YES");
  } else {
    p("NO");
  }
  return;
}
void Yes(bool cond) {
  if (cond) {
    p("Yes");
  } else {
    p("No");
  }
  return;
}
void line() {
  p("--------------------");
  return;
}

/*
ll fact[fact_table + 5], rfact[fact_table + 5];

void c3_init() {
    fact[0] = rfact[0] = 1;
    for (ll i = 1; i <= fact_table; i++) {
        fact[i] = (fact[i - 1] * i) % mod;
    }
    rfact[fact_table] = mypow(fact[fact_table], mod - 2, mod);
    for (ll i = fact_table; i >= 1; i--) {
        rfact[i - 1] = rfact[i] * i;
        rfact[i - 1] %= mod;
    }
    return;
}
ll c3(ll n, ll r) {
    return (((fact[n] * rfact[r]) % mod ) * rfact[n - r]) % mod;
}
*/
struct Timer {
  int64_t start;
  const int64_t CYCLES_PER_SEC = 2800000000;
  Timer() { reset(); }
  void reset() { start = getCycle(); }
  inline double get() { return (double)(getCycle() - start) / CYCLES_PER_SEC; }
  inline int64_t getCycle() {
    uint32_t low, high;
    __asm__ volatile("rdtsc" : "=a"(low), "=d"(high));
    return ((int64_t)low) | ((int64_t)high << 32);
  }
};

Timer timer;

bool multicase = false;

ll n, m, k, w, num, a, b, c, d, e, h, q, ans, sum, l;

typedef vector<ll> vec;
typedef vector<vector<ll>> mat;

ll table[105][105][105];

vec calc(string &s) {
  bool find = false;
  ll res = 0, digit = 0;
  for (int i = 0; i < s.size(); i++) {
    if (s[i] == '.') {
      find = true;
      continue;
    } else {
      res *= 10;
      res += s[i] - '0';
      if (find)
        digit++;
    }
  }
  ll two = 0, five = 0;
  while (res % 2 == 0) {
    res /= 2;
    two++;
  }
  while (res % 5 == 0) {
    res /= 5;
    five++;
  }
  two = min(18ll, two);
  five = min(18ll, five);
  vec n = {digit, two, five};
  return n;
}

void solve() {
  cin >> n;
  vec x(n);
  for (int i = 0; i < n; i++) {
    string s;
    cin >> s;
    vec ret = calc(s);
    for (int i1 = 0; i1 <= 9; i1++) {
      ll min2 = max(0ll, ret[0] - ret[1] + i1);
      ll min3 = max(0ll, ret[0] - ret[2] + i1);
      for (int i2 = min2; i2 <= 18; i2++) {
        for (int i3 = min3; i3 <= 18; i3++) {
          ans += table[i1][i2][i3];
        }
      }
    }
    table[ret[0]][ret[1]][ret[2]]++;
  }
  p(ans);
  return;
}

int main() {
  // init();
  ios::sync_with_stdio(false);
  cin.tie(nullptr);
  ll q, testcase = 1;
  if (multicase) {
    cin >> q;
  } else {
    q = 1;
  }
  while (q--) {
    // pu("Case ");pu("#");pu(testcase);pu(": ");
    solve();
    testcase++;
  }
  // solve();
  return 0;
}


Error: TLE

Fixed submission:

#include <bits/stdc++.h>
#include <stdlib.h>
#include <sys/time.h>
#include <unistd.h>

using namespace std;

typedef long long ll;
typedef pair<ll, ll> P;
typedef pair<P, ll> T;
typedef pair<long double, ll> Ps;
typedef pair<ll, bool> Pb;

const ll INF = 1e18;
const ll fact_table = 3200008;
long double Pi = 3.1415926535897932384626;

priority_queue<ll> pql;
priority_queue<P> pqp;
priority_queue<P> bag;
// big priority queue
priority_queue<ll, vector<ll>, greater<ll>> pqls;
priority_queue<P, vector<P>, greater<P>> pqps;
// small priority queue
// top pop

ll dx[8] = {1, 0, -1, 0, 1, 1, -1, -1};
ll dy[8] = {0, 1, 0, -1, 1, -1, -1, 1};
// ↓,→,↑,←

/*
#define endl "\\n"
#ifdef ENJAPMA
#undef endl
#endif
*/
#define p(x) cout << x << endl;
#define el cout << endl;
#define pe(x) cout << x << " ";
#define ps(x) cout << fixed << setprecision(25) << x << endl;
#define pu(x) cout << (x);
#define pb push_back
#define lb lower_bound
#define ub upper_bound
#define CLEAR(a) a = decltype(a)();
#define pc(x) cout << x << ",";
#define rep(i, n) for (ll i = 0; i < (n); i++)

// const ll mod = 998244353ll;
const ll mod = 1000000007ll;

ll mypow(ll a, ll b, ll mod) {
  ll x = 1;
  while (b) {
    while (!(b & 1)) {
      (a *= a) %= mod;
      b >>= 1;
    }
    (x *= a) %= mod;
    b--;
  }
  return x;
}
void YES(bool cond) {
  if (cond) {
    p("YES");
  } else {
    p("NO");
  }
  return;
}
void Yes(bool cond) {
  if (cond) {
    p("Yes");
  } else {
    p("No");
  }
  return;
}
void line() {
  p("--------------------");
  return;
}

/*
ll fact[fact_table + 5], rfact[fact_table + 5];

void c3_init() {
    fact[0] = rfact[0] = 1;
    for (ll i = 1; i <= fact_table; i++) {
        fact[i] = (fact[i - 1] * i) % mod;
    }
    rfact[fact_table] = mypow(fact[fact_table], mod - 2, mod);
    for (ll i = fact_table; i >= 1; i--) {
        rfact[i - 1] = rfact[i] * i;
        rfact[i - 1] %= mod;
    }
    return;
}
ll c3(ll n, ll r) {
    return (((fact[n] * rfact[r]) % mod ) * rfact[n - r]) % mod;
}
*/
struct Timer {
  int64_t start;
  const int64_t CYCLES_PER_SEC = 2800000000;
  Timer() { reset(); }
  void reset() { start = getCycle(); }
  inline double get() { return (double)(getCycle() - start) / CYCLES_PER_SEC; }
  inline int64_t getCycle() {
    uint32_t low, high;
    __asm__ volatile("rdtsc" : "=a"(low), "=d"(high));
    return ((int64_t)low) | ((int64_t)high << 32);
  }
};

Timer timer;

bool multicase = false;

ll n, m, k, w, num, a, b, c, d, e, h, q, ans, sum, l;

typedef vector<ll> vec;
typedef vector<vector<ll>> mat;

ll table[105][105][105];

vec calc(string &s) {
  bool find = false;
  ll res = 0, digit = 0;
  for (int i = 0; i < s.size(); i++) {
    if (s[i] == '.') {
      find = true;
      continue;
    } else {
      res *= 10;
      res += s[i] - '0';
      if (find)
        digit++;
    }
  }
  ll two = 0, five = 0;
  while (res % 2 == 0) {
    res /= 2;
    two++;
  }
  while (res % 5 == 0) {
    res /= 5;
    five++;
  }
  vec n = {digit, two, five};
  return n;
}

void solve() {
  cin >> n;
  vec x(n);
  for (int i = 0; i < n; i++) {
    string s;
    cin >> s;
    vec ret = calc(s);
    for (int i1 = 0; i1 <= 9; i1++) {
      for (int i2 = 0; i2 <= 52; i2++) {
        if (ret[0] + i1 > ret[1] + i2)
          continue;
        for (int i3 = 0; i3 <= 21; i3++) {
          if (ret[0] + i1 > ret[2] + i3)
            continue;
          ans += table[i1][i2][i3];
        }
      }
    }
    table[ret[0]][ret[1]][ret[2]]++;
  }
  p(ans);
  return;
}

int main() {
  // init();
  ios::sync_with_stdio(false);
  cin.tie(nullptr);
  ll q, testcase = 1;
  if (multicase) {
    cin >> q;
  } else {
    q = 1;
  }
  while (q--) {
    // pu("Case ");pu("#");pu(testcase);pu(": ");
    solve();
    testcase++;
  }
  // solve();
  return 0;
}