2022年天梯赛模拟赛 L1-8 均是素数 (20 分)(python)

在给定的区间 [m,n] 内，是否存在素数 p、q、r（p

输入格式：

输入给出区间的两个端点 0

输出格式：

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

输入样例：

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

代码

import math
def f(x):
    flag=1
    for i in range(2,int(math.sqrt(x))+1):
        if x%i==0:
            flag=0
    if x<=1:
        flag=0
    return flag
a,b=map(int,input().split())
m=[]
for i in range(a,b+1):
    if f(i)==1:
        m.append(i)
# print(m)
s=0
for i in range(0,len(m)):
    for j in range(i+1,len(m)):
        for k in range(j+1,len(m)):
            if f(m[i]*m[j]+m[k]) and f(m[j]*m[k]+m[i]) and f(m[k]*m[i]+m[j]):
#                 print(m[i],m[j],m[k])
                s+=1
print(s)