【PAT】 (Basic Level) Practice （中文） 1007 素数对猜想 (20 分)

思路：先构造一个判断是否为素数的函数，然后将n以内的素数加到列表，然后遍历列表。但是最后一个测试点会超时，主要是判断素数的函数运算量有点大

import math
def isprime(n):
    for i in range(2,int(math.sqrt(n))+1):
        if n % i == 0:
            return False
    return True

def getPlist(n):
    P = []
    for i in range(3,n+1):
        if isprime(i):
            P.append(i)
    return P

n = int(input())
Plst = getPlist(n)

cnt = 0
for i in range(len(Plst)-1):
    if Plst[i+1] - Plst[i] == 2:
        cnt += 1
print(cnt )
                   

对上面判断函数进行改进，在除以2到math.sqrt(n)+1之前先看看是不是偶数，也就是只对奇数进行判断，被除数是3到math.sqrt(n)+1之间的奇数。这样就减少运算量了（但是还是慢，最后一个测试点138ms）

import math
def prime(n):
    if n % 2 == 0:
        return False
    for i in range(3,int(math.sqrt(n))+1,2):
        if n % i == 0:
            return False
    return True
n=int(input())
l1=[i for i in range(3,n+1,2) if prime(i) and prime(i+2) and (i+2<=n)]
print(len(l1))

继续改进判断素数的函数。素数有一个分布规律，就是大于5的素数分布在6倍数的两侧，例如7，11，13，17，19。那么如果一个数除以6的模不是5或者1，那就不是素数。因此判断一个数是不是素数，只需要将它除以5到sqrt（n）之间的、在6两侧的数即可

import math
def prime(n):
    if n % 2==0:
        return False
    if n == 3:
        return True
    if n % 6 != 1 and n % 6 != 5:
        return False
    for i in range(5,int(math.sqrt(n))+1,6):
        if n%i==0 or n % (i+2) == 0:
            return False
    return True

n=int(input())
l1=[i for i in range(3,n+1,2) if prime(i) and prime(i+2) and (i+2<=n)]
print(len(l1))

但是最后一个测试点还是100多ms。。