【ACM】孪生素数问题

 
#include <iostream>
#include <cmath>
#include <string>
#include <cstdio>

using namespace std;

bool isPrime(int n){

    if (n<=1)
    {
        return false;
    }
    if (n==2)
    {
        return true;
    }
    for (int i = 2 ; i <= floor(sqrt(n*1.0)+0.5); i++)
    {
        if (n%i==0)
        {
            return false;
        }
    }
    return true;

}

int main(){

    int a[1000010] = {0};

    for (int k = 2 ; k < 1000010 ; k++)
    {
        if (isPrime(k))
        {
            a[k] = 1;
        }
    }

    int n,m;
    cin>>n;
    while(n--){

        cin>>m;
        int count = 0;
        for (int i = 2 ; i <= m ; i++)
        {
            if (a[i]==1)
            {
                if (i+1<=m && a[i+1]==1)
                {
                    count++;
                }
                if (i+2<=m && a[i+2]==1)
                {
                    count++;
                }
            }
        }
        cout<<count<<endl;
    }

    return 0;
}