C++学习之路 | PTA乙级—— 1007 素数对猜想 (20分)（精简）

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

#include<iostream>
#include<math.h>
using namespace std;
bool judge(int n)//判断素数，常用
{
	if (n < 2) 
		return false;
	for (int i = 2; i <= sqrt(n); i++)
	{
		if (n % i == 0) 
			return false;
	}
	return true;
}
int main()
{
	int n, sum = 0;
	int vist[10001] = { 0 };//标记是否为素数
	cin >> n;
	for (int i = 1; i <= n; i++)
	{
		if (judge(i))//判断
		{
			vist[i] = 1;//如果是素数，标记为1
		}
	}
	for (int i = 1; i <= n; i++)
	{
		if (vist[i] == 1 && vist[i + 2] == 1 && i + 2 <= n)//判断相邻且相差2的数是否为素数，i+2必须小于等于n
		{
			sum++;
		}
	}
	cout << sum;
}