PAT 1007 素数对猜想 python代码（运行超时）

1007 素数对猜想

输入格式:

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

输出格式:

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

输入样例:

20

输出样例:

4

代码如下

import math
def prime(i):
    if i <3:
        return True
    elif i%2==0:
        return False
    else:
        count = 1
        for j in range(1,int(math.sqrt(i))+1):
            if i%j==0:
                count+=1
        if count==2:
            return True
        else:
            return False
#输入
n=eval(input())
#找素数
prime_num = []
for i in range(1,n+1):
    if prime(i):
        prime_num.append(i)
#满足要求的素数对
out=0
for i in range(1,len(prime_num)):
    if prime_num[i]-prime_num[i-1]==2:
        out+=1
print(out)

得分