第一个没有优化,而第二个稍微优化了一些。当然,“Sieve of Eratosthenes”是最好的。此函数按顺序生成素数,但没有上限。
简单且未优化:
def simple_prime_list(num):
list_of_prime = (2, )
current_num = 2
is_prime = True
while len(list_of_prime) != num:
current_num += 1
for i in list_of_prime:
if current_num % i == 0:
is_prime = False
if is_prime == True:
list_of_prime += (current_num,)
#To reset the status
is_prime = True
return list_of_prime
通过在数字不是素数时不检查所有偶数和break 进行轻微优化:
def prime_list(num):
list_of_prime = (2, )
current_num = 2
is_prime = True
while len(list_of_prime) != num:
current_num += 1
if current_num % 2 != 0:
for i in list_of_prime:
if current_num % i == 0:
is_prime = False
break
if is_prime == True:
list_of_prime += (current_num, )
#To reset the status
is_prime = True
return list_of_prime
尝试测量 2 个不同的运行时间:
import time
def measureTime(fn):
start = time.clock()
fn()
end = time.clock()
#return value in millisecond
return (end - start)*1000
print('Simple Prime List:', measureTime(lambda: simple_prime_list(1000)), 'ms')
print('Optimised Prime List:', measureTime(lambda: prime_list(1000)), 'ms')
输出:
简单素数列表:775.798 毫秒
优化的 Prime 列表:69.48299999999996 毫秒