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

#include <bits/stdc++.h>
using namespace std;
#define ll long long
#define fast                                                                   \\
  ios_base::sync_with_stdio(false);                                            \\
  cin.tie(NULL)
#define endl "\\n"
#define pb push_back
#define mod 1000000007
#define MAX 200009
#define MAXN 1000001
int dx[] = {1, -1, 0, 0};
int dy[] = {0, 0, 1, -1};
int dx8[] = {1, -1, 0, 0, 1, 1, -1, -1};
int dy8[] = {0, 0, 1, -1, 1, -1, 1, -1};
int spf[MAXN];
// ll binexp(ll a, ll b, ll m) {
// 	a %= m;
// 	ll res = 1;
// 	while (b > 0) {
// 		if (b & 1)
// 			res = res * a % m;
// 		a = a * a % m;
// 		b >>= 1;
// 	}
// 	return res;
// }
// ll modinvfermat(ll a, ll m)
// {
// 	return binexp(a, m - 2, m);
// }
// void task(bool flag)
// {
// 	if (flag)
// 		cout << "YES\\n";
// 	else
// 		cout << "NO\\n";
// }
// ll lcm(ll a, ll b)
// {
// 	return ((1LL * a * b) / (__gcd(a, b)));
// }
// // int spf[MAXN];
// void sieve()
// {
// 	int i,j;
// 	for(i=0;i<MAXN;i++)
// 	spf[i]=i;
// 	for(i=4;i<MAXN;i+=2)
// 	spf[i]=2;
// 	for(i=3;i*i<MAXN;i++)
// 	{
// 		if(spf[i]==i)
// 		{
// 			for(j=i*i;j<MAXN;j+=i)
// 			{
// 				if(spf[j]==j)spf[j]=i;
// 			}
// 		}
// 	}
// }
// int fac(int n)
// {
//     if(n==1)return 1;
//     int ans = 1;
//     int y = spf[n];
//     int ctr = 1;
//     int j = n / y;
//     while (j != 1) {
//         if (spf[j] == y)
//             ctr ++;
//         else {
//             y = spf[j];
//             ans = ans*(ctr + 1);
//             ctr = 1;
//         }
//         j = j / spf[j];
//     }
//     ans = ans *(ctr + 1);
//     return ans;
// }

// bool sortbysec(const pair<int,int> &a,
//               const pair<int,int> &b)
// {
//     return (a.second < b.second);
// }
// bool sortbyme(const pair<int,int> &a,
//               const pair<int,int> &b){
// 	return (a.first+a.second > b.first+b.second);

// }
// struct test
// {
//   int x, y, z;
// };

void sieve() {
  int i, j;
  for (i = 0; i < MAXN; i++)
    spf[i] = i;
  for (i = 4; i < MAXN; i += 2)
    spf[i] = 2;
  for (i = 3; i * i < MAXN; i++) {
    if (spf[i] == i) {
      for (j = i * i; j < MAXN; j += i) {
        if (spf[j] == j)
          spf[j] = i;
      }
    }
  }
}
ll fac(ll n) {

  if (n == 1)
    return 1;

  ll ans = 1, y = spf[n];

  ll ctr = 1, j = n / y;
  while (j != 1) {
    if (spf[j] == y)
      ctr++;
    else {
      y = spf[j];
      ans = ans * (ctr + 1);
      ctr = 1;
    }
    j = j / spf[j];
  }
  ans = ans * (ctr + 1);
  return ans;
}
// ll n;
// ll m;
// ll k;

// vector<pair<ll,ll>> v;
// ll vis[500000];
// ll dist[500000];

// void dfs(ll node){
//     vis[node]=1;
//     for(auto i:v[node]){
//         if(vis[i]==0){
//             dfs(i);

//         }
//     }
// }
// void bfs(ll node){
//     queue<ll> q;
//     vis[node]=1;
//     dist[node]=0;

//     q.push(node);
//     while(!q.empty()){
//         ll curr=q.front();
//         q.pop();

//         for(auto i:v[curr]){
//             if(vis[i]==0){
//                 vis[i]=1;
//                 dist[i]=dist[curr]+1;
//                 q.push(i);

//             }
//         }

//     }
// }

// string alpha="abcdefghijklmnopqrstuvwxyz";
// int dx[]={1,-1,0,0};
// int dy[]={0,0,1,-1};
// int dx8[]={1,-1,0,0,1,1,-1,-1};
// int dy8[]={0,0,1,-1,1,-1,1,-1};
//  int spf[MAXN];
//  void sieve()
//  {
//  	int i,j;
//  	for(i=0;i<MAXN;i++)
//  	spf[i]=i;
//  	for(i=4;i<MAXN;i+=2)
//  	spf[i]=2;
//  	for(i=3;i*i<MAXN;i++)
//  	{
//  		if(spf[i]==i)
//  		{
//  			for(j=i*i;j<MAXN;j+=i)
//  			{
//  				if(spf[j]==j)spf[j]=i;
//  			}
//  		}
//  	}
//  }
//  int fac(int n)
//  {
//      if(n==1)return 1;
//      int ans = 1;
//      int y = spf[n];
//      int ctr = 1;
//      int j = n / y;
//      while (j != 1) {
//          if (spf[j] == y)
//              ctr ++;
//          else {
//              y = spf[j];
//              ans = ans*(ctr + 1);
//              ctr = 1;
//          }
//          j = j / spf[j];
//      }
//      ans = ans *(ctr + 1);
//      return ans;
//  }

