从零开始的pat算法题Basic Level 1007

1007 素数对猜想 (20 分)

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

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

输入格式:

输入在一行给出正整数N。

输出格式:

在一行中输出不超过N的满足猜想的素数对的个数。

输入样例:

20

输出样例:

4

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

using namespace std;

//判断是否为素数
int sushu(int n) {
	for (int i = 2; i <= sqrt(n); i++) {
		if (n % i == 0)
			return 0;
	}
	return 1;
}

int main() {
	int N,sum=0;
	cin >> N;
	int a[100000];
	for (int i = 0; i < N; i++) {
		a[i] = 0;
	}
	//记录到N为止的素数
	int t = 0;
	for (int i = 3; i <= N; i++) {
		if (sushu(i)) {
			a[t] = i;
			t++;
		}
	}
	//比较前后2个素数差
	for (int j = 1; j < t; j++) {
		if (a[j] - a[j-1]== 2)
			sum++;
	}
	cout << sum;
}