PAT乙级真题 1007 素数对猜想 C++实现

题目

让我们定义d​n为：d​n=pn+1−p​n，其中pi是第i个素数。显然有d1​​ =1，且对于n>1有dn是偶数。“素数对猜想”认为“存在无穷多对相邻且差为2的数”。
现给定任意正整数N(<10^5)，请计算不超过N的满足猜想的素数对的个数。
输入格式:
输入在一行给出正整数N。
输出格式:
在一行中输出不超过N的满足猜想的素数对的个数。
输入样例:
20
输出样例:
4

思路

每次只存储上一个素数，判断当前素数与其差值即可。

判断某数是否为素数时，需遍历小于等于其平方根的数，使用了cmath.h中的sqrt()函数。

注意：求平方根一开始使用的是pow(n, 0.5)，结果最后一个测试用例经常会超时，换成了sqrt(n)后立即通过。pow的效率比较低下，慎用！！！

算法

#include <iostream>
#include <cmath>
using namespace std;

bool isPrime(int n){
    //用pow(n, 0.5)会超时，sqrt效率大增
    for (int i=2; i<=sqrt(n); i++){
        if (n%i==0){
            return false;
        }
    }
    return true;
}

int main(){
    int n;
    cin >> n;

    int count = 0;
    int pre = 3;
    for (int i=5; i<=n; i++){
        if (isPrime(i)){
            if (i-pre == 2){
                count++;
            }
            pre = i;
        }
    }

    cout << count << endl;
    return 0;
}