PTA 均是素数 python

在给定的区间 [m,n] 内，是否存在素数 p、q、r（p<q<r），使得 pq+r、qr+p、rp+q 均是素数？

输入格式：

输入给出区间的两个端点 0<m<n≤1000，其间以空格分隔。

输出格式：

在一行中输出满足条件的素数三元组的个数。

输入样例：

1 35

输出样例：

10

样例解读

满足条件的 10 组解为：

2, 3, 5
2, 3, 7
2, 3, 13
2, 3, 17
2, 5, 7
2, 5, 13
2, 5, 19
2, 5, 31
2, 7, 23
2, 13, 17

分析公式，会发现，如果三个数都是奇数，则（2n1+1）*(2n2+1)+2n3+1=4n1n2+2(n1+n2+n3)+2

显然，结果必为偶数，也即必不为素数。而素数中只有2为偶数，即三个数中必有一个数为2，而且2为最小的素数。所以，只要两重循环即可，python的代码就不会超时，当然，需要多加分支。

代码如下：

import math

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

a,b=map(int,input().split())
c,d=[],0
for i in range(a,b+1):
    if isPrime(i):
        c.append(i)
for a2 in range(len(c)):
    for a3 in range(a2+1,len(c)):
        if isPrime(2*c[a2]+c[a3]) and isPrime(2*c[a3]+c[a2]) and isPrime(c[a2]*c[a3]+2):
            d+=1
if a<=2:
    print(d)
else:
    print(0)

 纯新人，写的菜，别见怪。