Problem p02722 (C++) — buggy submission, verdict: Runtime Error

#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
using namespace __gnu_pbds;
using namespace std;
typedef long long ll;
typedef long double ld;
typedef vector<ll> vl;
typedef pair<ll, ll> pll;
#define MOD 1000000007
#define INF 1000000000
#define mp make_pair
#define pb push_back
#define ss second
#define ff first
#define endl '\\n'
#define pl cout << endl;
void fast() {
  ios_base::sync_with_stdio(false);
  cin.tie(NULL);
  cout.tie(NULL);
}
const long long INFll = 1ll * INF * INF;
const long double PI =
    3.141592653589793238462643383279502884197169399375105820974944;
#define all(v) v.begin(), v.end()
#define rall(v) v.rbegin(), v.rend()
#define gcd(a, b) __gcd((a), (b))
#define lcm(a, b) ((a) * (b)) / gcd((a), (b))
#define debug(x) cout << x << endl;
#define sz(a) int((a).size())
mt19937 rng(chrono::steady_clock::now().time_since_epoch().count());
struct custom_hash {
  static uint64_t splitmix64(uint64_t x) {
    // http://xorshift.di.unimi.it/splitmix64.c
    x += 0x9e3779b97f4a7c15;
    x = (x ^ (x >> 30)) * 0xbf58476d1ce4e5b9;
    x = (x ^ (x >> 27)) * 0x94d049bb133111eb;
    return x ^ (x >> 31);
  }

  size_t operator()(uint64_t x) const {
    static const uint64_t FIXED_RANDOM =
        chrono::steady_clock::now().time_since_epoch().count();
    return splitmix64(x + FIXED_RANDOM);
  }
};
long long genrand(long long L, long long R) {
  return uniform_int_distribution<long long>(L, R)(rng);
}
// use brackets before operations if taking MOD
// use map only when necessary
// give long long to arrays and variables properly
int mul(int a, int b, int mod = MOD) { return int(a * 1ll * b % mod); }
int norm(int a, int mod = MOD) {
  while (a >= mod)
    a -= mod;
  while (a < 0)
    a += mod;
  // a = (a%mod+mod)%mod;
  return a;
}
int powmod(int x, int y, int mod = MOD) {
  int res = 1;
  while (y > 0) {
    if (y & 1)
      res = mul(res, x, mod);
    x = mul(x, x, mod);
    y = y >> 1;
  }
  return res;
}
int inv(int a, int mod = MOD) { return powmod(a, mod - 2); }
int main() {
  fast();
  long long n;
  cin >> n;
  unordered_map<long long, bool> m;
  for (long long k = 1; k * k <= n; k++) {
    if (n % k == 0) {
      long long A = k, B = n / k;
      long long X = A, Y = B;
      long long temp = n;
      if (X != 1) {
        while (temp % X == 0) {
          temp /= X;
        }
        if (temp == 1)
          m[X] = 1;
      }
      if (Y != 1) {
        temp = n;
        while (temp % Y == 0) {
          temp /= Y;
        }
        if (temp == 1)
          m[Y] = 1;
      }
      for (long long i = 1; i * i <= A - 1; i++) {
        if ((A - 1) % i == 0) {
          long long X = i, Y = (A - 1) / i;
          long long temp = B;
          if (X != 1) {
            while (temp % X == 0) {
              temp /= X;
            }
            if (temp == 1)
              m[X] = 1;
          }
          if (Y != 1) {
            temp = B;
            while (temp % Y == 0) {
              temp /= Y;
            }
            if (temp == 1)
              m[Y] = 1;
          }
        }
      }

      swap(A, B);

      for (long long i = 1; i * i <= A - 1; i++) {
        if ((A - 1) % i == 0) {
          long long X = i, Y = (A - 1) / i;
          long long temp = B;
          if (X != 1) {
            while (temp % X == 0) {
              temp /= X;
            }
            if (temp == 1)
              m[X] = 1;
          }
          if (Y != 1) {
            temp = B;
            while (temp % Y == 0) {
              temp /= Y;
            }
            if (temp == 1)
              m[Y] = 1;
          }
        }
      }
    }
  }
  for (long long k = 1; k * k <= n - 1; k++) {
    if ((n - 1) % k == 0) {
      m[k] = true;
      m[(n - 1) / k] = true;
    }
  }
  m[n] = true;
  long long res = m.size() - m[1];
  // for(auto e: m){
  // 	cerr<<e.ff<<' ';
  // }
  cout << res << endl;
  return 0;
}

Error: WA

Fixed submission:

