1007 素数对猜想 Python实现

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

问题分析：

解题思路很简单，找出n以内的素数，再寻找相邻且差为二的素数对

 在实现的过程中，遇到一个十分麻烦的问题，算法超出时间限制，优化方向当然是素数判断算法的优化。所以之后会着重讲解素数判断算法的优化（需要代码的同学直接拉到最后）。

素数判断算法一：

直接利用素数定义编写的算法：

def prime(a):
    if a<=1:
        return False
    for i in range(2,a-1):
        if a%i==0:
            return False
    return True

利用该算法输出100000（十万）以内素数，程序以及运行时间如下：

程序：

import time
def prime(a):
    if a<=1:
        return False
    for i in range(2,a-1):
        if a%i==0:
            return False
    return True
strat = time.clock()
for i in range(2,100000):
    if prime(i):
        print(i)
end = time.clock()
print(end-strat)

时间：

163.62294357333334

优化一：

在判断素数时并不需要计算2—n-1的值，实际上只需要计算2—，第一步优化大家都比较清楚，经过这步优化，减少了的计算量。我们来看看程序，和运行时间。

程序：

import time
import math
def prime(a):
    if a<=1:
        return False
    for i in range(2,int(math.sqrt(a)+1)):
        if a%i==0:
            return False
    return True
strat = time.clock()
for i in range(2,100000):
    if prime(i):
        print(i)
end = time.clock()
print(end-strat)

时间：

59.89916501333333

可以看出时间少了多于一半。

优化二：

我们可以看到，除了2（2是素数）以外的其他偶数都不是素数，除了3（3是素数）以外其他能被3整除的数都不是素数，所以我们每找到一个素数，那么它的整数倍都不是素数。

程序思路：

创建一个大小为n初始值为1的列表，当找到一个素数时，将它的倍数变为0.

程序：

import time
def prime(n):
    flag = [1]*(n+2)
    p=2
    while(p<=n):
        print(p)
        for i in range(2*p,n+1,p):
            flag[i] = 0
        while 1:
            p += 1
            if(flag[p]==1):
                break
start = time.time()
prime(100000)
end = time.time()
print(end-start)

时间：

59.229109048843384

这种优化时间消耗更少。

用优化一做题会出现超时

优化一，题目代码：

def prime(a):
    if a<=1:
        return False
    for i in range(2,int(a**0.5)+1):
        if a%i==0:
            return False
    return True
def sustr(a):
    result = []
    for i in range(a-1):
        if su(i+2)==True:
            result.append(i+2)
    if su(i+2)==True:
        result.append(i+2)
    return result

a = int(input())
b = sustr(a)
num = 0
for i in range(len(b)-1):
    if (b[i+1]-b[i]==2):
        num+=1
print(num)

用优化二做题不会出现超时

优化二，题目代码：

def prime(n,result):
    flag = [1]*(n+2)
    p=2
    while(p<=n):
        result.append(p)
        for i in range(2*p,n+1,p):
            flag[i] = 0
        while 1:
            p += 1
            if(flag[p]==1):
                break

a = int(input())
result = []
prime(a,result)
num = 0
for i in range(len(result)-1):
    if (result[i+1]-result[i]==2):
        num+=1
print(num)