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

#include <bits/stdc++.h>
using namespace std;

#define mp make_pair
#define sz(x) (int)((x).size())
#define all(x) (x).begin(), (x).end()
#define FO(x)                                                                  \\
  {                                                                            \\
    freopen(#x ".in", "r", stdin);                                             \\
    freopen(#x ".out", "w", stdout);                                           \\
  }
#define eb emplace_back

typedef pair<int, int> pii;
typedef pair<long long, long long> pll;
typedef long long ll;
typedef unsigned long long ull;
typedef vector<int> vi;
typedef vector<vi> vvi;
typedef vector<ll> vll;
typedef vector<vll> vvll;
typedef vector<pii> vpii;
typedef vector<vpii> vvpii;
typedef vector<pll> vpll;
typedef long double ld;
typedef vector<ld> vld;
typedef unsigned int uint;

template <uint MD> struct ModInt {
  using M = ModInt;
  const static M G;
  uint v;
  ModInt(ll _v = 0) { set_v(uint(_v % MD + MD)); }
  M &set_v(uint _v) {
    v = (_v < MD) ? _v : _v - MD;
    return *this;
  }
  explicit operator bool() const { return v != 0; }
  M operator-() const { return M() - *this; }
  M operator+(const M &r) const { return M().set_v(v + r.v); }
  M operator-(const M &r) const { return M().set_v(v + MD - r.v); }
  M operator*(const M &r) const { return M().set_v(uint(ull(v) * r.v % MD)); }
  M operator/(const M &r) const { return *this * r.inv(); }
  M &operator+=(const M &r) { return *this = *this + r; }
  M &operator-=(const M &r) { return *this = *this - r; }
  M &operator*=(const M &r) { return *this = *this * r; }
  M &operator/=(const M &r) { return *this = *this / r; }
  bool operator==(const M &r) const { return v == r.v; }
  const bool operator<(const M &o) const { return v < o.v; }
  M pow(ll n) const {
    M x = *this, r = 1;
    while (n) {
      if (n & 1)
        r *= x;
      x *= x;
      n >>= 1;
    }
    return r;
  }
  M inv() const { return pow(MD - 2); }
  friend ostream &operator<<(ostream &os, const M &r) { return os << r.v; }
};

using Mint = ModInt<(int)1e9 + 7>;

int main() {
#ifdef BLAT
  freopen("stdin", "r", stdin);
  freopen("stderr", "w", stderr);
#endif

  ios::sync_with_stdio(false);
  cin.tie(0);
  cout.tie(0);
  cout.precision(12);
  srand(time(NULL));

  ll n;
  cin >> n;

  int ans = 0;

  for (int i = 2; 1ll * i * i <= n; ++i) {
    int e = 0;
    while (n % i == 0) {
      n /= i;
      ++e;
    }

    for (int j = 1; j <= e; ++j) {
      e -= j;
      ++ans;
    }
  }

  if (n != 1)
    ++ans;
  cout << ans << '\\n';
  return 0;
}

Error: TLE

Fixed submission:

#include <bits/stdc++.h>
using namespace std;

#define mp make_pair
#define sz(x) (int)((x).size())
#define all(x) (x).begin(), (x).end()
#define FO(x)                                                                  \\
  {                                                                            \\
    freopen(#x ".in", "r", stdin);                                             \\
    freopen(#x ".out", "w", stdout);                                           \\
  }
#define eb emplace_back

typedef pair<int, int> pii;
typedef pair<long long, long long> pll;
typedef long long ll;
typedef unsigned long long ull;
typedef vector<int> vi;
typedef vector<vi> vvi;
typedef vector<ll> vll;
typedef vector<vll> vvll;
typedef vector<pii> vpii;
typedef vector<vpii> vvpii;
typedef vector<pll> vpll;
typedef long double ld;
typedef vector<ld> vld;
typedef unsigned int uint;

template <uint MD> struct ModInt {
  using M = ModInt;
  const static M G;
  uint v;
  ModInt(ll _v = 0) { set_v(uint(_v % MD + MD)); }
  M &set_v(uint _v) {
    v = (_v < MD) ? _v : _v - MD;
    return *this;
  }
  explicit operator bool() const { return v != 0; }
  M operator-() const { return M() - *this; }
  M operator+(const M &r) const { return M().set_v(v + r.v); }
  M operator-(const M &r) const { return M().set_v(v + MD - r.v); }
  M operator*(const M &r) const { return M().set_v(uint(ull(v) * r.v % MD)); }
  M operator/(const M &r) const { return *this * r.inv(); }
  M &operator+=(const M &r) { return *this = *this + r; }
  M &operator-=(const M &r) { return *this = *this - r; }
  M &operator*=(const M &r) { return *this = *this * r; }
  M &operator/=(const M &r) { return *this = *this / r; }
  bool operator==(const M &r) const { return v == r.v; }
  const bool operator<(const M &o) const { return v < o.v; }
  M pow(ll n) const {
    M x = *this, r = 1;
    while (n) {
      if (n & 1)
        r *= x;
      x *= x;
      n >>= 1;
    }
    return r;
  }
  M inv() const { return pow(MD - 2); }
  friend ostream &operator<<(ostream &os, const M &r) { return os << r.v; }
};

using Mint = ModInt<(int)1e9 + 7>;

int main() {
#ifdef BLAT
  freopen("stdin", "r", stdin);
  freopen("stderr", "w", stderr);
#endif

  ios::sync_with_stdio(false);
  cin.tie(0);
  cout.tie(0);
  cout.precision(12);
  srand(time(NULL));

  ll n;
  cin >> n;

  int ans = 0;

  for (int i = 2; i * i <= n; ++i) {
    int e = 0;
    while (n % i == 0) {
      n /= i;
      ++e;
    }

    for (int j = 1; j <= e; ++j) {
      e -= j;
      ++ans;
    }
  }

  if (n != 1)
    ++ans;
  cout << ans << '\\n';
  return 0;
}