PAT乙级 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 <stdio.h>
#include <math.h>
int main(){
    int num[10000], k = 0;
    int n;
    scanf("%d", &n);
    for(int i = 2; i <= n; i++){
        int flag = 0;
        for(int j = 2; j <= sqrt((double)i); j++)
            if(i % j == 0)
                flag = 1;
        if(flag == 0)
            num[k++] = i;
    }
    int cnt = 0;
    for(int i = 0; i < k - 1; i++){
        if(num[i+1] - num[i] == 2)
            cnt++;
    }
    printf("%d\n", cnt);
    return 0;
}