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

#include <cmath>
#include <cstdio>
#include <cstdlib>
#include <cstring>

#include <algorithm>
#include <bitset>
#include <complex>
#include <iostream>
#include <map>
#include <numeric>
#include <queue>
#include <set>
#include <stack>
#include <string>
#include <unordered_map>
#include <unordered_set>
#include <vector>

#include <cassert>
#include <functional>

typedef long long ll;
using namespace std;

#ifndef LOCAL
#define debug(...) ;
#else
#define debug(...)                                                             \\
  cerr << __LINE__ << " : " << #__VA_ARGS__ << " = " << _tostr(__VA_ARGS__)    \\
       << endl;

template <typename T> ostream &operator<<(ostream &out, const vector<T> &v);

template <typename T1, typename T2>
ostream &operator<<(ostream &out, const pair<T1, T2> &p) {
  out << "{" << p.first << ", " << p.second << "}";
  return out;
}

template <typename T> ostream &operator<<(ostream &out, const vector<T> &v) {
  out << '{';
  for (const T &item : v)
    out << item << ", ";
  out << "\\b\\b}";
  return out;
}

void _tostr_rec(ostringstream &oss) { oss << "\\b\\b \\b"; }

template <typename Head, typename... Tail>
void _tostr_rec(ostringstream &oss, Head &&head, Tail &&...tail) {
  oss << head << ", ";
  _tostr_rec(oss, forward<Tail>(tail)...);
}

template <typename... T> string _tostr(T &&...args) {
  ostringstream oss;
  int size = sizeof...(args);
  if (size > 1)
    oss << "{";
  _tostr_rec(oss, forward<T>(args)...);
  if (size > 1)
    oss << "}";
  return oss.str();
}
#endif

#define mod 1000000007              // 1e9+7(prime number)
#define INF 1000000000              // 1e9
#define LLINF 2000000000000000000LL // 2e18
#define SIZE 200010

int main() {
  ll N;
  int ans = 0;

  cin >> N;

  for (ll i = 2; i * i <= N; i++) {
    int counter = 0;
    while (N % i == 0) {
      N /= i;
      counter++;
    }
    for (int j = 1; j * (j + 1) / 2 <= counter; j++) {
      ans++;
    }
  }

  if (N > 1)
    ans++;

  cout << ans << endl;

  return 0;
}

Error: TLE

Fixed submission:

#include <cmath>
#include <cstdio>
#include <cstdlib>
#include <cstring>

#include <algorithm>
#include <bitset>
#include <complex>
#include <iostream>
#include <map>
#include <numeric>
#include <queue>
#include <set>
#include <stack>
#include <string>
#include <unordered_map>
#include <unordered_set>
#include <vector>

#include <cassert>
#include <functional>

typedef long long ll;
using namespace std;

#ifndef LOCAL
#define debug(...) ;
#else
#define debug(...)                                                             \\
  cerr << __LINE__ << " : " << #__VA_ARGS__ << " = " << _tostr(__VA_ARGS__)    \\
       << endl;

template <typename T> ostream &operator<<(ostream &out, const vector<T> &v);

template <typename T1, typename T2>
ostream &operator<<(ostream &out, const pair<T1, T2> &p) {
  out << "{" << p.first << ", " << p.second << "}";
  return out;
}

template <typename T> ostream &operator<<(ostream &out, const vector<T> &v) {
  out << '{';
  for (const T &item : v)
    out << item << ", ";
  out << "\\b\\b}";
  return out;
}

void _tostr_rec(ostringstream &oss) { oss << "\\b\\b \\b"; }

template <typename Head, typename... Tail>
void _tostr_rec(ostringstream &oss, Head &&head, Tail &&...tail) {
  oss << head << ", ";
  _tostr_rec(oss, forward<Tail>(tail)...);
}

template <typename... T> string _tostr(T &&...args) {
  ostringstream oss;
  int size = sizeof...(args);
  if (size > 1)
    oss << "{";
  _tostr_rec(oss, forward<T>(args)...);
  if (size > 1)
    oss << "}";
  return oss.str();
}
#endif

#define mod 1000000007              // 1e9+7(prime number)
#define INF 1000000000              // 1e9
#define LLINF 2000000000000000000LL // 2e18
#define SIZE 200010

int main() {
  ll N;
  int ans = 0;

  cin >> N;

  for (int i = 2; i * i <= N; i++) {
    int counter = 0;
    while (N % i == 0) {
      N /= i;
      counter++;
    }
    for (int j = 1; j * (j + 1) / 2 <= counter; j++) {
      debug(i, j);
      ans++;
    }
  }

  if (N > 1)
    ans++;

  cout << ans << endl;

  return 0;
}