在Python中判断一个数是否为质数,可以通过以下几种方法:
暴力枚举法
def is_prime_brute_force(num):
if num < 2:
return False
for i in range(2, num):
if num % i == 0:
return False
return True
优化枚举法
import math
def is_prime_optimized(num):
if num < 2:
return False
for i in range(2, int(math.sqrt(num)) + 1):
if num % i == 0:
return False
return True
使用数学函数
def is_prime_math_function(num):
if num < 2:
return False
if num in [2, 3]:
return True
if num % 2 == 0:
return False
for i in range(3, int(math.sqrt(num)) + 1, 2):
if num % i == 0:
return False
return True
生成器方法
def is_prime_generator(num):
if num < 2:
return False
if num in [2, 3]:
return True
if num % 2 == 0:
return False
for i in range(3, int(math.sqrt(num)) + 1, 2):
if num % i == 0:
return False
yield True
判断一个数是否为素数
def is_prime(num):
if num < 2:
return False
for i in range(2, int(math.sqrt(num)) + 1):
if num % i == 0:
return False
return True
测试代码
num = int(input("请输入一个正整数:"))
if is_prime(num):
print(num, "是质数")
else:
print(num, "不是质数")
以上方法中,优化枚举法通过只检查到数的平方根来提高效率,而生成器方法则可以在处理大量数据时节省内存。选择合适的方法可以提高代码的性能和效率