PAT 1007. 素数对猜想

1007. 素数对猜想 (20)

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

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

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

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

输入样例：

20

输出样例：

4

#include<iostream>
#include<math.h>
#include<vector>
using namespace std;
int main(){
     int N,n;
     cin>>N;
     vector<int> vi;
     for(int j=3;j<=N;j++){
         int flag=1; n=sqrt(j);
         for(int i=2;i<=n;i++){
          if(flag==0) break;
          if(j%i==0) flag=0;
         }
        if(flag==1) vi.push_back(j);
     }
     int cnt=0;
     for(auto b=vi.begin()+1;b<vi.end();b++)
     if(*b-*(b-1)==2) cnt++;
     cout<<cnt;
     return 0;
}