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

#include <algorithm>
#include <bitset>
#include <cassert>
#include <complex>
#include <deque>
#include <exception>
#include <fstream>
#include <functional>
#include <iomanip>
#include <ios>
#include <iosfwd>
#include <iostream>
#include <istream>
#include <iterator>
#include <limits>
#include <list>
#include <locale>
#include <map>
#include <memory>
#include <new>
#include <numeric>
#include <ostream>
#include <queue>
#include <set>
#include <sstream>
#include <stack>
#include <stdexcept>
#include <streambuf>
#include <string>
#include <typeinfo>
#include <unordered_map>
#include <unordered_set>
#include <utility>
#include <valarray>
#include <vector>
#define _GLIBCXX_DEBUG
#define INF 100000010
#define PI 3.14159265359

using namespace std;
const int64_t mod = 1000000007;
const int64_t tmpMod = 998244353;

int num(int n) {
  // cout<<n<<endl;
  vector<int> x(n + 1);
  int num = n; // 素因数分解する変数num
  int sq = n;  // 素因数分解する変数num
  int aaa = n;
  int ans = 1;
  bool s = false;

  for (int i = 2; (i - 1) * (i - 1) <= n; i++) {
    while (num % i == 0) {
      // if(double(i)==double(sqrt(sq)))s=true;
      x.at(i)++;
      num /= i;
    }
  }
  if (num == n)
    x[n] = 1;
  for (int i = 2; i * i <= n; i++) {
    ans *= x.at(i) + 1;
  }
  ans *= x.at(n) + 1;
  // return ans;
  // if(s)cout<<aaa<<endl;
  // cout<<'s'<<endl;
  // cout<<ans<<endl;
  // return fmod(sqrt(n),0)==0 ? ans : ans*2;
  return ans;
}

int main() {

  int64_t N, a = 1, b, c;
  cin >> N;
  // vector<int64_t> A(N);
  for (int i = 2; i < N; i++) {
    a += 2;
    // cout<<i<<" "<<num(i)<<endl;
    for (int j = 2; j * j <= i; j++) {
      if (i % j == 0)
        a += 2;
    }
    if (fmod(sqrt(i), 1) == 0)
      a--;
    // a+=num(i);
  }
  // vector<vector<int> > A(N,vector<int>(M));
  // for(int i=0; i<N; i++){
  //    for(int j=0; j<M; j++){
  //        cin>>A[i][j];
  //    }
  //  }
  // sort(A.begin(),A.end());
  // for (int i = 0; i < N; i++) { maximum = max(maximum , A[i]);}
  // for (int i = 0; i < N; i++) { minimum = min(minimum , A[i]);}
  cout << a << endl;
}


Error: TLE

Fixed submission:

#include <algorithm>
#include <bitset>
#include <cassert>
#include <complex>
#include <deque>
#include <exception>
#include <fstream>
#include <functional>
#include <iomanip>
#include <ios>
#include <iosfwd>
#include <iostream>
#include <istream>
#include <iterator>
#include <limits>
#include <list>
#include <locale>
#include <map>
#include <memory>
#include <new>
#include <numeric>
#include <ostream>
#include <queue>
#include <set>
#include <sstream>
#include <stack>
#include <stdexcept>
#include <streambuf>
#include <string>
#include <typeinfo>
#include <unordered_map>
#include <unordered_set>
#include <utility>
#include <valarray>
#include <vector>
#define _GLIBCXX_DEBUG
#define INF 100000010
#define PI 3.14159265359

using namespace std;
const int64_t mod = 1000000007;
const int64_t tmpMod = 998244353;

int num(int n) {
  // cout<<n<<endl;
  vector<int> x(n + 1);
  int num = n; // 素因数分解する変数num
  int sq = n;  // 素因数分解する変数num
  int aaa = n;
  int ans = 1;
  bool s = false;

  for (int i = 2; i <= n; i++) {
    while (num % i == 0) {
      if (double(i) == double(sqrt(sq)))
        s = true;
      x.at(i)++;
      num /= i;
    }
  }

  for (int i = 2; i <= n; i++) {
    ans *= x.at(i) + 1;
  }
  // return ans;
  // if(s)cout<<aaa<<endl;
  // cout<<'s'<<endl;
  // cout<<ans<<endl;
  // return fmod(sqrt(n),0)==0 ? ans : ans*2;
  return ans;
}

int main() {

  int64_t N, a = 1, b, c;
  cin >> N;
  // vector<int64_t> A(N);
  for (int i = 2; i < N; i++) {
    // cout<<i<<" "<<num(i)<<endl;
    a += num(i);
  }
  // vector<vector<int> > A(N,vector<int>(M));
  // for(int i=0; i<N; i++){
  //    for(int j=0; j<M; j++){
  //        cin>>A[i][j];
  //    }
  //  }
  // sort(A.begin(),A.end());
  // for (int i = 0; i < N; i++) { maximum = max(maximum , A[i]);}
  // for (int i = 0; i < N; i++) { minimum = min(minimum , A[i]);}
  cout << a << endl;
}