【PAT 乙】1007 素数对猜想

题目

让我们定义d​n为：d​n=pn+1−p​n，其中p​i是第i个素数。显然有d1=1且对于n>1有dn是偶数。“素数对猜想”认为“存在无穷多对相邻且差为2的素数”。现给定任意正整数N(<100000)，请计算不超过N的满足猜想的素数对的个数。
输入格式:
输入在一行给出正整数N。
输出格式:
在一行中输出不超过N的满足猜想的素数对的个数。
输入样例:
20
输出样例:
4

分析

给定一个整数，判断出不超过这个数的素数，然后看看有多少对相差2的素数对就可以了。
虽然逻辑是很简单的，但是运行的时候一直显示超时，看了很多别人的程序，明白时间主要花在了，素数判断的那一部分，所以要想缩短时间，就要修改判断素数的程序。
任意一个合数都可分解为素数因子的乘积，观察素数的分布可以发现：除 2,3 以外的素数必定分布在 6k (k为大于1的整数) 的两侧。6k % 6 = 0, (6k+2) % 2= 0,(6k+3) %3=0,(6k+4)%2=0,所以2,3外的素数形式只能写成 6k+1 或 6k-1的形式。据此，我们可以缩小因子范围。

程序

from math import sqrt
n=int