python 3判断素数函数
代码:
#函数用来判断是否为素数
def isprime(n):
#n为2或3,可以直接判断是素数
if n == 2 or n == 3:
return True
#n可以被2或3整除,可以直接判断不是素数
if n % 2 == 0 or n % 3 == 0:
return False
#观察素数的分布可以发现,除2&3以外的素数,必定分布在6k(k为大于1的整数)的两侧。
#6k%6==0
#(6k+2)%2==0
#(6k+3)%3==0
#(6k+4)%2==0
#所以2&3以外的素数为6k+1或6k-1的形式,据此可以缩小因子范围
for k in range(6, int(math.sqrt(n)) + 2, 6):
if n % (k - 1) == 0 or n % (k + 1) == 0:
return False
return True