#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
using namespace __gnu_pbds;
using namespace std;
typedef long long ll;
typedef long double ld;
typedef vector<ll> vl;
typedef pair<ll, ll> pll;
#define MOD 1000000007
#define INF 1000000000
#define mp make_pair
#define pb push_back
#define ss second
#define ff first
#define endl '\\n'
#define pl cout << endl;
void fast() {
  ios_base::sync_with_stdio(false);
  cin.tie(NULL);
  cout.tie(NULL);
}
const long long INFll = 1ll * INF * INF;
const long double PI =
    3.141592653589793238462643383279502884197169399375105820974944;
#define all(v) v.begin(), v.end()
#define rall(v) v.rbegin(), v.rend()
#define gcd(a, b) __gcd((a), (b))
#define lcm(a, b) ((a) * (b)) / gcd((a), (b))
#define debug(x) cout << x << endl;
#define sz(a) int((a).size())
mt19937 rng(chrono::steady_clock::now().time_since_epoch().count());
struct custom_hash {
  static uint64_t splitmix64(uint64_t x) {
    // http://xorshift.di.unimi.it/splitmix64.c
    x += 0x9e3779b97f4a7c15;
    x = (x ^ (x >> 30)) * 0xbf58476d1ce4e5b9;
    x = (x ^ (x >> 27)) * 0x94d049bb133111eb;
    return x ^ (x >> 31);
  }

  size_t operator()(uint64_t x) const {
    static const uint64_t FIXED_RANDOM =
        chrono::steady_clock::now().time_since_epoch().count();
    return splitmix64(x + FIXED_RANDOM);
  }
};
long long genrand(long long L, long long R) {
  return uniform_int_distribution<long long>(L, R)(rng);
}
// use brackets before operations if taking MOD
// use map only when necessary
// give long long to arrays and variables properly
int mul(int a, int b, int mod = MOD) { return int(a * 1ll * b % mod); }
int norm(int a, int mod = MOD) {
  while (a >= mod)
    a -= mod;
  while (a < 0)
    a += mod;
  // a = (a%mod+mod)%mod;
  return a;
}
int powmod(int x, int y, int mod = MOD) {
  int res = 1;
  while (y > 0) {
    if (y & 1)
      res = mul(res, x, mod);
    x = mul(x, x, mod);
    y = y >> 1;
  }
  return res;
}
int inv(int a, int mod = MOD) { return powmod(a, mod - 2); }
int main() {
#ifndef ONLINE_JUDGE
  freopen("/home/jam1729/Codes/in.txt", "r", stdin);
  freopen("/home/jam1729/Codes/out.txt", "w", stdout);
#endif
  fast();
  long long n;
  cin >> n;
  unordered_map<long long, bool> m;
  for (long long k = 1; k * k <= n; k++) {
    if (n % k == 0) {
      long long A = k, B = n / k;
      long long X = A, Y = B;
      long long temp = n;
      if (X != 1) {
        while (temp % X == 0) {
          temp /= X;
        }
        if (temp == 1)
          m[X] = 1;
      }
      if (Y != 1) {
        temp = n;
        while (temp % Y == 0) {
          temp /= Y;
        }
        if (temp == 1)
          m[Y] = 1;
      }
      for (long long i = 1; i * i <= A - 1; i++) {
        if ((A - 1) % i == 0) {
          long long X = i, Y = (A - 1) / i;
          long long temp = B;
          if (X != 1) {
            while (temp % X == 0) {
              temp /= X;
            }
            if (temp == 1)
              m[X] = 1;
          }
          if (Y != 1) {
            temp = B;
            while (temp % Y == 0) {
              temp /= Y;
            }
            if (temp == 1)
              m[Y] = 1;
          }
        }
      }

      swap(A, B);

      for (long long i = 1; i * i <= A - 1; i++) {
        if ((A - 1) % i == 0) {
          long long X = i, Y = (A - 1) / i;
          long long temp = B;
          if (X != 1) {
            while (temp % X == 0) {
              temp /= X;
            }
            if (temp == 1)
              m[X] = 1;
          }
          if (Y != 1) {
            temp = B;
            while (temp % Y == 0) {
              temp /= Y;
            }
            if (temp == 1)
              m[Y] = 1;
          }
        }
      }
    }
  }
  for (long long k = 1; k * k <= n - 1; k++) {
    if ((n - 1) % k == 0) {
      m[k] = true;
      m[(n - 1) / k] = true;
    }
  }
  m[n] = true;
  long long res = m.size() - m[1];
  // for(auto e: m){
  // 	cerr<<e.ff<<' ';
  // }
  cout << res << endl;
  return 0;
}