Оптимизация кода

#include <stdio.h>
#include <stdlib.h>
#include <iostream>
#include <cmath>
#include <ctime>
using namespace std;
bool prost(long long x);
long gcd(long a, long b) {
    long nod = 1L;
    long tmp;
    if (a == 0L)
        return b;
    if (b == 0L)
        return a;
    if (a == b)
        return a;
    if (a == 1L || b == 1L)
        return 1L;
    while (a != 0 && b != 0) {
        if (a % 2L == 0L && b % 2L == 0L) {
            nod *= 2L;
            a /= 2L;
            b /= 2L;
            continue;
        }
        if (a % 2L == 0L && b % 2L != 0L) {
            a /= 2L;
            continue;
        }
        if (a % 2L != 0L && b % 2L == 0L) {
            b /= 2L;
            continue;
        }
        if (a > b) {
            tmp = a;
            a = b;
            b = tmp;
        }
        tmp = a;
        a = (b - a) / 2L;
        b = tmp;
    }
    if (a == 0)
        return nod * b;
    else
        return nod * a;
}
int main(int ac, char *av[])
{
    int X; long i, j, k = 1, g;
    cin >> X;
    if (prost(X)) { cout << "3"; }
    else {
        if (ac > 1) X = atol(av[1]);
 
        for (i = 1; i <= X / 2; i++) {
            if (X%i) continue;
            for (j = i + 1; j <= X; j++) {
                if (X%j) continue;
                g = gcd(j, i);
                if (g*X == i * j) {
 
                    k += 2;
                }
            }
        }
        printf("%ld\n", k);
    }
    //system("pause");
    return 0;
}
bool prost(long long x) {
    for (long long i = 2; i <= sqrt(x); i++)
        if (x%i == 0)
            return false;
    return true;
}

#include <iostream>
 
using namespace std;
 
int f(int n) {
    int k = 1, i = 2;
    while(i*i <= n) {
        int t = 0;
        while (n%i == 0) {
            t++;
            n/=i;
        }
        k *= t+1;
        i++;
    }
    if (n!=1) k*=2;
    return k;
}
 
int main() {
    int n;
    cin >> n;
    cout << f(n*n) << endl;
    return 0;
}