PAT BASIC LEVEL 1007. 素数对猜想 (20)

1007. 素数对猜想 (20)

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

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

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

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

输入样例：

20

输出样例：

4

Answer:

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

bool is_prime(int n) {
double max = sqrt(n);
for(int i = 2; i <= max; i ++)
if(n % i == 0)
return false;
return true;
}

int count_prime_pair(int n) {
if(n < 5)
return 0;
int odd;
if(n % 2 == 1)
odd = n / 2 + 1;
else
odd = n / 2;
bool result[odd];
result[0] = false;
result[1] = true;
for(int i = 2; i < odd; i ++)
if(is_prime(i * 2 + 1))
result[i] = true;
else
result[i] = false;
int correct = 0;
for(int i = 1; i < odd - 1; i ++)
if(result[i] && result[i + 1])
correct ++;
return correct;
}

int main() {
int num;
cin >> num;
cout << count_prime_pair(num) << endl;
}