Cần gấp c++ mn ơi pls

`*` Nhận thấy rằng số nguyên tố thì không thể có bất kì một ước nguyên dương nào khác `1` và chính nó.

`->` `a_i` và `a_j` thoã mãn `<=>` 2 số đó có thể là số `1` và số còn lại là số nguyên tố.

Vậy ta chỉ cần xét nếu có số `1` trong mảng thì kết quả là số lượng số nguyên tố trong mảng, ngược lại thì kết quả sẽ là `0`

$\texttt{C++}$

#include <bits/stdc++.h>

using namespace std;
const int nmax = 1e7 + 5;

bool f[nmax + 1];
void sangnt()
{
    memset(f, true, sizeof(f));
    f[0] = false;
    f[1] = false;
    for (int i = 2; i * i <= nmax; ++i)
    {
        if (f[i] == true)
        {
            for (int j = i * i; j <= nmax; j += i)
            {
                f[j] = false;
            }
        }
    }
}
int n, a[1000005];
int main()
{
    freopen("BAI3.inp", "r", stdin);
    freopen("BAI3.out", "w", stdout);
    sangnt();
    ios_base::sync_with_stdio(0), cin.tie(0), cout.tie(0);
    cin >> n;
    int check = 0, nt = 0;
    for (int i = 1; i <= n; i++)
    {
        cin >> a[i];
        if (a[i] == 1)
            check++;
        if (f[a[i]])
            nt++;
    }
    cout << (long long) nt * check;
    return 0;
}