void solve() {

  ll n;
  cin >> n;
  ll ans = 0;
  sieve();
  for (ll i = 1; i <= n - 1; i++) {
    // ll temp=n-i;
    // ll cc=0;
    // vector<ll>fact;
    //   ll f=temp;
    //       ll loc=1;

    // for(ll j=2;j*j<=temp;j++){
    //     if(temp%j==0){

    //         ll cnt=0;

    //         while(f%j==0&&f!=0){
    //             f=f/j;

    //             cnt++;

    //         }
    //         // fact.pb(cnt);
    //         loc*=(cnt+1);

    //     }
    // }
    // // if(f!=1){
    // //     fact.pb(1);

    // // }

    // // for(auto i:fact){
    // //     loc*=(i+1);

    // // }
    // if(f!=1){
    //     loc*=(2);

    // }
    ans += fac(n - i);
  }
  cout << ans << endl;

  return;
}

int main() {
#ifndef ONLINE_JUDGE
  // for getting input from input.txt
  freopen("input.txt", "r", stdin);
  // for writing output to output.txt
  freopen("output.txt", "w", stdout);
#endif
  fast;

  ll t;
  t = 1;
  // cin>>t;
  while (t--) {
    solve();
  }

#ifndef ONLINE_JUDGE
  cout << "\\nTime Elapsed: " << 1.0 * clock() / CLOCKS_PER_SEC << " sec\\n";
#endif
  return 0;
}

Error: TLE

Fixed submission:

#include <bits/stdc++.h>
using namespace std;
#define ll long long
#define fast                                                                   \\
  ios_base::sync_with_stdio(false);                                            \\
  cin.tie(NULL)
#define endl "\\n"
#define pb push_back
#define mod 1000000007
#define MAX 200009
int dx[] = {1, -1, 0, 0};
int dy[] = {0, 0, 1, -1};
int dx8[] = {1, -1, 0, 0, 1, 1, -1, -1};
int dy8[] = {0, 0, 1, -1, 1, -1, 1, -1};

ll binexp(ll a, ll b, ll m) {
  a %= m;
  ll res = 1;
  while (b > 0) {
    if (b & 1)
      res = res * a % m;
    a = a * a % m;
    b >>= 1;
  }
  return res;
}
ll modinvfermat(ll a, ll m) { return binexp(a, m - 2, m); }
void task(bool flag) {
  if (flag)
    cout << "YES\\n";
  else
    cout << "NO\\n";
}
ll lcm(ll a, ll b) { return ((1LL * a * b) / (__gcd(a, b))); }

bool sortbysec(const pair<int, int> &a, const pair<int, int> &b) {
  return (a.second < b.second);
}
bool sortbyme(const pair<int, int> &a, const pair<int, int> &b) {
  return (a.first + a.second > b.first + b.second);
}
struct test {
  int x, y, z;
};
// ll n;
// ll m;
// ll k;

// vector<pair<ll,ll>> v;
// ll vis[500000];
// ll dist[500000];

// void dfs(ll node){
//     vis[node]=1;
//     for(auto i:v[node]){
//         if(vis[i]==0){
//             dfs(i);

//         }
//     }
// }
// void bfs(ll node){
//     queue<ll> q;
//     vis[node]=1;
//     dist[node]=0;

//     q.push(node);
//     while(!q.empty()){
//         ll curr=q.front();
//         q.pop();

//         for(auto i:v[curr]){
//             if(vis[i]==0){
//                 vis[i]=1;
//                 dist[i]=dist[curr]+1;
//                 q.push(i);

//             }
//         }

//     }
// }

void solve() {

  ll n;
  cin >> n;
  ll ans = 0;
  for (ll i = 1; i <= n - 1; i++) {
    ll temp = n - i;
    ll cc = 0;
    vector<ll> fact;
    ll f = temp;
    ll loc = 1;

    for (ll j = 2; j * j <= temp; j++) {
      if (temp % j == 0) {

        ll cnt = 0;

        while (f % j == 0 && f != 0) {
          f = f / j;

          cnt++;
        }
        // fact.pb(cnt);
        loc *= (cnt + 1);
      }
    }
    // if(f!=1){
    //     fact.pb(1);

    // }

    // for(auto i:fact){
    //     loc*=(i+1);

    // }
    if (f != 1) {
      loc *= (2);
    }
    ans += loc;
  }
  cout << ans << endl;

  return;
}

int main() {
#ifndef ONLINE_JUDGE
  // for getting input from input.txt
  freopen("input.txt", "r", stdin);
  // for writing output to output.txt
  freopen("output.txt", "w", stdout);
#endif
  fast;

  ll t;
  t = 1;
  // cin>>t;
  while (t--) {
    solve();
  }

#ifndef ONLINE_JUDGE
  cout << "\\nTime Elapsed: " << 1.0 * clock() / CLOCKS_PER_SEC << " sec\\n";
#endif
  return 0;
}