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

#include <bits/stdc++.h>
#include <iomanip>
#include <math.h>
// #include <unistd.h>
// #include <stdio.h>
#define rep(i, x) for (ll i = 0; i < x; i++)
#define all(a) (a).begin(), (a).end()
using ll = long long;
using ld = long double;
using namespace std;
using dou = double;
const ll inf = 2147483647;
const ll INF = 1LL << 60;
const ld pi = 3.14159265358;
const ll mod = 1000000007;
typedef pair<ll, ll> P;
using graph = vector<vector<ll>>;
template <class T> inline bool chmax(T &a, T b) {
  if (a < b) {
    a = b;
    return true;
  }
  return false;
}
template <class T> inline bool chmin(T &a, T b) {
  if (a > b) {
    a = b;
    return true;
  }
  return false;
}
ll gcd(ll a, ll b) {
  if (a < b)
    swap(a, b);
  if (a % b == 0)
    return b;
  else
    return gcd(b, a % b);
}
// 素数判定O(sqrt(N))
bool isp(ll n) {
  bool res = true;
  if (n == 1)
    return false;
  else {
    for (ll i = 2; i * i <= n; i++) {
      if (n % i == 0) {
        res = false;
        break;
      }
    }
    return res;
  }
}

ll fac[510000], finv[510000], inv[510000];

// テーブルを作る前処理
void COMinit() {
  fac[0] = fac[1] = 1;
  finv[0] = finv[1] = 1;
  inv[1] = 1;
  for (ll i = 2; i < 510000; 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;
  }
}

// 二項係数計算
ll nCk(ll n, ll k) {
  if (n < k)
    return 0;
  if (n < 0 || k < 0)
    return 0;
  return fac[n] * (finv[k] * finv[n - k] % mod) % mod;
}
// nのm乗をMODで割ったあまりO(logN)
ll modpow(ll n, ll m, ll MOD) {
  ll res = 1;
  while (m) {
    if (m & 1)
      res = (res * n) % MOD;
    m >>= 1;
    n *= n;
    n %= MOD;
  }
  return res;
}

// printf("%.10f\\n", N);
/*  vector<ll> vec = {1,2,5,6,8,9,10};
    cout << binary_search(all(vec),5) << endl; -> true*/
/*
    vector<ll> vec = {1,2,5,6,8,9,10};
    auto n = upper_bound(all(vec),4);
    cout << *n << endl; -> 5*/

//    cout << fixed << setprecision(15);

int main() {
  ios::sync_with_stdio(false);
  cin.tie(0);
  ll N;
  cin >> N;
  map<ll, ll> ma;
  ll n = N;
  ll ans = 0;
  for (ll i = 2; i * i <= N; i++) {
    // cout << ans << endl;
    ll cnt = 0;
    if (n % i == 0) {
      while (n % i == 0) {
        n /= i;
        cnt++;
      }
    }
    // cout << n << "\\n";
    // cout << cnt << "\\n";
    if (cnt != 0) {
      for (ll i = 1; i <= 10; i++) {
        ll sa = i * (i + 1) / 2;
        if (cnt < sa) {
          ans += i - 1;
          break;
        }
      }
    }
  }
  if (n != 1) {
    ans++;
  }
  cout << ans << endl;
}

Error: TLE

Fixed submission:

#include <algorithm> // min, max, swap, sort, reverse, lower_bound, upper_bound
#include <bitset>    // bitset
#include <cctype>    // isupper, islower, isdigit, toupper, tolower
#include <cstdint>   // int64_t, int*_t
#include <cstdio>    // printf
#include <cstdlib>
#include <deque>    // deque
#include <iostream> // cout, endl, cin
#include <map>      // map
#include <math.h>
#include <queue>         // queue, priority_queue
#include <set>           // set
#include <stack>         // stack
#include <string>        // string, to_string, stoi
#include <tuple>         // tuple, make_tuple
#include <unordered_map> // unordered_map
#include <unordered_set> // unordered_set
#include <utility>       // pair, make_pair
#include <vector>        // vector
#define rep(i, x) for (int i = 0; i < x; i++)
#define all(a) (a).begin(), (a).end()
using namespace std;
using ld = long double;
using ll = long long;
const ll INF = 10010010010;
typedef pair<ll, ll> P;
const ll mod = 1000000007;
ll gcd(ll a, ll b) { // a>b
  if (a < b)
    swap(a, b);
  if (a % b == 0)
    return b;
  else
    return gcd(b, a % b);
}
bool isp(ll m) {
  bool res = true;
  if (m == 1)
    return false;
  else {
    for (ll i = 2; i * i <= m; i++) {
      if (m % i == 0) {
        res = false;
        break;
      }
    }
  }
  return res;
}
int main() {
  ll N;
  cin >> N;
  ll ans = 0;
  ll n = N;
  if (isp(N))
    cout << "1" << endl;
  else {
    map<ll, ll> vec;
    for (int i = 2; i <= N; i++) {
      while (N % i == 0) {
        vec[i]++;
        N /= i;
      }
      if (isp(N)) {
        vec[N]++;
        break;
      }
    }
    for (auto x : vec) {
      if (x.second > 0) {
        ll cnt = 1;

        while (x.second >= cnt) {

          x.second -= cnt;
          ans++;
          cnt++;
        }
      }
    }
    cout << ans << endl;
  }
}