PAT乙级练习题B1007. 素数对猜想

题目描述

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

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

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

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

输入样例：
20
输出样例：
4

题目解析

输入N；
找出所有不大于N的素数，存入vector；
-判断一个数是否为素数，即只能被1和本身整除；
在vector里遍历找出所有相差为2的素数对；

这里要注意素数判断程序的速度，循环到sqrt(N)即可;

代码

#include<iostream>
#include<vector>
#include<math.h>
using namespace std;

bool is_prime(int val)
{
    if (val == 1||val==2)
    {
        return 1;
    }
    if (val % 2 == 0)
    {
        return 0;
    }
    for (int i = 3; i <= sqrt(val); i+=2)
    {
        if (val%i == 0)
        {
            return 0;
        }
    }
    return 1;
}

int main()
{
    int N;
    vector<int> primes;
    int cnt = 0;
    cin >> N;
    for (int i = 1; i <= N; ++i)
    {
        if (is_prime(i))
        {
            primes.push_back(i);
        }
    }
    if (primes.empty())
    {
        cout << cnt << endl;
        return 0;
    }
    for (auto it = primes.begin(); it != (primes.end()-1); ++it)
    {
        if (*(it + 1) - *it == 2)
        {
            ++cnt;
        }
    }
    cout << cnt << endl;
    system("pause");
    return 0;
}