PAT (Basic Level) Practice 1003 我要通过 Python解法

一、题目

让我们定义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

四、题目分析

这道题目原理很简单，但要写出最高效率的代码需要深入思考。
首先关于求素数。
1.求素数如果采用遍历的方法，则是i从2到sqrt(N)，判断N是否能被i整除，运行效率为192ms(详见代码1)。
2.判断素数的函数is_prime()和主程序如果共用一个primes列表类型参数，虽然耦合加大，但效率明显提高，从192ms提高到108ms(详见代码2)；
3.核心思想：素数的倍数一定不是素数 偶数一定不是素数，从108ms提高到90ms(详见代码3)。

五、Python代码

1.代码1：

import math
N = int(input())
def is_prime(N):
    if(N==1):
        return 0
    k = int(math.sqrt(N) + 1)
    for i in range(2, k):
        if(N % i==0):
            return 0
    return 1
primes = []

count =0

primes = [-1]
for i in range(1, N+1):
    if(is_prime(i)):
        primes.append(i)
        if(primes[-1]-primes[-2]==2):
            count+=1
print(count)

2.代码2：

import math

def isP(n,prime):
    k=int(math.sqrt(n))
    for i in prime:
        if i>k : break
        if n%i==0 : return(False)
    return(True)

n=int(input())
prime=[2,3]
sum=0

if n>4:
    for j in range(5,n+1):
        if isP(j,prime):
            prime.append(j)
            if prime[-1]-prime[-2]==2:
                sum+=1
print(sum)

3.代码3：

def easyprm(N):
    a = [0] * (N + 1)                                #初始化认为所有均为素数
    a[0] = a[1] = 1
    for i in range(2, N // 2 + 1):                    #查找所有素数的倍数
        for j in range(2, N // i + 1):
            a[i * j] = 1
    res=[2]
    for i in range(3, N+1, 2):                 #只寻找奇数
        if a[i]==0:
            res.append(i)
    return res
N=int(input())
A=easyprm(N)
sum=0
for i in range(len(A)-1):
    if A[i+1]-A[i]==2:
        sum+=1
print(sum)