限定行数,输入两个正方形边长,求面积之差是否是质数(素数):
from numpy import *
while True:
try:
Q = int(input()) # 先输入一个数字代表需要输入几行,比如Q=4,那么就需要再输入4行数据
for i in range(Q):
a,b = [int(n) for n in input().split()]
s = a**2-b**2
if s<2:
print("这个数不是素数!")
else:
for i in range(2,s):
if s%i==0:
print("no")
break
else:
print("yes")
except:
break
结果:
2
9 3
no
4 3
yes