PAT乙级 1007素数对猜想 测试点5超时解决

1007 素数对猜想 (20分)

让我们定义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

判断某个数a是否为素数的循环条件为i<=sqrt(a)或i*i<=a就可以，如果循环条件为i<a，则测试点5超时

#include <iostream>
#include <algorithm>
#include <vector>
#include <math.h>

using namespace std;

bool ifvegetable(int a){//判断a是否为素数
    for(int i=2;i<=sqrt(a);i++){//循环条件若为i<a，则测试点5超时
        if(a%i==0){
            return false;
        }
    }
    return true;
}

int main() {
    int n;
    cin >> n;
    vector<int> num;
    for(int i=3;i<=n;i++){
        if(ifvegetable(i)){//若为素数
           num.push_back(i);
        }
    }
    int result=0;
    int len=num.size();
    for(int i=2;i<=len;i++){
        if(num[i-1]-num[i-2]==2){
            result++;
        }
    }
    cout << result << endl;
    return 0;
}

感觉PAT乙级里面好多自己去年C语言实验的作业题，我相信是作业题借鉴的PAT (lll￢ω￢)