python列表生成器二维筛选_如何加速Eratosthenes python列表生成器的筛选

编辑：已经意识到在SO上有很多优化,但很少有其他人为主筛算法解释它们,因此很难让初学者或第一次算法创建者接近它们.所有解决方案都将在python中,以便在速度和优化的同一页面上.这些解决方案将逐渐变得更快,更复杂. 🙂

香草溶液

def primesVanilla(n):

r = [True] * n

r[0] = r[1] = False

for i in xrange(n):

if r[i]:

for j in xrange(i+i,n,i):

r[j]=False

return r

这是Sieve非常简单的实现.在继续之前,请确保您了解上面发生的事情.唯一需要注意的是你在i i而不是i开始标记not-primes,但这很明显. (因为你认为我本身就是一个素数).为了使测试公平,所有数字将高达2500万.

real 0m7.663s

user 0m7.624s

sys 0m0.036s

次要改进1(平方根)：

我将尝试根据直接向不太直接的变化对它们进行排序.注意我们不需要迭代到n,而只需要上升到n的平方根.原因是n下的任何复合数必须具有低于或等于n的平方根的素因子.当你手工筛选时,你会注意到n的平方根上的所有“未被发现”的数字都是默认的素数.

另一个注意事项是,当平方根变成一个整数时,你必须要小心一点,所以你应该在这种情况下添加一个,以便它覆盖它. IE,在n = 49时,你想循环到7(包括7),或者你可能得出结论49是素数.

def primes1(n):

r = [True] * n

r[0] = r[1] = False

for i in xrange(int(n**0.5+1)):

if r[i]:

for j in xrange(i+i,n,i):

r[j]=False

return r

real 0m4.615s

user 0m4.572s

sys 0m0.040s

请注意,它要快得多.当你考虑它时,你只是循环到平方根,所以现在需要2500万顶级迭代只有5000顶级.

次要改进2(跳过内循环)：

观察在内循环中,我们可以从i * i开始,而不是从i i开始.这是从平方根的一个非常相似的论证得出的,但是最大的想法是i和i * i之间的任何复合都已经被较小的素数标记.

def primes2(n):

r = [True] * n

r[0] = r[1] = False

for i in xrange(int(n**0.5+1)):

if r[i]:

for j in xrange(i*i,n,i):

r[j]=False

return r

real 0m4.559s

user 0m4.500s

sys 0m0.056s

那有点令人失望.但是,嘿,它仍然更快.

一些重大改进3(甚至跳过)：

这里的想法是我们可以预先标记所有偶数索引,然后在主循环中跳过迭代2.之后我们可以在3开始外循环,内循环可以跳过2 * i. (因为我改为暗示它会是偶数,(我)(我是我)等)

def primes3(n):

r = [True] * n

r[0] = r[1] = False

for i in xrange(4,n,2):r[i]=False

for i in xrange(3,int(n**0.5+1),2):

if r[i]:

for j in xrange(i*i,n,2*i):

r[j]=False

return r

real 0m2.916s

user 0m2.872s

sys 0m0.040s

酷改进4(wim的想法)：

此解决方案是一个相当高级的技巧.切片分配比循环更快,因此使用python的切片表示法. [R [开始：结束：跳过]

def primes4(n):

r = [True] * n

r[0] = r[1] = False

r[4::2] = [False] * len(r[4::2])

for i in xrange(3,int(1 + n**0.5),2):

if r[i]:

r[i*i::2*i] = [False] * len(r[i*i::2*i])

return r

10 loops, best of 3: 1.1 sec per loop

轻微改进5

请注意,python在计算长度时会复制r [4 :: 2],因此这需要相当多的额外时间,因为我们需要的只是计算长度.我们确实使用了一些讨厌的数学来实现这个目标.

def primes5(n):

r = [True] * n

r[0] = r[1] = False

r[4::2] = [False] * ((n+1)/2-2)

for i in xrange(3,int(1 + n**0.5),2):

if r[i]:

r[i*i::2*i] = [False] * ((n+2*i-1-i*i)/(2*i))

return r

10 loops, best of 3: 767 msec per loop

分配加速(Padraic Cunningham)

请注意,我们为所有True分配一个数组,然后将half(均匀)设置为False.实际上我们可以从一个交替的布尔数组开始.

def primes6(n):

r = [False,True] * (n//2)+[True]

r[1],r[2]=False, True

for i in xrange(3,int(1 + n**0.5),2):

if r[i]:

r[i*i::2*i] = [False] * ((n+2*i-1-i*i)/(2*i))

return r

10 loops, best of 3: 717 msec per loop

不要引用我这个,但我认为没有一些讨厌的数学方法,这个最后版本没有明显的改进.我尝试过的一个可爱的属性,但结果并不快,注意到2,3以外的素数必须是6k 1或6k-1的形式. (注意,如果它是6k,那么可以被6,6k 2 | 2,6k 3 | 3,6k 4 | 2,6k 5整除与-1 mod 6一致.这表明我们每次可以跳过6并检查两者无论是从我身边的糟糕实施,还是蟒蛇内部,我都无法找到任何有意义的速度提升.:(