Problem p02900 (C++)

Buggy submission (status: Time Limit Exceeded):
#include <algorithm>
#include <cstdio>
#include <iomanip>
#include <iostream>
#include <map>
#include <math.h>
#include <queue>
#include <set>
#include <stack>
#include <string.h>
#include <string>
#include <tuple>
#include <vector>

#define FOR(i, a, b) for (int i = a; i < b; i++)
#define REP(i, a, b) for (int i = a; i > b; i--)
#define vint vector<int>
#define vvint vector<vint>
#define CST(x) cout << fixed << setprecision(x) // 小数点以下の桁数指定
#define ct(a) cout << a << endl
static const double pi = 3.141592653589793;
using namespace std;
typedef long long ll;
typedef pair<int, int> P;
const ll MOD = 1e9 + 7;
const int INF = 1e9 + 7;

class unionfind {
public:
  vector<int> par;
  vector<int> rnk;

  unionfind(int N) : par(N), rnk(N) { // コンストラクタ
    for (int i = 0; i < N; i++) {
      par[i] = i;
      rnk[i] = 0;
    }
  }

  int root(int x) { // データｘが属する木の根を再帰で得る:root(x)={ｘの木の根}
    if (par[x] == x)
      return x;
    return par[x] = root(par[x]);
  }

  void unite(int x, int y) {
    int rx = root(x);
    int ry = root(y);
    if (rx == ry)
      return; // ｘとｙの根が同じ（＝同じ木にある）時はそのまま
    if (rnk[rx] < rnk[ry]) {
      par[rx] =
          ry; // xとyの根が同じでない（＝同じ木にない）時：ｘの根をｒｘをｙの根につける
    } else if (rnk[rx] > rnk[ry]) {
      par[ry] = rx;
    } else {
      par[rx] = ry;
      rnk[ry]++;
    }
  }

  bool same(int x, int y) {
    int rx = root(x);
    int ry = root(y);
    return rx == ry;
  }
};

ll gcd(ll x, ll y) {
  if (y == 0)
    return x;
  return gcd(y, x % y);
}

int main() {
  cin.tie(0);
  cout.tie(0);
  ios::sync_with_stdio(false);
  ll A, B;
  cin >> A >> B;
  ll G = gcd(A, B);
  ll ans = 1;

  for (int i = 2; i * i <= G; i++) {
    if (G % i == 0) {
      ans++;
      while (G % i == 0)
        G /= i;
    }
  }
  if (G > 1)
    ans++;

  cout << ans << endl;

  return 0;
}


Fixed submission:
#include <algorithm>
#include <cstdio>
#include <iomanip>
#include <iostream>
#include <map>
#include <math.h>
#include <queue>
#include <set>
#include <stack>
#include <string.h>
#include <string>
#include <tuple>
#include <vector>

#define FOR(i, a, b) for (int i = a; i < b; i++)
#define REP(i, a, b) for (int i = a; i > b; i--)
#define vint vector<int>
#define vvint vector<vint>
#define CST(x) cout << fixed << setprecision(x) // 小数点以下の桁数指定
#define ct(a) cout << a << endl
static const double pi = 3.141592653589793;
using namespace std;
typedef long long ll;
typedef pair<int, int> P;
const ll MOD = 1e9 + 7;
const int INF = 1e9 + 7;

class unionfind {
public:
  vector<int> par;
  vector<int> rnk;

  unionfind(int N) : par(N), rnk(N) { // コンストラクタ
    for (int i = 0; i < N; i++) {
      par[i] = i;
      rnk[i] = 0;
    }
  }

  int root(int x) { // データｘが属する木の根を再帰で得る:root(x)={ｘの木の根}
    if (par[x] == x)
      return x;
    return par[x] = root(par[x]);
  }

  void unite(int x, int y) {
    int rx = root(x);
    int ry = root(y);
    if (rx == ry)
      return; // ｘとｙの根が同じ（＝同じ木にある）時はそのまま
    if (rnk[rx] < rnk[ry]) {
      par[rx] =
          ry; // xとyの根が同じでない（＝同じ木にない）時：ｘの根をｒｘをｙの根につける
    } else if (rnk[rx] > rnk[ry]) {
      par[ry] = rx;
    } else {
      par[rx] = ry;
      rnk[ry]++;
    }
  }

  bool same(int x, int y) {
    int rx = root(x);
    int ry = root(y);
    return rx == ry;
  }
};

ll gcd(ll x, ll y) {
  if (y == 0)
    return x;
  return gcd(y, x % y);
}

int main() {
  cin.tie(0);
  cout.tie(0);
  ios::sync_with_stdio(false);
  ll A, B;
  cin >> A >> B;
  ll G = gcd(A, B);
  ll ans = 1;

  for (ll i = 2; i * i <= G; i++) {
    if (G % i == 0) {
      ans++;
      while (G % i == 0)
        G /= i;
    }
  }
  if (G > 1)
    ans++;

  cout << ans << endl;

  return 0;
}