pat 素數對猜想

讓我們定義d​n​​為：d​n​​=p​n+1​​−p​n​​，其中p​i​​是第i個素數。顯然有d​1​​=1，且對於n>1有d​n​​是偶數。“素數對猜想”認為“存在無窮多對相鄰且差為2的素數”。

現給定任意正整數N(<10​5​​)，請計算不超過N的滿足猜想的素數對的個數。

輸入格式:

輸入在一行給出正整數N。

輸出格式:

在一行中輸出不超過N的滿足猜想的素數對的個數。

輸入樣例:

20

輸出樣例:

4

#include <iostream>
#include <math.h>
using namespace std;
int a[100001];
int prime(int n)
{
    int k=sqrt(n);
    int i;
    if(n==0||n==1)
        return 0;
    for(i=2;i<=k;i++)
    {
        if(n%i==0)
        {
            return 0;
        }
    }
    return 1;
}
int main()
{
    int n;
    cin>>n;
    int i;
    int j=0;
    for(i=2;i<=n;i++)
    {
        if(prime(i))
            a[j++]=i;
        else;
    }
    int s=0;
    for(i=1;i<=j;i++)
    {
        if(a[i]-a[i-1]==2)
            s++;
    }
    cout<<s<<endl;
}