PAT 1007 素数对猜想

https://pintia.cn/problem-sets/994805260223102976/problems/994805317546655744

让我们定义 d~n~ 为：d~n~ = p~n+1~ - p~n~，其中 p~i~ 是第i个素数。显然有 d~1~=1 且对于n&gt1有 d~n~ 是偶数。“素数对猜想”认为“存在无穷多对相邻且差为2的素数”。

现给定任意正整数N (< 10^5^)，请计算不超过N的满足猜想的素数对的个数。

输入格式：每个测试输入包含1个测试用例，给出正整数N。

输出格式：每个测试用例的输出占一行，不超过N的满足猜想的素数对的个数。

输入样例：

20

输出样例：

4

 代码：

#include <bits/stdc++.h>

using namespace std;

const int maxn=2e5+10;
int a[maxn];

int A(int x)
{
    for(int i=2; i*i<=x; i++)
    {
        if(x%i==0)
            return 0;
    }
    return 1;
}
int main()
{
    int cnt=0;
    for(int i=2;i<=maxn;i++)
    {
        if(A(i)==1)
        {
            cnt++;
            a[cnt]=i;
        }
    }
    int n;
    scanf("%d",&n);
    int ans=0;
    for(int i=2;i<=n;i++)
    {
            if(a[i]-a[i-1]==2&&a[i]<=n)
                //cout<<a[i]<<" ";
                ans++;
    }
    printf("%d",ans);
    return 0;
}