PAT乙级 1007 素数对猜想 (python)

1007 素数对猜想 (20 分)

让我们定义 d_n为：d_n = p_n+1 − p_n ，其中 p_i 是第i个素数。显然有d₁ =1，且对于n>1有 d_n 是偶数。“素数对猜想”认为“存在无穷多对相邻且差为2的素数”。
现给定任意正整数 N(<10⁵)，请计算不超过 N 的满足猜想的素数对的个数。

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

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

输入样例:

20

输出样例:

4

python代码：

# 耗时177ms
import math
count = 0
n = int(input())
def isPrime(n):
    if n < 2:
        return False
    for i in range(2,int(math.sqrt(n)) + 1):    # 经典的判断素数的方法：从3开始，若该数能被[2,它的平方根]区间的任一整数整除，则它不是素数
        if n % i == 0:
            return False
    return True
for i in range(3,n-1,2):   # 一开始只能得18分：测试点2答案错误：将for条件从for i in range(3,n+1,2)改成for i in range(3,n-1,2)之后成功：因为需要满足i+2<=n,则i<=n-2,即i<n-1
    if(isPrime(i) == True and isPrime(i+2) == True):
        count += 1
print(count)

看了柳婼的github后，发现她的代码更优，耗时更短：

# 耗时75ms:
N = int(input())
flag = [1]*(N+2)
flag[0] = flag[1] = 0
result = []
for i in range(2,N+1):
    if flag[i]:
        result.append(i)
        p = 2
        while i*p <= N:
            flag[i*p] = 0
            p += 1
c = 0
for i in range(len(result)-1):
    if (result[i+1] - result[i]) == 2:
        # print('result[i+1]',result[i+1])
        # print('result[i]',result[i])
        # print('--------')
        c += 1
print(c)

参考链接：[https://github.com/liuchuo/PAT/blob/master/BasicLevel_Python/1007%20素数对猜想%20（20%20分).py